resize函数c++Pytorch（⼀）——相关库和函数查看Pytorch版本
print(torch.__version__)
查看cuda版本
print(torch.version.cuda)
查看cudnn版本
print(torch.backends.cudnn.version())
查看GPU型号
print(_device_name(0))
Pytorch是否可以使⽤计算机的GPU
torch.cuda.is_available()
True就是可以被使⽤
tensor 数据类型转换
torch.long() #将tensor转换为long类型torch.half() #将tensor转换为半精度浮点类型torch.int() #将该tensor转换为int类型torch.double() #将该tensor转换为double类型torch.float() #将该tensor转换为float类型torch.char() #将该tensor转换为char类型torch.byte() #将该tensor转换为byte类型torch.short() #将该tensor转换为short类型
将tensor转换为另⼀个tensor的shape
x = x.type_as(t1)
torch.spmm()/()  稀疏矩阵相乘
稀疏矩阵: 矩阵中⾮零元素的个数远远⼩于矩阵元素的总数，并且⾮零元素的分布没有规律，则称该矩阵为稀疏矩阵(sparse matrix) (a,b)⽤法上类似(a,b)
但a是稀疏矩阵
python3.7是torch.spmm()和()都有

版权声明：本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198，我们将在24小时内删除。