.mean(dim=[2, 3]) 代码举例说明

假设有一个形状为 (4, 3, 5, 5) 的张量 `x`，我们可以对它的第二和第三维进行平均：

import torch

x = torch.randn(4, 3, 5, 5)  # 创建一个形状为 (4, 3, 5, 5) 的张量
y = x.mean(dim=[2, 3])     # 对第二和第三维进行平均
print(y.shape)             # 输出新张量的形状，应为 (4, 3)

输出结果为：

torch.Size([4, 3])

这里的 `y` 是一个新的张量，它的形状为 (4, 3)，表示对 `x` 的第二和第三维进行了平均，压缩后的结果。