摘要: 转自:https://blog.csdn.net/xiexu911/article/details/80820028 1.torch.squeeze() 只会去掉维度为1的那个维度。它只会去掉维度为1的维度,像下面的没有为1的维度,就不会改变: >>> aaa=np.ones((3,2)) >>> 阅读全文
posted @ 2021-06-22 15:18 lypbendlf 阅读(112) 评论(0) 推荐(0) 编辑