The same operation can be directly performed on # x by calling squeeze and the dimension to squeeze out z2 = x.squeeze(1) assert torch.all(z1 z2) # all the ...
確定! 回上一頁