![](https://img-blog.csdnimg.cn/20201014180756923.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 68
没心没肺的快乐
这个作者很懒,什么都没留下…
展开
-
Pytorch系列-torch.Tensor
目录属性数据类型devicelayoutmemory_formatViews属性数据类型device是在cpu上还是cuda上# 设置devicedevice = torch.device('cuda:0')device = torch.device('cuda',0)# 在cuda上初始化张量torch.randn((2,3), device=device)torch.randn((2,3), device='cuda:1')# 获取张量位置a.get_device() #原创 2021-05-12 16:30:57 · 758 阅读 · 1 评论 -
Pytorch系列-torch基础
目录TensorsTensors属性含义is_tensor是不是一个PyTorch tensoris_storage是不是一个PyTorch storage objectis_complex是不是一个复数: torch.complex64, torch.complex128is_floating_point是不是浮点类型: torch.float64, torch.float32, torch.float16is_nonzero是不是非零set原创 2021-05-12 15:26:25 · 1169 阅读 · 1 评论