[torch 参数更多]torch.distributions.dirichlet.Dirichlet

torch.distributions.dirichlet.Dirichlet

torch.distributions.dirichlet.Dirichlet(concentration, validate_args=None)

paddle.distribution.Dirichlet

paddle.distribution.Dirichlet(concentration)

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

参数映射

PyTorch PaddlePaddle 备注
concentration concentration 浓度参数,即公式中 α 参数。
validate_args - 是否添加验证环节。Paddle 无此参数,一般对训练结果影响不大,可直接删除。