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.

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

The updated tensor (same shape as x).