Tensor的访问操作
Tensor的主要运算操作通常分为四大类:
- Reshaping operations(重塑操作)
- Element-wise operations(元素操作)
- Reduction operations(缩减操作)
- Access operations(访问操作)
访问操作
当我们希望从一个张量中获得具体的标量值时,可以对张量进行访问操作。
(1)访问单元素张量(item张量法)
t = torch.tensor([
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
], dtype=torch.float32)
t.mean()
显示结果:
tensor(5.)
访问标量值
t.mean().item()
显示结果:
5.0
(2)访问张量中的多个值
-
将输出张量转换成python列表
t.mean(dim=0).tolist()
显示结果:
[4.0, 5.0, 6.0]
-
进入numpy数组来访问
t.mean(dim=0).numpy()
显示结果:
array([4., 5., 6.], dtype=float32)