unsqueeze 用來改變Tensor的維度,把低維的Tensor變為高維的Tensor。 ... print(a.unsqueeze(0).shape) print(a.unsqueeze(1)) print('-'*10) ...
確定! 回上一頁