
Pytorch
Mic_hu
这个作者很懒,什么都没留下…
展开
-
Pytorch.item()的使用
Pytorch.item()是得到一个元素张量里面的元素值 >>> a = torch.Tensor([15]) >>> a tensor([15.]) >>> a.item() 15.0 >>>原创 2020-08-10 10:58:44 · 573 阅读 · 0 评论 -
Pytorch入门
Pytorch入门张量新建张量张量运算张量的转换 张量 新建张量 1.生成一个空张量torch.empty import torch x = torch.empty(5,3) print(x) 2.rand(*size, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) -> Tensor返回一个张量,该张量由间隔上均匀分布的随机数填充:math:[0,1) import torch转载 2020-08-01 19:54:01 · 276 阅读 · 0 评论