[torch 参数更多]torch.Tensor.size¶
torch.Tensor.size¶
torch.Tensor.size(dim=None)
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
dim | - | 表示获取大小的轴,Paddle 无此参数,需要转写。 |
转写示例¶
# PyTorch 写法
torch.tensor([-1, -2, 3]).size(0)
# Paddle 写法
paddle.to_tensor([-0.4, -0.2, 0.1, 0.3]).shape[0]