Skip to content

[Feature] Force log_prob to return a tensordict when kwargs are passed to ProbabilisticTensorDictSequential.log_prob #1079

[Feature] Force log_prob to return a tensordict when kwargs are passed to ProbabilisticTensorDictSequential.log_prob

[Feature] Force log_prob to return a tensordict when kwargs are passed to ProbabilisticTensorDictSequential.log_prob #1079

Annotations

1 error

pytorch/tensordict (pytorch/tensordict, test/smoke_test.py, tensordict)  /  ...  /  manywheel-py3_9-cuda12_6

failed Dec 18, 2024 in 9s