code: def from_logits(behaviour_policy_logits,
                  target_policy_logits,
                  actions,
                  discounts,
                  rewards,
                  values,
                  bootstrap_value,
                  clip_rho_threshold=1.0,
                  clip_pg_rho_threshold=1.0,
                  name="vtrace_from_logits"):
      
  
      res = multi_from_logits(
          [behaviour_policy_logits], [target_policy_logits], [actions],
          discounts,
          rewards,
          values,
          bootstrap_value,
          clip_rho_threshold=clip_rho_threshold,
          clip_pg_rho_threshold=clip_pg_rho_threshold,
          name=name)
  
      return VTraceFromLogitsReturns(
          vs=res.vs,
          pg_advantages=res.pg_advantages,
          log_rhos=res.log_rhos,
          behaviour_action_log_probs=tf.squeeze(
              res.behaviour_action_log_probs, axis=0),
          target_action_log_probs=tf.squeeze(
              res.target_action_log_probs, axis=0),
      )
docs: multi_from_logits wrapper used only for tests