[ 参数不一致 ]torch.nn.functional.one_hot¶

torch.nn.functional.one_hot¶

torch.nn.functional.one_hot(tensor,
                            num_classes=- 1)

paddle.nn.functional.one_hot¶

paddle.nn.functional.one_hot(x,
                             num_classes,
                             name=None)

两者功能一致，但 Paddle 的 num_classes 没有指定默认值，需要手动指定，具体如下：

参数映射¶

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | tensor | x | 表示输入的 Tensor 。 | | num_classes | num_classes | 表示一个 one-hot 向量的长度， Paddle 无默认值，必须显式指定，需要转写。 | | 返回值 | 返回值 | PyTorch 返回值类型为 int64，Paddle 返回值类型为 float32，需要转写。 |

转写示例¶

# PyTorch 写法
y = torch.nn.functional.one_hot(x)

# Paddle 写法
y = paddle.nn.functional.one_hot(x, num_classes=x.max().item() + 1)

# PyTorch 写法
y = torch.nn.functional.one_hot(x, num_classes=2)

# Paddle 写法
y = paddle.nn.functional.one_hot(x, num_classes=2).astype('int64')