我怎样才能得到一个轴上5个张量的平均值?

你没有提到这5个张量是如何存储的,但让我们假设它们在一个列表中。以下是一种方法:

import torch

x = [torch.rand((7, 20, 180)) for _ in range(5)]
y = torch.stack(x).mean(dim=0)

print(y.shape)
# >>> torch.Size([7, 20, 180])

我也假设你说 dim=1 就好像PyTorch是基于1的索引,但事实并非如此。

_{我看到你最近问了很多问题,这不是问题。我说过一次,但又一次:始终努力提供 Minimal, Reproducible Example 展现出一些努力总是好的。问之前你试过什么吗?}