PyTorch: How to get the shape of a Tensor as a list of int ... import torch >>> var = torch.tensor([[1,0], [0,1]]) # Using .size function, returns a torch.
確定! 回上一頁