Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf_agents.bandits.environments.drifting_linear_environment.update_row

View source on GitHub

Updates the i-th row of tensor x with the values given in updates.

tf_agents.bandits.environments.drifting_linear_environment.update_row(
    input_x, updates, row_index
)

Args:

  • input_x: the input tensor.
  • updates: the values to place on the i-th row of x.
  • row_index: which row to update.

Returns:

The updated tensor (same shape as x).