Language

unilab.algos.torch.flash_sac.update.compute_categorical_td_target¶

unilab.algos.torch.flash_sac.update.compute_categorical_td_target(support, target_log_probs, reward, dones, truncated, actor_entropy, gamma)[source]¶

Parameters:

support (Tensor)
target_log_probs (Tensor)
reward (Tensor)
dones (Tensor)
truncated (Tensor)
actor_entropy (Tensor)
gamma (float)

Return type:

Tensor