摘要: 1 numpy与CUDA之间的转换 1.tensor张量与numpy相互转换 tensor >numpy import torcha=torch.ones([2,5]) tensor([[1., 1., 1., 1., 1.], [1., 1., 1., 1., 1.]])# *********** 阅读全文
posted @ 2020-02-22 19:29 lqchen 阅读(23391) 评论(0) 推荐(1) 编辑
摘要: torch.ge torch.ge(input, other, out=None) → Tensor 逐元素比较input和other,即是否 input>=otherinput>=other。 如果两个张量有相同的形状和元素值,则返回True ,否则 False。 第二个参数可以为一个数或与第一个 阅读全文
posted @ 2020-02-22 19:24 lqchen 阅读(4766) 评论(0) 推荐(0) 编辑
点击右上角即可分享
微信分享提示