【Pytorch】Tensor的访问操作

Tensor的访问操作

Tensor的主要运算操作通常分为四大类：

1. Reshaping operations（重塑操作）
2. Element-wise operations（元素操作）
3. Reduction operations（缩减操作）
4. 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）访问张量中的多个值

1. 将输出张量转换成python列表

   t.mean(dim=0).tolist()
   

   显示结果：

   [4.0, 5.0, 6.0]
   

2. 进入numpy数组来访问

   t.mean(dim=0).numpy()
   

   显示结果：

   array([4., 5., 6.], dtype=float32)