tensor([-8.2506, -3.6277, -2.0539, -4.3296, -9.1765, -4.4646, -5.9651, -4.4289,
        -3.9373, -0.4725, -2.3407, -5.3857, -3.7996, -5.1204, -4.2914, -6.8435,
        -5.2565, -4.1634, -5.0635, -5.7674], device='cuda:0',
       grad_fn=<SubBackward0>)
