Tensor 0 device cuda 0 (5/31)

Tensor 0 device cuda 0