参考:
torch.Tensor是一种包含单一数据的多维矩阵。
Torch定义了七种CPU张量和八种GPU张量类型:
图片转自https://ptorch.com/docs/8/torch-tensor
torch.Tensor是默认的tensor类型(torch.FloatTensor)的简称。
# torch.Tensor https://ptorch.com/docs/8/torch-tensor
#张量可以从Python的list或者序列构成
t1 = torch.FloatTensor([[1,2,3],[4,5,6]])
print("torch.FloatTensor([[1,2,3],[4,5,6]]): \n",t1)
#可以通过指定它的大小来构建一个空的张量
t2 = torch.IntTensor(2,4).zero_()
print("torch.FloatTensor([[1,2,3],[4,5,6]]): \n",t2)
运行结果:
#可以使用Python的索引和切片符号来访问和修改张量的内容:
x = torch.FloatTensor([[1,2,3],[4,5,6]])
print("x: \n",x)
x[0][1] = 8
print("x[0][1] = 8:\n",x)
运行结果:
#element_size() ---> int 返回单个元素的字节大小。
y1 = torch.FloatTensor().element_size()
print("torch.FloatTensor().element_size():\n",y1)
y2 = torch.FloatTensor(3,4).element_size()
print("torch.FloatTensor(3,4).element_size():\n",y2)
y3 = torch.ByteTensor().element_size()
print("torch.ByteTensor().element_size():\n",y3)
运行结果:
### expand(*sizes) 返回tensor的一个新视图,,单个维度扩大为更大的尺寸。tensor也可以扩大为更高维,新增加的维度将附在前面。
# 扩大tens