pytorch的函数(如squeeze,permute,flatten,stack,numel,matmul)
squeeze消去一个或多个维度。只有在指定的维度长度=1时才生效。a.squeeze(0):消去第1个维度(这里的索引是从0开始的,和数组索引一样)a.squeeze(0,2):消去第1、3个维度a.squeeze():消去所有长度=1的维度如果指定的维度长度不为1,不报错但是也不会有任何作用.摘自PyTorch入门——数组维度及squeeze、unsqueezeunsqueeze增加一个长度=