标量 (Scalar)

标量 (Scalar) 由一个值组成. (0维)

# Scalar
x = tensor(42,)
print(x)
print(x.dim())
print(x * 2)
print(x.item())

  

 

  
1
  
2
  
3
  
4
  
5
  
6
 

输出结果:

tensor(42)
0
tensor(84)

  

 

  
1
  
2
  
3
 

向量 (Vector)

向量 (Vector) 为一维.

例子:

# Vector
v = tensor([1.5, -0.5, 3.0])
print(v)
print(v.dim())
print(v.size())

  

 

  
1
  
2
  
3
  
4
  
5
 

输出结果:

42
tensor([ 1.5000, -0.5000,  3.0000])
1

  

 

  
1
  
2
  
3
 

矩阵 (Matrix)

矩阵 (Matrix) 是二维的.

# Matrix
M = tensor([[1.,2.], [3.,4.]])
print(M.matmul(M))
print(tensor([1.,0.]).matmul(M))

  

 

  
1
  
2
  
3
  
4
 

输出结果:

torch.Size([3])
tensor([[ 7., 10.],
        [15., 22.]])
tensor([1., 2.])

  

 

  
1
  
2
  
3
  
4
 

图片展示

文章来源: iamarookie.blog.csdn.net，作者：我是小白呀，版权归原作者所有，如需转载，请联系作者。

原文链接：iamarookie.blog.csdn.net/article/details/114176589