python - 在 PyTorch 中将 5D 张量转换为 4D 张量

Question

在 PyTorch 中，我有一个 5D 张量X的尺寸B x 9 x C x H x W。我想将其转换为Y具有维度的 4D 张量，B x 9C x H x W以便串联发生在通道上。

为了说明让，

a = X[1,0,:,:,:]
b = X[1,1,:,:,:]
c = X[1,2,:,:,:]
...
i = X[1,8,:,:,:]

然后在张量Y中，a to i应该按通道连接。

标签： pythonpytorchtensor

Answer 1

您可以使用以下命令轻松广播到新形状torch.reshape：

b, n, c, h, w = X.shape
X = X.reshape(b, n*c, h, w)