PyTorch: 将 [4, 256, 256, 3] 张量转换为 [4, 256, 256]

可以使用 PyTorch 的 squeeze() 函数将维度为 3 的维度移除，代码如下：

import torch

# 创建一个维度为 [4, 256, 256, 3] 的张量
x = torch.randn(4, 256, 256, 3)

# 将维度为 3 的维度移除
x = x.squeeze(3)

# 打印转换后的张量维度
print(x.shape)  # 输出: torch.Size([4, 256, 256])

在 squeeze() 函数中传入的参数 3 表示要移除的维度，由于在 PyTorch 中索引从 0 开始，因此 3 表示的是第四个维度。如果不确定要移除的维度是几，可以使用 dim 属性查看每个维度的大小，然后根据需要移除对应的维度。