class_docstr: Simple sonnet module to produce the mean of a tfp.Distribution.
class_docstr_tok: ["Simple","sonnet","module","to","produce","the","mean","of","a","tfp.Distribution","."]
code: def __call__(self, distribution: tfd.Distribution):
      return distribution.mean()
  
file_path: acme/acme/tf/networks/stochastic.py
imports: from acme import types
  from acme.tf import utils
  import sonnet
  import tensorflow
  import tensorflow_probability
  import tree
label_desc: Use image node_id 1 for calling the StochasticMeanHead obj's underlying member method code with example usage: obj.__call__(distribution) and returns: distribution