[Docs] Fix doc of ToTensorImage transforms.py (#1824) #147

Job	Run time
unittests-pettingzoo / linux-job	9m 15s
unittests-brax (3.9, 12.1) / linux-job	8m 55s
unittests-d4rl (3.9, 12.1) / linux-job	5m 35s
unittests-envpool (3.9, 12.1) / linux-job	8m 33s
unittests-gendgrl (3.9, 12.1) / linux-job	6m 36s
unittests-gym (3.9, 12.1) / linux-job	17m 5s
unittests-jumanji (3.9, 12.1) / linux-job	8m 29s
unittests-minari (3.9, 12.1) / linux-job	15m 15s
unittests-openx (3.9, 12.1) / linux-job	16m 51s
unittests-robohive (3.9, 12.1) / linux-job	9m 57s
unittests-roboset (3.9, 12.1) / linux-job	7m 15s
unittests-sklearn (3.9, 12.1) / linux-job	5m 41s
unittests-smacv2 (3.9, 12.1) / linux-job	18m 11s
unittests-vd4rl (3.9, 12.1) / linux-job	8m 0s
unittests-vmas (3.9, 12.1) / linux-job	12m 42s
	2h 38m 20s

Provide feedback