Ptt 大爆卦 | PyTorch torch - 前往 https://stackoverflow.com/questions/46826218/pytorch-how-to-get-the-shape-of-a-tensor-as-a-list-of-int

你即將離開本站

並前往https://stackoverflow.com/questions/46826218/pytorch-how-to-get-the-shape-of-a-tensor-as-a-list-of-int

PyTorch: How to get the shape of a Tensor as a list of int

For PyTorch v1.0 and possibly above: >>> import torch >>> var = torch.tensor([[1,0], [0,1]]) # Using .size function, returns a torch.

確定！回上一頁

查詢「PyTorch torch」的人也找了：

PyTorch tutorial

Pip install pytorch