2018.11.15——pytorch中的torch.squeeze()和torch.mean()

torch.squeeze(x ,3)对第3个维度上的1去掉

torch.squeeze(x ,7)对第7个维度上的1去掉

========================================

猜你喜欢

转载自blog.csdn.net/jingtingxu369/article/details/84101629