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 project\deeplearning> 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