One using the size() method and another by using the shape attribute of a tensor in PyTorch. ... v = torch.tensor([[ 1 , 0 ],[ 0 , 1 ]]).
確定! 回上一頁