tensor([[ 3.6046e+00,  1.2190e+00, -1.2866e+00,  5.7429e-01, -2.1598e+00,
         -1.6126e+00],
        [-4.1878e+00, -9.6545e-01, -1.6600e+00,  4.7929e-01,  2.6955e+00,
          2.0181e+00],
        [-2.8121e+00, -2.8722e+00, -1.6128e+00, -1.2074e+00,  1.8425e+00,
          1.4802e+00],
        [ 1.9531e+00,  9.2122e-01, -4.8800e-01,  5.7239e-01,  1.2107e+00,
          1.2528e+00],
        [ 2.2102e+00, -6.9767e-01,  2.0063e+00, -4.6251e-02, -1.2286e+00,
         -1.1705e+00],
        [ 1.3050e+00,  3.0710e-01,  2.0148e+00, -5.3435e-01, -1.4927e+00,
         -1.0529e+00],
        [ 2.3298e+00,  2.4714e-01,  9.1082e-01,  5.2415e-01, -1.7508e+00,
         -1.7501e+00],
        [ 2.0022e+00,  5.2038e-01, -1.1329e+00,  2.8990e-01, -2.2111e+00,
         -3.7945e-01],
        [-2.6225e+00, -2.3827e+00, -2.1624e+00,  1.4399e-01,  1.5553e+00,
          1.0705e+00],
        [-3.0957e+00, -3.3503e+00, -2.2644e+00, -7.2087e-01,  1.8929e+00,
          2.1273e+00],
        [ 1.1477e+00, -3.7663e-02, -8.5491e-01,  1.8003e-01,  1.3143e+00,
         -3.1798e-01],
        [ 2.4680e+00, -2.6932e-01,  2.2278e+00,  6.5248e-01, -5.3225e-01,
          9.3230e-01],
        [ 1.9146e+00, -3.8687e-01,  2.2509e+00, -4.6245e-01, -1.7544e+00,
          4.4629e-01],
        [ 2.6945e+00,  1.1209e+00,  3.3958e+00, -2.1423e-01, -1.4755e+00,
         -5.5258e-01],
        [ 2.2644e+00,  8.7243e-01, -7.7537e-01, -1.3940e-05, -1.3505e+00,
         -5.1227e-01],
        [-3.5341e+00, -1.9181e+00, -2.2729e+00,  1.4508e-02,  2.0904e+00,
          6.6582e-01],
        [-2.7847e+00, -2.5998e+00, -2.1508e+00, -7.2106e-01,  2.1997e+00,
          1.7041e+00],
        [ 1.9262e+00,  3.9249e-02, -7.7798e-01, -2.2438e-02,  1.1500e+00,
         -3.5026e-01],
        [ 2.4109e+00, -2.2505e-01,  2.6512e+00, -1.5815e-01, -1.0422e+00,
         -4.9462e-01],
        [ 2.6354e+00, -4.5863e-01,  3.2250e+00,  6.2449e-01, -7.7366e-01,
         -9.5432e-01]], device='cuda:0')
tensor([ 3.6046,  1.2190, -1.2866,  0.5743, -2.1598, -1.6126], device='cuda:0')
tensor([[ -3.0602,  -1.5580,  -2.2190,  -1.3443,  -5.9504,  -4.0771],
        [ -3.5716,  -1.4371,  -2.0460,  -0.9506,  -3.2620,  -2.0790],
        [ -2.9097,  -2.3226,  -2.7811,  -1.5052, -10.4608,  -3.6685],
        [ -3.5511,  -2.0786,  -3.1180,  -1.1408,  -7.7321,  -3.9459],
        [ -3.6227,  -1.3646,  -2.0543,  -1.1212,  -3.7186,  -2.3821],
        [ -4.1018,  -1.5257,  -1.9575,  -1.4120,  -7.1950,  -3.2365],
        [ -4.4620,  -1.4515,  -1.8506,  -1.4030,  -5.0658,  -3.0185],
        [ -5.2485,  -1.5753,  -1.9943,  -0.9668,  -3.9227,  -2.2725],
        [ -3.8439,  -1.7297,  -1.9991,  -1.0194,  -4.6167,  -2.3060],
        [ -3.0845,  -2.2067,  -3.3924,  -1.3474,  -9.6347,  -3.5836],
        [ -3.5213,  -1.8914,  -2.7315,  -1.1934,  -6.1471,  -3.8281],
        [ -4.3340,  -1.4421,  -2.1462,  -1.2673,  -4.0709,  -2.7331],
        [ -4.5152,  -1.4197,  -2.0295,  -1.2764,  -6.0244,  -3.1861],
        [ -4.3218,  -1.6413,  -1.9702,  -1.3421,  -5.2884,  -3.3141],
        [ -4.4769,  -1.5208,  -1.8653,  -1.1920,  -3.5987,  -2.3225],
        [ -4.6912,  -1.4044,  -2.2130,  -0.9696,  -4.6412,  -2.8677],
        [ -3.1317,  -2.3898,  -3.2545,  -1.3320, -10.3278,  -3.7891],
        [ -3.3748,  -1.4877,  -2.1719,  -1.0512,  -4.0361,  -2.4606],
        [ -4.3639,  -1.3994,  -2.2609,  -1.2186,  -4.1258,  -3.0945],
        [ -4.4901,  -1.5078,  -1.9535,  -1.3870,  -5.2937,  -3.0614]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-18.2089, -13.3464, -23.6477, -21.5663, -14.2635, -19.4285, -17.2514,
        -15.9800, -15.5148, -23.2493, -19.3129, -15.9937, -18.4513, -17.8779,
        -14.9761, -16.7871, -24.2249, -14.5824, -16.4631, -17.6934],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[ 3.6046,  1.2190, -1.2866,  0.5743, -2.1598, -1.6126],
        [-4.1878, -0.9654, -1.6600,  0.4793,  2.6955,  2.0181]],
       device='cuda:0')
