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