在 Pytorch 中计算 4D 张量的一个特定维度的平均值

Question

我有一个形状为 [66,7,7,1024] 的 PyTorch 视频特征张量，我需要将其转换为 [1024,66,7,7]。如何重新排列张量形状？另外，如何在dimension=1 中执行均值？即，在执行尺寸为 66 的平均之后，我需要张量为 [1024,1,7,7]。

我尝试计算维度=1 的平均值，但未能将其替换为平均值。我无法想象一个 4D 张量，其中一维被其均值取代。

编辑： 我试过torch.mean（my_tensor，dim = 1）。但这会给我返回一个形状为 [1024,7,7] 的张量。 4D 张量正在转换为 3D。但我希望它保持 4D 形状，形状为 [1024,1,7,7]。

非常感谢。

Answer 1

问题的第一部分已在 cmets 部分得到解答。所以我们可以使用tensor.transpose([3,0,1,2])将张量转换为[1024,66,7,7]的形状。

现在时间维度上的平均值可以通过 torch.mean(my_tensor, dim=1) 这将给出一个形状为[1024,7,7] 的 3D 张量。 为了获得形状为[1024,1,7,7] 的张量，我必须在维度=1 中解压：

tensor = tensor.unsqueeze(1)