Tensor对于pytorch就如同ndarray,而且和ndarray很像,但有一个重要的区别就是Tensor支持GPU,ndarray不支持。
注意为了说明方便,
1.Tensor分类:
ByteTensor,唯一的一个无符号类型Tensor,8位无符号整型(下面皆有符号)
CharTensor,8位有符号整型
ShortTensor,16位整型
IntTensor,32位整型
LongTensor,64位整型
FloatTensor,32位浮点,默认类型
DoubleTensor,64位浮点
2.初始化
(1)通过其他类型对象初始化
torch.Tensor([[1,2],[3,4]])
torch.from_numpy(ndarray) 返回的Tensor与参数ndarray共享内存
(2) 通过函数初始化
torch.zeros((n1,n2…),out=None)用元组传入维数信息
torch.eye(n,m=None,out=None) 只能返回2维对角阵
很多函数运算结果也会返回一个Tensor
(3)随机初始化
3.获取Tensor的相关参数
torch.numel(a)获取元素个数
a.size()返回一个torch.Size类型,可以当作tensor,支持索引
4.类型转换
a.float()
等即可
a.numpy()
可转为ndarray,同样也共享内存
5.获取及修改
支持索引和切片。