1. torch.squeeze(x,N) 主要对数据维度进行压缩torch.squeeze(x,N) #也可以写为格式x.squeeze(dim=N) 含义:当N未给定时,去掉x中所有.
確定! 回上一頁