libtorch - Torch C++ (libtorch) 中是否提供概率分布？

Question

我正在尝试移植一个 Q 网络，它使用概率分布采样（NormalPython 中的类及其rsample函数）。这些在 PyTorch C++ 中可用吗？

Answer 1

正态分布在 C++ 前端中有点可用torch::randn（正态分布，均值 = 0 和偏差 = 1）函数，尽管它确实不提供rsample功能。您必须自己实现重新参数化技巧，但这不应该超过几行代码。

Answer 2

截至目前，libtorch 中还没有用于分发的前端（您可以在此处查看命名空间）。根据这个线程，显然它在未来更新的路线图上。
还有指向该线程中分发的内部 C++ 实现的链接。