tensor  一共包含的数据类型，数据

class DoubleTensor(Tensor): ...
class FloatTensor(Tensor): ...
class LongTensor(Tensor): ...
class IntTensor(Tensor): ...
class ShortTensor(Tensor): ...
class CharTensor(Tensor): ...
class ByteTensor(Tensor): ...
class BoolTensor(Tensor): ...

 例子：

import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.autograd import Variable
b = torch.rand(3,3)#得到的是floattensor值，
print(type(b),b,b.type())
a = b.long()#得到的是longtensor值
print(type(a),a,a.type())
d=b.int()
print(type(d),d,d.type())

运行结果：

PS F:Graduation projectdeeplearning> python .c.py
<class 'torch.Tensor'> tensor([[0.4874, 0.3303, 0.6307],
        [0.4765, 0.2642, 0.4723],
        [0.2643, 0.5587, 0.0493]]) torch.FloatTensor
<class 'torch.Tensor'> tensor([[0, 0, 0],
        [0, 0, 0],
        [0, 0, 0]]) torch.LongTensor
<class 'torch.Tensor'> tensor([[0, 0, 0],
        [0, 0, 0],
        [0, 0, 0]], dtype=torch.int32) torch.IntTensor

总结：

查看数据类型使用a.type() ， 数据转换.long()  其余类似

参考：https://blog.csdn.net/hustchenze/article/details/79154139

最后

以上就是不安芝麻为你收集整理的pytorch tensor 数据类型的转换总结：查看数据类型使用a.type() ， 数据转换.long()  其余类似的全部内容，希望文章能够帮你解决pytorch tensor 数据类型的转换总结：查看数据类型使用a.type() ， 数据转换.long()  其余类似所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。