[ 输入参数类型不一致 ]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)