[torch 参数更多]torch.nn.functional.gumbel_softmax

torch.nn.functional.gumbel_softmax

torch.nn.functional.gumbel_softmax(logits, tau=1, hard=False, eps=1e-10, dim=- 1)

paddle.nn.functional.gumbel_softmax

paddle.nn.functional.gumbel_softmax(x, temperature=1.0, hard=False, axis=- 1, name=None)

Pytorch 相比 Paddle 支持更多其他参数,具体如下:

参数映射

PyTorch PaddlePaddle 备注
logits x 一个 N-D Tensor,前 N-1 维用于独立分布 batch 的索引,最后一维表示每个类别的概率,仅参数名不一致。
tau temperature 大于 0 的标量,仅参数名不一致。
hard hard 如果是 True,返回离散的 one-hot 向量。如果是 False,返回软样本。
eps - eps 参数,Pytorch 已废弃不会生效,Paddle 无此参数,对网络无影响,可直接删除。
dim axis 按照维度 axis 计算 softmax,仅参数名不一致。