torch.squeeze()
压缩大小为1的维度
参考:https://pytorch.org/docs/stable/generated/torch.squeeze.html
torch.unsqueeze()
在指定的维度处增加一个维度,大小为1
参考:https://pytorch.org/docs/stable/generated/torch.unsqueeze.html
压缩大小为1的维度
参考:https://pytorch.org/docs/stable/generated/torch.squeeze.html
在指定的维度处增加一个维度,大小为1
参考:https://pytorch.org/docs/stable/generated/torch.unsqueeze.html