TensorFlow 2.0 Beta is available Learn more

tff.utils.update_state

Returns a new state with new values for fields in kwargs.

tff.utils.update_state(
    state,
    **kwargs
)

Defined in python/core/utils/computation_utils.py.

Args:

  • state: the structure with named fields to update.
  • **kwargs: the list of key-value pairs of fields to update in state.

Raises:

  • KeyError: if kwargs contains a field that is not in state.
  • TypeError: if state is not a structure with named fields.