[ 返回参数类型不一致 ]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，需要转写。

转写示例

num_classes: one-hot 向量的长度

# 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')