def from_logits()

in torchbeast/core/vtrace.py [0:0]

• 10 lines of code
• 1 McCabe index (conditional complexity)

def from_logits(
    behavior_policy_logits,
    target_policy_logits,
    actions,
    discounts,
    rewards,
    values,
    bootstrap_value,
    clip_rho_threshold=1.0,
    clip_pg_rho_threshold=1.0,