Pytorch squeeze() 和 unsqueeze() 方法区别

1 增加维度 unsqueeze()

tensor = tensor.unsqueeze(0)

2 减少维度 squeeze()

tensor = tensor.squeeze(0)

猜你喜欢

转载自blog.csdn.net/ao1886/article/details/109152433
今日推荐