class_docstr: A base class for modules that output distributions.
class_docstr_tok: ["A","base","class","for","modules","that","output","distributions","."]
docstr: Forward pass of the module.
docstr_tok: ["Forward","pass","of","the","module","."]
code: def __call__(self, *args, **kwargs) -> tfp.distributions.Distribution:
      """Forward pass of the module."""
  
file_path: acme/acme/tf/networks/base.py
imports: import abc
  from typing import Tuple, TypeVar
  from acme import types
  import sonnet
  import tensorflow_probability
label_desc: Use image node_id 1 for calling the DistributionalModule obj's underlying member method code with example usage: obj.__call__() without return types