[Bug] Fixes bug in GAE advantage estimation when `gammalmbda` is a Tensor #2127

Job	Run time
generate-matrix / generate	8s
pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl) / manywheel-py3_9-cpu	9m 1s
pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl) / manywheel-py3_9-cuda11_8	6m 37s
pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl) / manywheel-py3_9-cuda12_4	9m 54s
pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl) / manywheel-py3_9-cuda12_6	9m 28s
pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl) / manywheel-py3_9-cuda12_8	10m 34s
pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl) / manywheel-py3_9-rocm6_2_4	15m 53s
pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl) / manywheel-py3_9-rocm6_3	15m 30s
pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl) / upload / manywheel-py3_9-cpu	14s
pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl) / upload / manywheel-py3_9-cuda11_8	20s
pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl) / upload / manywheel-py3_9-cuda12_4	17s
pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl) / upload / manywheel-py3_9-cuda12_6	14s
pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl) / upload / manywheel-py3_9-cuda12_8	13s
pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl) / upload / manywheel-py3_9-rocm6_2_4	18s
pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl) / upload / manywheel-py3_9-rocm6_3	14s
	1h 18m 55s

Provide feedback