(discount, discount_mask)
return fastmath.numpy.where(discount_mask, discount, 1.0)
Computes a discount to apply at a given timestep, based on the mask.
Computes a discount to apply at a given timestep, based on the mask.
Computes a discount to apply at a given timestep based on the mask .
def mask_discount(discount, discount_mask):
  """Computes a discount to apply at a given timestep, based on the mask."""
  return fastmath.numpy.where(discount_mask, discount, 1.0)
def mask_discount ( discount , discount_mask ) : return fastmath . numpy . where ( discount_mask , discount , 1.0 )