\u200E

[ 仅 paddle 参数更多 ]torch.nn.Module.float¶

torch.nn.Module.float ¶

          torch.nn.Module.float()

         

paddle.nn.Layer.to ¶

          paddle.nn.Layer.to(dtype="float32")

         

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

参数映射¶

PyTorch	PaddlePaddle	备注
-	dtype	转换的数据类型，Paddle 为 float32，需要转写。

转写示例¶

dtype 参数：转换的数据类型¶

           # PyTorch 写法:
module = torch.nn.Module()
module.float()

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

          