ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


Set bandit policy type tensor to BanditPolicyType.UNIFORM based on mask.

Set bandit policy type values to BanditPolicyType.UNIFORM; returns tensor where output[i] is BanditPolicyType.UNIFORM if mask[i] is True, otherwise it is left as values[i].

values Tensor containing BanditPolicyType enumerations.
mask Tensor of the same shape as values with boolean flags indicating values to set to BanditPolicyType.UNIFORM.

Tensor containing BanditPolicyType enumerations with masked values.