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

tf_agents.bandits.environments.drifting_linear_environment.apply_givens_rotation

View source on GitHub

Applies a Givens rotation on tensor x.

tf_agents.bandits.environments.drifting_linear_environment.apply_givens_rotation(
    cosa, sina, axis_i, axis_j, input_x
)

Reference on Givens rotations: https://en.wikipedia.org/wiki/Givens_rotation

Args:

  • cosa: the cosine of the angle.
  • sina: the sine of the angle.
  • axis_i: the first axis of rotation.
  • axis_j: the second axis of rotation.
  • input_x: the input tensor.

Returns:

The rotated tensor (same shape as x).