[ 输入参数用法不一致 ]torch.set_default_tensor_type¶

torch.set_default_tensor_type ¶

          torch.set_default_tensor_type(t)

         

          paddle.set_default_dtype(d)

         

其中 PyTorch 与 Paddle 的参数类型不一致，具体如下：

PyTorch	PaddlePaddle	备注
t	d	指定的默认张量类型，参数类型不一致。PyTorch 支持张量类型或其名称字符串（如 `torch.FloatTensor`，Paddle 支持直接指定 `dtype`（如 `paddle.float32`），需要转写。

           # PyTorch
torch.set_default_tensor_type(torch.FloatTensor)

# Paddle
paddle.set_default_dtype(paddle.float32)