torch.nn.Module.float()
paddle.nn.Layer.to(dtype="float32")
Paddle 相比 PyTorch 支持更多其他参数,具体如下:
# PyTorch 写法: module = torch.nn.Module() module.float() # Paddle 写法: module = paddle.nn.Layer() module.to(dtype="float32")