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

【问题标题】：How can I get the mean of 5 tensors along an axis?如何获得沿轴的 5 个张量的平均值？
【发布时间】：2020-07-08 04:39:36
【问题描述】：

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

我想得到每个沿 dim=1 的平均值，这样我的最终形状将是 torch.Size([7, 20, 180])。

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

【问题讨论】：

【解决方案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。表现出一些努力总是好的。问之前有没有试过？

【讨论】：