pytorch将一个维度为4 256 256 3的tensor转换为4 256 256

可以使用PyTorch的squeeze()函数来完成这个操作。

import torch

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

# 使用squeeze()函数将最后一维去掉
x = x.squeeze()

# 打印新的tensor的shape
print(x.shape)  # 输出: torch.Size([4, 256, 256])

这样就可以将一个维度为[4, 256, 256, 3]的tensor转换为[4, 256, 256]的tensor。