torch.size ():查看当前Tensor的维度,用法也很简单:终端进入Python环境>>>import torch>>>a = torch.Tensor([[1, 2, 3], [4, 5, 6]])>>>a.size()(2 ...
確定! 回上一頁