[参数不一致]torch.nn.Module.type

torch.nn.Module.type

torch.nn.Module.type(dst_type)

paddle.nn.Layer.astype

paddle.nn.Layer.astype(dtype=None)

两者参数用法不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
dst_type dtype PyTorch 为字符串或 PyTorch 数据类型,Paddle 为 字符串或 Paddle 数据类型,需要转写。

转写示例

dst_type 参数:数据类型

# PyTorch 写法:
module = torch.nn.Module()
module.type(dst_type=torch.float32)

# Paddle 写法:
module = paddle.nn.Layer()
module.astype(dtype=paddle.float32)