Pytorch类型转换

2019-10-04  本文已影响0人  顾北向南

1. 直接调用变量类型

2. 变量调用pytorch中的type函数

self = torch.LongTensor(3, 5)
# 转换为其他类型
print self.type(torch.FloatTensor)

3. 变量调用pytorch中的type_as函数

self = torch.Tensor(3, 5)
tesnor = torch.IntTensor(2,3)
print self.type_as(tesnor)
上一篇 下一篇

猜你喜欢

热点阅读