[torch 参数更多]torch.Tensor.size

torch.Tensor.size

torch.Tensor.size(dim=None)

paddle.Tensor.shape

paddle.Tensor.shape

PyTorch 相比 Paddle 支持更多其他参数,具体如下:

参数映射

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]