ray.rllib.models.tf.tf_modelv2.TFModelV2.update_ops

TFModelV2.update_ops() → List[numpy.array | jnp.ndarray | tf.Tensor | torch.Tensor]

Return the list of update ops for this model.

For example, this should include any BatchNorm update ops.