Pytorch中的Tensor常用的类型转换函数
tensor数据类型转换
例如:
a = tensor(282, device=’cuda:0′)
b = int(a)
print(b) # 282,int类型
在tensor的后面添加: .int()、.float()、.double()等.
同时也可以使用 .to(type) 进行实现。
同时也可以使用 type()函数 ,data为Tensor数据类型,data.type()为给出data的类型,如果使用data.type(torch.FloatTensor)则强制转换为torch.FloatTensor类型张量。当你不知道要转换为什么类型时,但需要求a1,a2两个张量的乘积,可以使用a1.type_as(a2)将a1转换为a2同类型。
tensor
Original: https://blog.csdn.net/qq_42262610/article/details/124624770
Author: passion-ma
Title: tensor中数据类型的相互转换
原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/672536/
转载文章受原作者版权保护。转载请注明原作者出处!