如何获得沿轴的5个张量的平均值？

Question

我有5个torch.Size([7, 20, 180])形的张量

我想获取dim=1的平均值，以便最终形状为torch.Size([7, 20, 180])。

基本上，我希望第一个张量中的第一个元素，然后第二个张量中的第一个元素，依此类推。

Answer 1

您没有提到这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。付出一些努力总是一件好事。你问之前有尝试过什么吗？}