Pytorch之Tensor

Posted 2020-12-12 szcloud

Tensor 维度可为 0，1，.......

1、

import torch as t

tensorA=t.tensor([[1,2,3]])
tensorB=t.tensor(10)
testResult= tensorA + tensorB

print(tensorB.size())
print(testResult)
print(testResult.size())

运行结果：

torch.Size([]) #0维
tensor([[11, 12, 13]]) #2维
torch.Size([1, 3]) #结果维度数tensorA  2维

2、

import torch as t

tensorA=t.tensor([[1,2,3]])
tensorB=t.tensor([10])
testResult= tensorA + tensorB

print(tensorB.size())
print(testResult)
print(testResult.size())

运行结果：

torch.Size([1]) #1维
tensor([[11, 12, 13]]) #2维
torch.Size([1, 3])  #结果维数同tensorA 2维

3、

import torch as t

tensorA=t.tensor([[1,2,3]])
tensorB=t.tensor([[10]])
testResult= tensorA + tensorB

print(tensorB.size())
print(testResult)
print(testResult.size())

运行结果：

torch.Size([1, 1])   #2维
tensor([[11, 12, 13]])   #2维

torch.Size([1, 3])  #结果维数同tensorA  2维

4、

import torch as t

tensorA=t.tensor([[1,2,3]])
tensorB=t.tensor([[[10]]])
testResult= tensorA + tensorB

print(tensorB.size())
print(testResult)
print(testResult.size())

运行结果：

torch.Size([1, 1, 1])  #3维
tensor([[[11, 12, 13]]]) #3维
torch.Size([1, 1, 3])  #结果维数同tensorB  3维

。。。

结论：PyTorch中一个tensor(tensorA)加上一个数tensorB(实际也是tensor ，只不过可能是0维、1维等) ,结果维数为两者的最大值