删除 `TorchDistributionMixin` 的 `Callable` 父类
作者: kosiokarchev创建于 2025年7月19日更新于 2025年7月24日
是否可以从 TorchDistributionMixin 的父类中移除 Callable (在 1.9 版本中引入的)?我想创建一个通用的子类,如
class MyDist(TorchDistribution, Generic[_T]): ... 并将其用作 MyDist[SomeConcreteType],但这会失败,因为 Callable 强制了特定的使用模式 MyDist[[arg_type, ...], return_type]。另一方面,在我的经验中,反向工具和内置的 callable 已经从 __call__ 方法中获取了“可调用性”。
内容来源: pyro-ppl/pyro