tensor([[ 2.9350,  0.2918, -0.4897,  0.3836, -2.4002, -1.7767],
        [-3.0132, -1.3639, -1.5633,  0.4019,  1.8576,  0.1175],
        [-1.6710, -2.6417, -1.1535, -1.2443,  1.8596,  0.9802],
        [ 0.4087,  1.6377, -2.4196, -1.0887, -0.0451, -0.3500],
        [ 2.5069, -0.3497,  2.3280,  0.4492, -0.8279,  1.1063],
        [ 2.6649,  0.4352,  1.9089, -0.4538, -1.3340,  0.0047],
        [ 3.3994,  0.3506,  2.1801, -0.3085, -1.5252, -1.3828],
        [ 2.4163,  0.2993, -1.0213, -0.1624, -1.9537, -1.4897],
        [-2.2224, -1.4381, -2.1006,  0.1982,  2.3182,  0.2727],
        [-2.0120, -2.3852, -2.2080, -1.0960,  2.9265,  1.1258],
        [ 1.4842,  0.7582, -0.7773, -0.2215,  0.9467, -0.1965],
        [ 2.0057, -0.1358,  2.7503,  0.1871, -1.7959, -0.3964],
        [ 3.0931, -0.1314,  2.1404,  0.5124, -1.5250, -0.6298],
        [ 2.1118, -0.1193,  0.2202,  1.3650, -1.6383, -1.5157],
        [-2.6929, -2.2954, -2.6278,  0.7667,  2.6766,  0.5531],
        [-1.8883, -1.3206, -1.9868, -0.3382,  2.0445,  0.3493],
        [ 0.7095, -0.0698, -1.6586, -0.5196,  0.3009,  0.2629],
        [ 2.6481,  0.3333,  2.7039,  0.1053, -1.4299, -0.8059],
        [ 2.5768,  0.1283,  1.6876,  1.0990, -0.4195, -0.5691],
        [ 1.8568, -0.6955, -1.6327,  1.9171, -2.0449, -1.4038]],
       device='cuda:0')
tensor([ 2.9350,  0.2918, -0.4897,  0.3836, -2.4002, -1.7767], device='cuda:0')
tensor([[ -6.0211,  -0.8741,  -1.0836,  -1.2289,  -5.3713,  -2.3073],
        [ -6.1949,  -0.7054,  -1.0062,  -1.1646,  -5.6284,  -2.0063],
        [ -5.0280,  -0.6869,  -2.1974,  -1.5506,  -9.1591,  -1.4492],
        [ -4.8384,  -0.7307,  -1.5476,  -1.3975,  -5.8561,  -1.5763],
        [ -5.6877,  -0.9106,  -1.0718,  -1.2402,  -6.5036,  -2.0966],
        [ -5.9566,  -0.9694,  -1.2290,  -1.2309,  -7.0632,  -1.6844],
        [ -6.3677,  -0.8677,  -1.0660,  -1.3259,  -5.6004,  -2.0936],
        [ -5.5144,  -0.9024,  -1.1123,  -1.1141,  -6.1081,  -2.2252],
        [ -6.8531,  -0.6747,  -1.1413,  -1.2216,  -6.5398,  -1.7731],
        [ -4.8118,  -0.7195,  -2.0044,  -1.4820,  -6.4995,  -1.4993],
        [ -4.7832,  -0.6527,  -1.5814,  -1.2760,  -6.0429,  -1.5889],
        [ -5.2312,  -0.9291,  -1.0555,  -1.0311,  -6.0675,  -2.0164],
        [ -5.8626,  -1.0088,  -1.0508,  -1.2848,  -6.3818,  -1.6904],
        [ -6.5123,  -0.9299,  -1.0030,  -1.3000,  -5.4221,  -2.0456],
        [ -6.0655,  -0.8252,  -1.0304,  -1.0880,  -6.9881,  -2.0396],
        [ -5.1835,  -0.5909,  -2.0676,  -1.4498, -12.7833,  -1.4128],
        [ -4.6665,  -0.6572,  -1.6440,  -1.2675,  -5.8114,  -1.5871],
        [ -5.7181,  -0.9171,  -1.0506,  -1.0854,  -5.9419,  -1.9775],
        [ -6.0926,  -1.0231,  -1.0488,  -1.3382,  -5.9690,  -1.8156],
        [ -5.6796,  -0.9371,  -0.9807,  -1.1833,  -6.0149,  -2.0213]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-16.8863, -16.7059, -20.0713, -15.9466, -17.5106, -18.1335, -17.3214,
        -16.9764, -18.2036, -17.0166, -15.9251, -16.3309, -17.2792, -17.2129,
        -18.0369, -23.4878, -15.6337, -16.6906, -17.2873, -16.8170],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[ 2.9350,  0.2918, -0.4897,  0.3836, -2.4002, -1.7767],
        [-3.0132, -1.3639, -1.5633,  0.4019,  1.8576,  0.1175]],
       device='cuda:0')
