qq_41626743的博客

torch.squeeze / torch.unsqueeze 消除tensor维度数为 1 的维度，添加维度数为 1 的维度。1- 首先squeeze，当且仅当tensor存在维度数为 1 时，生效。先从二维来考虑。当tensor的形状为1*3时，a = torch.randn(1,3) print ( a )b = torch.squeeze( a , dim = 0) # dim =0对应（1,3）中的 1print ( b )显示结果：a = [ [ 1 ...