[ 返回参数类型不一致 ]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')