1、torch.squeeze()torch.squeeze() 对数据的维度进行压缩;去掉维数为1的的维度。squeeze(a)将a中所有为1的维度删掉,不为1的维度没有影响。a.squeeze(N) 就是去掉a中 ...
確定! 回上一頁