Pytorch类型转换
作者:
顾北向南 | 来源:发表于
2019-10-04 13:12 被阅读0次
1. 直接调用变量类型
- 通过newtensor= tensor.long() (int(),byte(),float(),byte(),double(),half()),返回值才是改变类型后的结果，调用的变量类型不变.
2. 变量调用pytorch中的type函数
- type(new_type=None, async=False)如果未提供new_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)
本文标题：Pytorch类型转换
本文链接：https://www.haomeiwen.com/subject/ayxspctx.html
网友评论