tensor([[ -2.9684,  -3.4014,  -2.0095,  -1.4574,  -6.7830,  -1.2359],
        [ -0.8889,  -3.6135,  -1.1353,  -3.3458,  -0.8562,  -1.1741],
        [ -6.5085,  -1.1030,  -2.9286,  -1.2129,  -4.7519,  -5.6026],
        [ -1.9646,  -1.2378,  -1.1358,  -2.1171,  -2.7243,  -0.7384],
        [ -8.7790,  -1.2444,  -3.6610,  -5.2956,  -3.9955,  -3.2031],
        [ -3.9672,  -3.1031,  -2.1246,  -1.0424,  -2.6532,  -3.5371],
        [ -1.2853,  -0.6505,  -1.9386,  -1.5930,  -0.5268,  -1.4056],
        [ -4.4664,  -1.1804,  -2.3873,  -1.1299,  -1.7042,  -0.9837],
        [ -4.1294,  -1.1922,  -2.1739,  -0.9211,  -2.7585,  -2.2968],
        [ -2.5573,  -1.2021,  -5.0752,  -0.8410,  -1.0085,  -1.2118],
        [ -3.5939,  -1.1385,  -1.2657,  -0.7873,  -2.4899,  -2.3558],
        [ -8.4347,  -2.3121,  -3.6365,  -1.0700,  -2.4081,  -0.8068],
        [ -1.4471, -12.7542,  -2.4971,  -1.0606,  -5.4621,  -1.1241],
        [ -2.0968,  -0.8718,  -1.7320,  -1.0203,  -1.5445,  -1.3520],
        [ -3.0549,  -1.4118,  -4.0597,  -1.1704,  -1.7968,  -1.2509],
        [ -6.8003,  -1.3953,  -2.4990,  -0.9068,  -1.8186,  -0.8803],
        [ -2.9895,  -1.4976,  -2.0928,  -0.7551,  -1.4813,  -1.2700],
        [ -5.9215,  -1.4736,  -1.5158,  -0.8531,  -2.5731,  -1.2827],
        [ -7.9745,  -1.3022,  -2.7585,  -1.0109,  -2.0905,  -0.7710],
        [ -3.2919,  -4.5109,  -2.7728,  -0.8740,  -4.6371,  -1.4330]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-17.8555, -11.0138, -22.1074,  -9.9181, -26.1786, -16.4277,  -7.3998,
        -11.8520, -13.4719, -11.8961, -11.6313, -18.6682, -24.3452,  -8.6175,
        -12.7445, -14.3004, -10.0863, -13.6198, -15.9076, -17.5197],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.8406e+00, -2.3102e+00, -1.7565e+00, -1.5491e+00, -2.6012e+00,
         -7.8544e-02],
        [-2.5505e-01, -2.8389e+00, -6.1586e-03, -3.6742e+00, -1.7833e-01,
         -1.1566e-01],
        [-3.2164e+00, -2.5783e-03, -3.4579e+00, -2.5927e-01, -3.1144e+00,
         -4.1599e+00],
        [-1.4943e+00, -3.2663e-05, -5.8128e-03, -7.7257e-01, -1.8081e+00,
         -1.1281e-01],
        [-4.6262e+00, -8.4532e-01, -4.4985e+00, -4.8766e+00, -3.1286e+00,
         -1.9571e+00],
        [-3.0885e+00, -2.7896e+00, -2.1975e+00, -1.0831e-01, -2.2441e+00,
         -2.9295e+00],
        [-2.3498e-01, -2.3681e-01, -1.6576e+00, -6.1693e-01, -1.3110e-02,
         -3.4823e-01],
        [-2.4082e+00, -3.2216e-02, -2.3866e+00, -6.1337e-02, -1.1138e+00,
         -2.2424e-02],
        [-3.0913e+00, -1.8810e-01, -1.8797e+00, -5.4574e-04, -1.5234e+00,
         -2.4791e+00],
        [-2.3595e+00, -1.4784e-01, -5.0627e+00, -1.6123e-02, -7.4215e-01,
         -9.3505e-01],
        [-3.2336e+00, -6.7420e-02, -6.9944e-01, -8.4319e-03, -1.9176e+00,
         -2.3167e+00],
        [-3.8049e+00, -2.5161e+00, -3.9038e+00, -1.3753e-02, -2.2901e+00,
         -1.4903e-01],
        [-1.5551e+00, -5.7970e+00, -3.1830e+00, -7.9112e-01, -2.6412e+00,
         -1.8576e-03],
        [-8.4479e-01, -1.6105e-03, -1.5718e+00, -7.4126e-01, -7.9370e-01,
         -3.0683e-01],
        [-1.5815e+00, -3.6043e-01, -4.8421e+00, -1.0586e-01, -1.1129e+00,
         -1.3350e-01],
        [-4.2083e+00, -2.9075e-01, -2.6464e+00, -1.4450e-01, -7.1815e-01,
         -2.3324e-03],
        [-2.5169e+00, -6.9919e-01, -1.9549e+00, -7.0955e-03, -1.0976e+00,
         -9.8650e-01],
        [-3.9319e+00, -6.0521e-01, -8.9839e-01, -1.9075e-01, -1.9963e+00,
         -8.6080e-01],
        [-3.8035e+00, -1.1193e+00, -2.8973e+00, -5.1236e-04, -1.7324e+00,
         -1.2412e-01],
        [-2.7798e+00, -2.9168e+00, -3.6913e+00, -3.3705e-01, -2.5444e+00,
         -2.9909e-01]], device='cuda:0')
tensor([-11.1362,  -7.0683, -14.2105,  -4.1937, -19.9323, -13.3575,  -3.1077,
         -6.0245,  -9.1622,  -9.2634,  -8.2431, -12.6778, -13.9694,  -4.2600,
         -8.1363,  -8.0104,  -7.2622,  -8.4833,  -9.6772, -12.5685],
       device='cuda:0')
