tensor([[ -2.4597,  -1.6178,  -3.9700,  -1.3372,  -1.6455,  -1.1237],
        [ -1.8722,  -1.4484,  -2.0659,  -1.2477,  -1.8507,  -1.0839],
        [ -4.2646,  -1.4158,  -1.6904,  -1.9656,  -1.3273,  -0.9299],
        [ -5.3213,  -2.8906,  -2.3180,  -1.5782,  -2.2185,  -2.6494],
        [ -3.2405,  -2.2202,  -2.3961,  -0.9334,  -2.0114,  -1.6657],
        [ -1.1006,  -1.1800,  -2.4623,  -1.3414,  -1.3604,  -1.2864],
        [ -4.9140,  -1.5906,  -2.8162,  -1.6273,  -1.3769,  -1.0936],
        [ -4.1031,  -1.3548,  -1.5228,  -1.2456,  -1.2944,  -1.6405],
        [ -1.7707,  -1.5075,  -1.6689,  -1.4369,  -1.1861,  -1.2585],
        [ -1.2170,  -1.6831,  -2.3665,  -3.0256,  -1.4406,  -1.6085],
        [ -1.6497,  -2.6407,  -1.6919,  -2.0250,  -1.5510,  -2.4814],
        [ -2.1075,  -2.8196,  -2.9065,  -1.3181,  -4.6345,  -3.3023],
        [ -3.6011,  -3.6985,  -1.8711,  -1.4133,  -2.5980,  -2.1290],
        [ -1.4900,  -3.3313,  -2.3736,  -1.9453,  -2.8966,  -3.0969],
        [ -1.5740,  -1.2650,  -1.1866,  -1.6129,  -1.7450,  -1.7413],
        [ -1.5058,  -3.0583,  -1.2232,  -1.3027,  -2.2092,  -2.0936],
        [ -2.8074,  -1.5717,  -3.6644,  -1.3142,  -2.1337,  -2.1703],
        [ -1.5059,  -1.3349,  -1.3226,  -1.2589,  -1.2879,  -1.2203],
        [-12.8319,  -2.8549,  -3.7267,  -1.4709,  -5.5032,  -4.4106],
        [ -1.8061,  -4.7739,  -1.3310,  -0.9973,  -2.4576,  -1.8223]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.1539,  -9.5687, -11.5936, -16.9760, -12.4674,  -8.7312, -13.4188,
        -11.1612,  -8.8286, -11.3413, -12.0398, -17.0885, -15.3111, -15.1338,
         -9.1248, -11.3928, -13.6617,  -7.9306, -30.7981, -13.1883],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.2659e+00, -5.1862e-01, -4.9206e+00, -2.7946e-01, -1.7645e+00,
         -2.9556e-01],
        [-1.7060e+00, -1.2515e+00, -2.1276e+00, -3.0141e-01, -2.3321e+00,
         -5.7693e-02],
        [-4.0418e+00, -1.3062e-01, -1.6235e+00, -1.5464e+00, -5.5095e-02,
         -8.0161e-02],
        [-3.9126e+00, -4.1424e+00, -3.9804e+00, -1.5923e+00, -2.8014e+00,
         -4.2992e+00],
        [-1.8837e+00, -1.7000e+00, -2.0522e+00, -2.6227e-02, -2.3127e+00,
         -1.0921e+00],
        [-7.5617e-01, -2.3414e-01, -2.6823e+00, -3.2384e-01, -3.0539e-01,
         -5.3168e-01],
        [-4.1667e+00, -5.0754e-02, -4.2529e+00, -9.5814e-01, -1.5426e-02,
         -3.5170e-01],
        [-2.9169e+00, -4.8717e-03, -8.8725e-01, -1.6556e-01, -9.2554e-04,
         -9.2446e-01],
        [-7.3798e-01, -1.2115e+00, -1.5051e+00, -4.3862e-01, -1.7248e-01,
         -7.0853e-01],
        [-9.7535e-01, -1.7174e+00, -2.6767e+00, -2.5899e+00, -9.0398e-01,
         -9.7004e-01],
        [-1.6285e+00, -2.7295e+00, -1.3901e+00, -2.1235e+00, -9.5976e-01,
         -2.3191e+00],
        [-2.5096e+00, -2.0825e+00, -2.9284e+00, -4.6404e-01, -5.0431e+00,
         -3.8714e+00],
        [-2.6642e+00, -3.2288e+00, -1.7909e+00, -4.8038e-01, -3.6786e+00,
         -2.0478e+00],
        [-1.7150e-02, -3.7040e+00, -3.3453e+00, -7.2994e-01, -4.0194e+00,
         -4.3377e+00],
        [-1.9382e+00, -1.6794e-01, -2.0158e-01, -1.0571e+00, -1.6686e+00,
         -1.5302e+00],
        [-1.3194e+00, -3.7663e+00, -3.3287e-01, -3.9688e-01, -2.7122e+00,
         -2.3345e+00],
        [-3.0132e+00, -3.1849e-01, -5.1386e+00, -7.7119e-02, -3.2160e+00,
         -1.2932e+00],
        [-1.1056e+00, -2.1290e-01, -4.4706e-02, -4.0669e-01, -1.0176e-01,
         -3.1279e-03],
        [-8.3351e+00, -3.3846e+00, -5.2579e+00, -1.2925e+00, -7.0026e+00,
         -5.4613e+00],
        [-2.2104e-01, -4.2259e+00, -3.8223e-01, -1.7672e-02, -3.5238e+00,
         -1.5670e+00]], device='cuda:0')
tensor([-10.0445,  -7.7763,  -7.4775, -20.7282,  -9.0669,  -4.8335,  -9.7956,
         -4.9000,  -4.7742,  -9.8333, -11.1504, -16.8991, -13.8905, -16.1534,
         -6.5637, -10.8621, -13.0566,  -1.8748, -30.7340,  -9.9376],
       device='cuda:0')
tensor([-2.1094, -1.7923, -4.1160,  3.7521, -3.4004, -3.8978, -3.6232, -6.2612,
        -4.0544, -1.5079, -0.8893, -0.1894, -1.4205,  1.0196, -2.5611, -0.5306,
        -0.6052, -6.0558, -0.0641, -3.2507], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.9775, -2.7636, -1.7617, -1.1473, -2.4314, -1.6656],
        [-4.5098, -4.4984, -1.9579, -1.3871, -2.6814, -1.2231],
        [-1.1858, -1.8031, -4.2461, -1.4149, -1.3847, -2.0462],
        [-3.8172, -1.5777, -1.9733, -1.2474, -1.3933, -0.9303],
        [-2.5001, -1.1218, -3.5174, -1.2157, -1.3336, -1.3868],
        [-3.3823, -1.6492, -1.3413, -1.6710, -1.5513, -2.1204],
        [-4.2442, -2.3002, -1.7763, -1.2864, -1.8275, -1.4275],
        [-3.8674, -1.8623, -1.9706, -1.0252, -1.9966, -1.4522],
        [-1.1879, -1.5139, -1.6595, -0.9699, -1.4077, -1.3799],
        [-2.5297, -1.2300, -3.2468, -1.3562, -1.7497, -1.4223],
        [-5.5406, -1.4444, -2.6023, -1.2201, -1.5535, -1.0979],
        [-1.4094, -1.1294, -1.5783, -2.6572, -2.1863, -1.3796],
        [-6.8373, -2.1414, -1.8131, -2.4145, -2.3793, -1.6940],
        [-3.6121, -3.5959, -2.2875, -1.0907, -1.7438, -1.4429],
        [-1.3271, -1.1481, -3.3065, -1.0260, -1.6279, -1.4861],
        [-1.1944, -1.4170, -2.2594, -1.4470, -1.4799, -0.9627],
        [-2.2709, -1.6033, -1.7654, -1.2702, -1.7322, -1.3668],
        [-3.3619, -1.1510, -2.5649, -1.5370, -1.6832, -1.7365],
        [-1.7441, -1.3901, -1.0874, -1.1853, -1.6684, -3.7834],
        [-2.8824, -2.2666, -1.9527, -1.1624, -1.6776, -1.3303]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-14.7471, -16.2575, -12.0808, -10.9391, -11.0755, -11.7154, -12.8621,
        -12.1744,  -8.1188, -11.5348, -13.4588, -10.3403, -17.2796, -13.7728,
         -9.9217,  -8.7603, -10.0088, -12.0344, -10.8586, -11.2719],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.5284e+00, -3.0195e+00, -1.5871e+00, -3.7404e-01, -3.3391e+00,
         -1.1646e+00],
        [-2.9179e+00, -4.2065e+00, -1.7876e+00, -4.7216e-01, -3.1061e+00,
         -7.0970e-02],
        [-7.3580e-01, -1.9044e+00, -4.8416e+00, -6.3814e-01, -9.4188e-02,
         -2.0468e+00],
        [-3.8405e+00, -3.1144e-01, -1.9277e+00, -2.3246e-05, -6.6680e-01,
         -3.4928e-04],
        [-2.0469e+00, -2.8296e-02, -4.2670e+00, -2.4110e-01, -8.5280e-01,
         -5.2607e-01],
        [-3.2909e+00, -6.8243e-01, -8.2720e-01, -1.0886e+00, -1.3250e+00,
         -1.9730e+00],
        [-3.0997e+00, -2.0874e+00, -1.6142e+00, -6.0816e-01, -1.6239e+00,
         -3.4798e-01],
        [-2.2386e+00, -1.1055e+00, -1.8080e+00, -8.9480e-03, -2.0112e+00,
         -6.1805e-01],
        [-5.7273e-01, -1.2589e+00, -1.3175e+00, -1.0276e-04, -2.5217e-01,
         -7.0473e-01],
        [-2.6871e+00, -5.3250e-01, -4.3364e+00, -1.7936e-01, -1.7999e+00,
         -8.6587e-01],
        [-4.0535e+00, -7.0322e-03, -3.2906e+00, -1.8834e-01, -1.7203e+00,
         -7.1821e-02],
        [-9.2943e-01, -2.7075e-02, -1.3845e+00, -2.6698e+00, -2.5144e+00,
         -1.2884e-02],
        [-4.5352e+00, -1.7366e+00, -1.8532e+00, -2.8464e+00, -3.2511e+00,
         -1.3941e+00],
        [-2.6803e+00, -3.8806e+00, -2.5991e+00, -1.6026e-03, -1.1934e+00,
         -6.7317e-01],
        [-1.3150e+00, -2.8450e-01, -3.6929e+00, -1.1691e-01, -1.3929e+00,
         -1.0653e+00],
        [-8.3015e-01, -9.4171e-03, -2.7036e+00, -4.6930e-01, -9.2874e-01,
         -1.3945e-01],
        [-1.9500e+00, -4.4741e-01, -1.2651e+00, -1.1522e-01, -2.0937e+00,
         -7.2515e-02],
        [-3.2501e+00, -5.2126e-02, -2.4052e+00, -8.8739e-01, -1.5933e+00,
         -1.8111e+00],
        [-2.2135e+00, -2.8016e-01, -1.2136e-01, -1.8948e-01, -1.2471e+00,
         -2.6635e+00],
        [-1.6955e+00, -2.1484e+00, -2.3087e+00, -3.8985e-01, -1.1510e+00,
         -4.8643e-02]], device='cuda:0')
tensor([-13.0128, -12.5612, -10.2609,  -6.7469,  -7.9622,  -9.1871,  -9.3813,
         -7.7903,  -4.1061, -10.4010,  -9.3316,  -7.5381, -15.6168, -11.0282,
         -7.8676,  -5.0807,  -5.9439,  -9.9992,  -6.7150,  -7.7420],
       device='cuda:0')
tensor([-1.7343, -3.6963, -1.8200, -4.1922, -3.1133, -2.5283, -3.4808, -4.3841,
        -4.0127, -1.1337, -4.1273, -2.8022, -1.6628, -2.7447, -2.0542, -3.6796,
        -4.0649, -2.0352, -4.1435, -3.5298], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.1340, -1.1962, -1.3825, -1.2029, -1.7006, -1.2544],
        [-5.1697, -3.2299, -2.2605, -1.0591, -1.7384, -1.4539],
        [-4.1717, -3.9016, -1.7498, -1.0521, -2.6530, -2.0812],
        [-1.2278, -1.2421, -1.2023, -1.0782, -1.4353, -1.1497],
        [-2.8972, -1.4047, -3.2445, -1.3788, -1.2696, -0.9973],
        [-5.3724, -1.4429, -2.4971, -1.1356, -1.3902, -1.2718],
        [-6.0055, -1.4121, -1.4906, -1.1995, -2.3619, -1.4131],
        [-2.2116, -1.2382, -1.1267, -1.2476, -1.8290, -1.8460],
        [-5.7411, -4.3587, -2.2663, -1.0877, -1.9123, -2.0647],
        [-4.4133, -3.4392, -2.2760, -1.4130, -1.6816, -1.6664],
        [-2.1247, -1.7318, -2.1579, -1.0888, -1.4720, -1.2868],
        [-1.8773, -1.8163, -2.4563, -1.4994, -1.8663, -0.8725],
        [-3.3314, -1.2391, -2.3465, -1.1629, -1.5065, -1.8107],
        [-7.0466, -1.5799, -2.6234, -1.1984, -1.4511, -2.4371],
        [-1.7418, -1.2565, -1.6423, -1.2650, -2.3947, -1.7946],
        [-4.1645, -1.9734, -2.1670, -1.1479, -2.4221, -1.4480],
        [-3.3993, -4.2806, -2.2318, -1.7594, -2.2253, -1.9750],
        [-1.7829, -1.2898, -1.3389, -1.1057, -1.4719, -1.2691],
        [-4.9690, -1.5844, -1.9456, -1.4863, -1.9918, -1.9499],
        [-4.4217, -1.3012, -1.9163, -1.6376, -2.2397, -1.7456]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -7.8706, -14.9115, -15.6092,  -7.3355, -11.1921, -13.1101, -13.8828,
         -9.4991, -17.4307, -14.8895,  -9.8620, -10.3881, -11.3972, -16.3364,
        -10.0949, -13.3229, -15.8713,  -8.2583, -13.9270, -13.2621],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-9.0179e-01, -2.9217e-02, -8.6044e-01, -5.7985e-02, -1.3774e+00,
         -4.2215e-01],
        [-3.5405e+00, -4.1786e+00, -3.2297e+00, -4.1060e-02, -1.0950e+00,
         -3.3738e-01],
        [-3.7133e+00, -4.0961e+00, -1.1496e+00, -5.3693e-02, -3.8195e+00,
         -2.3792e+00],
        [-8.4990e-01, -1.5912e-01, -2.1090e-02, -7.6061e-03, -6.6128e-01,
         -1.0591e-02],
        [-3.6232e+00, -4.5910e-03, -4.4418e+00, -2.9630e-01, -1.2690e-01,
         -9.3549e-02],
        [-4.2129e+00, -1.1254e-01, -3.2919e+00, -1.0354e-01, -1.1701e+00,
         -6.5993e-01],
        [-5.1719e+00, -1.4059e+00, -5.3706e-01, -1.1789e-01, -3.1418e+00,
         -7.0370e-01],
        [-2.5785e+00, -2.4758e-02, -1.6012e-01, -1.4523e-01, -1.8641e+00,
         -1.3061e+00],
        [-4.5858e+00, -4.7221e+00, -2.7479e+00, -8.1043e-02, -1.9016e+00,
         -2.2810e+00],
        [-4.1731e+00, -3.7225e+00, -2.4916e+00, -2.8999e-01, -1.0914e+00,
         -1.4482e+00],
        [-3.0110e+00, -1.9963e+00, -2.2910e+00, -2.4344e-02, -4.7853e-01,
         -2.1221e-01],
        [-2.2044e+00, -7.9213e-01, -3.3583e+00, -4.5636e-01, -1.9772e+00,
         -7.1511e-02],
        [-3.1871e+00, -1.6236e-02, -2.7234e+00, -6.3671e-02, -1.2200e+00,
         -2.1736e+00],
        [-5.7600e+00, -2.0292e+00, -2.3966e+00, -4.8548e-02, -1.0274e+00,
         -2.3484e+00],
        [-1.8783e+00, -3.7241e-02, -1.5847e+00, -6.4169e-03, -2.8286e+00,
         -1.1866e+00],
        [-2.9074e+00, -1.1576e+00, -3.0640e+00, -1.9906e-01, -3.5353e+00,
         -2.6265e-02],
        [-2.8637e+00, -4.3442e+00, -2.1576e+00, -9.8484e-01, -2.9362e+00,
         -2.1058e+00],
        [-2.4583e+00, -4.3482e-01, -5.2109e-01, -4.9993e-03, -2.2903e-01,
         -3.4333e-02],
        [-5.3800e+00, -3.5791e-01, -1.8879e+00, -5.2536e-01, -2.2360e+00,
         -1.1424e+00],
        [-3.5518e+00, -3.9132e-01, -1.7410e+00, -1.2344e+00, -3.3045e+00,
         -1.8778e+00]], device='cuda:0')
tensor([ -3.6490, -12.4222, -15.2114,  -1.7096,  -8.5864,  -9.5509, -11.0782,
         -6.0789, -16.3194, -13.2168,  -8.0134,  -8.8600,  -9.3840, -13.6101,
         -7.5219, -10.8896, -15.3924,  -3.6825, -11.5296, -12.1009],
       device='cuda:0')
tensor([-4.2216, -2.4893, -0.3978, -5.6259, -2.6058, -3.5591, -2.8046, -3.4202,
        -1.1113, -1.6727, -1.8486, -1.5282, -2.0132, -2.7263, -2.5730, -2.4333,
        -0.4789, -4.5758, -2.3974, -1.1612], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[ -4.8426,  -1.1178,  -3.2777,  -1.2492,  -1.5898,  -1.0972],
        [ -5.3519,  -1.6941,  -4.0462,  -1.2480,  -1.5456,  -1.1403],
        [-15.2857,  -1.1959,  -1.7080,  -1.1894,  -1.6914,  -2.2523],
        [ -1.0711,  -1.2694,  -1.3719,  -1.2781,  -2.3005,  -2.1140],
        [ -6.9439,  -2.0886,  -1.9931,  -1.1026,  -2.0872,  -1.2621],
        [ -2.9504,  -3.2096,  -2.4042,  -1.0400,  -2.9560,  -2.0027],
        [ -1.3379,  -1.1677,  -1.4337,  -1.1280,  -1.6509,  -1.2937],
        [ -5.1105,  -1.4438,  -3.0487,  -1.2526,  -1.5778,  -1.0743],
        [ -2.1878,  -1.4397,  -4.5869,  -1.1804,  -1.4617,  -1.2291],
        [ -6.8503,  -1.1638,  -3.4356,  -1.1694,  -1.8212,  -1.6961],
        [ -2.3423,  -1.1805,  -1.4860,  -1.1597,  -1.8078,  -2.8291],
        [ -4.5301,  -2.1208,  -1.9295,  -1.1560,  -1.9687,  -1.4355],
        [ -3.6744,  -3.8264,  -2.8032,  -1.1724,  -2.5158,  -1.3965],
        [ -1.5060,  -1.8282,  -5.5292,  -1.7085,  -1.4840,  -1.2789],
        [ -1.7932,  -1.4632,  -2.0388,  -1.2884,  -1.2680,  -1.2012],
        [ -5.5635,  -1.3105,  -2.5989,  -1.2500,  -1.6142,  -1.2229],
        [ -5.5480,  -1.0795,  -2.0019,  -1.3746,  -2.0295,  -1.4561],
        [ -1.3068,  -1.3292,  -1.9248,  -1.2283,  -1.5533,  -2.1465],
        [ -5.5726,  -2.4992,  -1.9061,  -1.0727,  -1.9214,  -1.6013],
        [ -4.6941,  -3.2645,  -1.7249,  -1.4716,  -3.1741,  -1.1677]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.1742, -15.0261, -23.3227,  -9.4049, -15.4777, -14.5629,  -8.0119,
        -13.5077, -12.0857, -16.1365, -10.8054, -13.1406, -15.3888, -13.3348,
         -9.0528, -13.5600, -13.4897,  -9.4888, -14.5733, -15.4970],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-5.2430e+00, -9.1898e-04, -4.3145e+00, -6.6348e-02, -1.2837e+00,
         -3.8397e-02],
        [-3.9560e+00, -6.8231e-01, -5.0437e+00, -8.0415e-02, -1.3264e+00,
         -3.0182e-01],
        [-9.4130e+00, -5.6608e-01, -9.5123e-01, -8.2846e-02, -1.6153e+00,
         -2.6176e+00],
        [-6.7660e-01, -3.0002e-02, -9.5603e-01, -2.4716e-01, -2.8626e+00,
         -1.5940e+00],
        [-5.1820e+00, -1.3692e+00, -2.3760e+00, -1.1218e-03, -2.4413e+00,
         -1.0919e-02],
        [-2.1681e+00, -3.0839e+00, -2.6889e+00, -4.2878e-02, -4.3140e+00,
         -2.2478e+00],
        [-1.2932e+00, -4.6571e-02, -8.5634e-01, -1.7193e-01, -1.1164e+00,
         -3.4119e-01],
        [-4.9881e+00, -3.8628e-02, -4.3669e+00, -2.0378e-03, -1.0509e+00,
         -1.1230e-01],
        [-1.7793e+00, -1.6334e-02, -5.7774e+00, -3.7221e-02, -1.0039e+00,
         -5.2583e-02],
        [-5.3335e+00, -4.5451e-01, -3.1992e+00, -4.2939e-04, -2.0974e+00,
         -1.5525e+00],
        [-2.8138e+00, -1.1770e-01, -1.4221e+00, -8.2370e-03, -1.6954e+00,
         -2.0907e+00],
        [-3.2234e+00, -1.4748e+00, -1.9422e+00, -2.8110e-03, -2.0864e+00,
         -1.0814e-01],
        [-3.0314e+00, -3.7003e+00, -3.2294e+00, -2.4570e-01, -3.4365e+00,
         -5.9442e-01],
        [-1.9121e+00, -2.2019e+00, -5.6171e+00, -1.1498e+00, -1.7929e-01,
         -2.1403e-02],
        [-1.8363e+00, -2.6751e-02, -2.2331e+00, -6.0465e-03, -9.4021e-02,
         -3.1083e-01],
        [-4.3109e+00, -3.3617e-05, -3.2732e+00, -3.2712e-01, -1.8611e+00,
         -8.8626e-02],
        [-4.2762e+00, -3.0654e-01, -1.5122e+00, -4.3834e-01, -2.7062e+00,
         -8.9728e-01],
        [-1.3399e+00, -1.4331e-01, -2.1134e+00, -1.4593e-01, -1.0673e+00,
         -1.6273e+00],
        [-3.5440e+00, -2.6941e+00, -1.9194e+00, -1.5999e-01, -1.8970e+00,
         -5.9498e-01],
        [-3.7054e+00, -2.8860e+00, -1.1374e+00, -7.6894e-01, -3.9052e+00,
         -1.0595e-01]], device='cuda:0')
tensor([-10.9468, -11.3907, -15.2460,  -6.3664, -11.3805, -14.5456,  -3.8257,
        -10.5588,  -8.6667, -12.6376,  -8.1479,  -8.8378, -14.2377, -11.0816,
         -4.5070,  -9.8609, -10.1367,  -6.4372, -10.8095, -12.5089],
       device='cuda:0')
tensor([-2.2274, -3.6354, -8.0767, -3.0384, -4.0972, -0.0173, -4.1862, -2.9489,
        -3.4190, -3.4989, -2.6574, -4.3028, -1.1511, -2.2532, -4.5458, -3.6991,
        -3.3530, -3.0516, -3.7639, -2.9880], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-7.1200, -2.6329, -1.8322, -1.0548, -2.1974, -1.2607],
        [-4.5564, -2.4924, -2.4277, -1.0357, -1.9346, -2.0689],
        [-1.2490, -1.3752, -2.0145, -1.4463, -1.4250, -1.2554],
        [-5.4411, -1.3365, -2.0282, -1.4654, -1.3011, -2.8621],
        [-2.6868, -1.4824, -3.3723, -1.3114, -1.5616, -1.0920],
        [-1.3879, -1.6441, -1.1445, -2.0191, -1.7864, -1.6674],
        [-2.6936, -2.2917, -1.8751, -1.2921, -2.0985, -1.4710],
        [-2.2833, -3.9221, -2.3794, -1.4247, -2.2733, -1.3613],
        [-1.2471, -1.2569, -1.3951, -1.1103, -1.3829, -1.1585],
        [-1.9502, -1.4454, -2.5341, -1.4943, -1.9262, -2.8274],
        [-5.0583, -1.3947, -1.4501, -1.6563, -2.3500, -1.2225],
        [-3.6325, -1.9307, -1.3209, -2.3281, -2.0426, -1.5554],
        [-4.9836, -1.8360, -1.9932, -3.1822, -2.6645, -1.9277],
        [-1.7141, -3.0644, -2.3728, -1.4981, -1.8253, -1.4822],
        [-1.5757, -2.5081, -3.4750, -1.3410, -1.3479, -1.2811],
        [-1.9131, -1.4437, -2.3676, -1.4531, -1.4910, -2.0183],
        [-3.0096, -1.2117, -2.2085, -1.2032, -1.4256, -1.3980],
        [-3.1597, -1.1424, -1.9457, -1.1753, -1.7802, -1.6208],
        [-2.6069, -1.2234, -2.0000, -1.1406, -1.9733, -2.4331],
        [-2.9891, -2.0784, -2.2023, -1.1131, -1.7411, -1.7640]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-16.0980, -14.5157,  -8.7653, -14.4344, -11.5066,  -9.6492, -11.7221,
        -13.6440,  -7.5507, -12.1778, -13.1319, -12.8102, -16.5871, -11.9569,
        -11.5288, -10.6869, -10.4567, -10.8240, -11.3773, -11.8879],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-5.3816e+00, -2.5401e+00, -1.8118e+00, -8.2451e-02, -2.6042e+00,
         -1.8639e-01],
        [-3.7131e+00, -2.3513e+00, -3.2076e+00, -3.7203e-02, -1.9060e+00,
         -2.6006e+00],
        [-2.1722e-02, -8.5182e-01, -2.1012e+00, -6.1660e-01, -6.4099e-03,
         -2.2729e-01],
        [-5.0933e+00, -4.1517e-01, -1.9318e+00, -4.9579e-01, -1.7576e-01,
         -2.0303e+00],
        [-2.5713e+00, -8.4721e-02, -5.2147e+00, -3.5214e-01, -1.5714e+00,
         -3.0579e-01],
        [-9.1976e-01, -6.4842e-01, -2.6547e-01, -1.8797e+00, -2.0216e+00,
         -1.6603e+00],
        [-1.6610e+00, -2.1993e+00, -2.1177e+00, -8.3561e-01, -2.3277e+00,
         -4.9298e-01],
        [-1.0475e+00, -4.6228e+00, -3.3876e+00, -2.1834e-01, -2.8286e+00,
         -3.9088e-01],
        [-8.7300e-01, -6.0961e-01, -6.1486e-01, -1.2508e-01, -2.5999e-01,
         -1.5979e-03],
        [-2.3437e+00, -4.0656e-02, -3.0690e+00, -5.4899e-01, -2.2082e+00,
         -1.8576e+00],
        [-3.7076e+00, -1.0173e-02, -3.5740e-01, -1.2667e+00, -3.6733e+00,
         -2.5707e-01],
        [-3.0730e+00, -1.3274e+00, -8.7468e-01, -2.5011e+00, -2.6615e+00,
         -1.3758e+00],
        [-3.3717e+00, -8.6238e-01, -2.5362e+00, -3.6113e+00, -3.6020e+00,
         -2.0790e+00],
        [-2.4330e-01, -2.9657e+00, -2.8507e+00, -3.2756e-01, -1.4703e+00,
         -9.4882e-01],
        [-2.0742e+00, -3.5639e+00, -4.3989e+00, -3.8239e-01, -2.4760e-01,
         -2.5729e-01],
        [-2.2668e+00, -1.1728e-02, -2.7569e+00, -4.0490e-01, -7.3156e-01,
         -2.0951e+00],
        [-2.7118e+00, -2.4303e-02, -2.5838e+00, -1.8403e-02, -6.5115e-01,
         -4.3529e-01],
        [-3.4536e+00, -4.7684e-01, -1.5849e+00, -5.9185e-02, -1.7974e+00,
         -1.1674e+00],
        [-3.3116e+00, -1.3914e-02, -2.2978e+00, -1.3393e-03, -2.0625e+00,
         -1.7443e+00],
        [-1.7143e+00, -1.4119e+00, -3.2694e+00, -1.2644e-01, -1.0934e+00,
         -1.3279e+00]], device='cuda:0')
tensor([-12.6066, -13.8158,  -3.8251, -10.1421, -10.1001,  -7.3952,  -9.6343,
        -12.4957,  -2.4841, -10.0681,  -9.2722, -11.8135, -16.0626,  -8.8063,
        -10.9242,  -8.2669,  -6.4247,  -8.5393,  -9.4314,  -8.9433],
       device='cuda:0')
tensor([-3.4914, -0.7000, -4.9402, -4.2923, -1.4065, -2.2540, -2.0878, -1.1483,
        -5.0666, -2.1096, -3.8596, -0.9967, -0.5245, -3.1506, -0.6045, -2.4199,
        -4.0320, -2.2847, -1.9459, -2.9446], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.1960, -1.1976, -1.8513, -1.1098, -1.4074, -1.2964],
        [-2.7752, -1.2128, -2.7024, -1.2434, -1.6829, -0.9211],
        [-1.6509, -1.7530, -2.9387, -1.5455, -1.6110, -1.4929],
        [-1.4991, -1.2177, -1.5332, -1.8767, -1.7898, -2.5100],
        [-5.3662, -3.7884, -2.0180, -1.2602, -2.1170, -1.2103],
        [-4.0827, -2.6554, -1.8983, -1.7988, -1.7848, -1.5237],
        [-1.2530, -1.4261, -1.9938, -1.9764, -1.5730, -1.3922],
        [-2.4133, -1.4264, -2.1961, -1.3457, -1.2905, -0.8439],
        [-1.2601, -1.2857, -2.5553, -1.1005, -1.3689, -1.0817],
        [-1.8788, -1.2088, -1.5018, -1.4925, -1.3395, -1.9164],
        [-2.3585, -1.3660, -1.8204, -1.6256, -1.8438, -2.4798],
        [-3.3448, -3.4337, -1.7771, -1.1063, -2.3490, -1.3561],
        [-3.3485, -2.8392, -2.7270, -1.0191, -2.4324, -1.3523],
        [-1.2013, -1.6240, -4.3792, -1.2152, -1.3910, -1.4532],
        [-2.2995, -1.3701, -1.6770, -1.2690, -2.0254, -0.9955],
        [-2.4511, -1.5514, -1.7618, -2.4639, -1.3215, -1.2529],
        [-2.9917, -1.4650, -2.2656, -2.8679, -1.6361, -2.5082],
        [-5.2743, -2.9331, -1.6591, -1.2168, -1.9475, -1.4837],
        [-4.4254, -3.7290, -2.3258, -2.0572, -2.7096, -1.9092],
        [-1.1643, -1.4115, -1.4434, -1.4770, -1.4777, -1.1550]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.0585, -10.5378, -10.9920, -10.4266, -15.7601, -13.7437,  -9.6145,
         -9.5159,  -8.6522,  -9.3378, -11.4942, -13.3668, -13.7186, -11.2641,
         -9.6365, -10.8025, -13.7345, -14.5145, -17.1560,  -8.1288],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-7.4204e-01, -9.7060e-04, -1.6162e+00, -9.1169e-02, -3.7963e-02,
         -1.5144e-01],
        [-3.1401e+00, -3.6055e-03, -3.1363e+00, -1.9795e-03, -1.4377e+00,
         -5.4559e-03],
        [-1.3603e+00, -6.8572e-01, -3.6408e+00, -9.0699e-01, -1.7431e+00,
         -1.2034e+00],
        [-1.2770e+00, -3.3650e-02, -6.9368e-01, -1.5171e+00, -1.8740e+00,
         -3.0600e+00],
        [-4.1522e+00, -4.2774e+00, -2.3164e+00, -5.1495e-01, -2.2429e+00,
         -2.8242e-01],
        [-2.6875e+00, -3.2284e+00, -1.2649e+00, -6.9845e-01, -1.1747e+00,
         -5.8506e-01],
        [-1.4319e-01, -1.0211e+00, -2.0506e+00, -1.2561e+00, -5.7707e-01,
         -6.1742e-01],
        [-2.8415e+00, -4.5841e-02, -2.3419e+00, -1.3096e-01, -1.4963e-03,
         -1.1379e-03],
        [-8.6956e-01, -4.1131e-03, -3.4396e+00, -1.0732e-02, -1.1157e+00,
         -8.9214e-03],
        [-1.6338e+00, -9.4893e-03, -5.7773e-01, -7.8751e-01, -5.3699e-01,
         -2.1392e+00],
        [-2.9677e+00, -1.9950e-01, -1.9176e+00, -1.2272e+00, -1.8363e+00,
         -2.2716e+00],
        [-2.2907e+00, -4.0814e+00, -1.4663e+00, -7.3779e-04, -3.3394e+00,
         -7.0881e-03],
        [-2.6562e+00, -2.4773e+00, -3.2298e+00, -4.0701e-02, -2.7996e+00,
         -8.0792e-01],
        [-7.1315e-01, -1.6022e+00, -5.0901e+00, -3.1462e-01, -2.7828e-02,
         -7.9919e-01],
        [-2.3998e+00, -5.4881e-01, -9.8285e-01, -6.0081e-04, -2.6099e+00,
         -7.9395e-02],
        [-2.0044e+00, -1.2940e-01, -1.1493e+00, -2.5311e+00, -2.1204e-01,
         -1.3111e-01],
        [-3.0522e+00, -5.5520e-01, -2.4584e+00, -2.8151e+00, -1.4624e+00,
         -3.2613e+00],
        [-3.3168e+00, -3.5055e+00, -7.1787e-01, -5.3572e-01, -1.9956e+00,
         -6.6026e-01],
        [-2.7194e+00, -3.3074e+00, -2.4990e+00, -1.3527e+00, -3.1623e+00,
         -1.9230e+00],
        [-3.0015e-01, -7.7014e-01, -7.9743e-01, -5.0473e-01, -1.9942e-01,
         -2.8732e-02]], device='cuda:0')
tensor([ -2.6398,  -7.7251,  -9.5404,  -8.4555, -13.7863,  -9.6389,  -5.6654,
         -5.3628,  -5.4486,  -5.6847, -10.4198, -11.1857, -12.0116,  -8.5471,
         -6.6213,  -6.1573, -13.6046, -10.7317, -14.9639,  -2.6006],
       device='cuda:0')
tensor([-5.4186, -2.8126, -1.4516, -1.9711, -1.9738, -4.1047, -3.9490, -4.1531,
        -3.2036, -3.6531, -1.0744, -2.1811, -1.7070, -2.7169, -3.0152, -4.6452,
        -0.1300, -3.7828, -2.1922, -5.5282], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.6805, -1.2255, -2.0830, -1.2216, -1.4165, -1.4424],
        [-3.4746, -1.0464, -1.9640, -1.1982, -1.1415, -1.3237],
        [-1.9768, -1.4824, -1.0945, -1.0619, -1.6235, -3.5606],
        [-7.4071, -2.1600, -2.1630, -1.0729, -1.9209, -1.6296],
        [-5.3552, -2.6002, -2.1374, -1.5970, -2.3243, -1.6115],
        [-1.4611, -1.2479, -1.4302, -1.5078, -1.5259, -1.3478],
        [-4.4370, -1.3323, -2.7212, -1.2687, -3.7008, -1.8921],
        [-4.6907, -1.4207, -2.4893, -1.1857, -2.2653, -1.1367],
        [-2.9538, -1.3350, -1.7571, -2.3545, -1.5643, -1.3092],
        [-4.1487, -2.4255, -1.8887, -1.3625, -1.9953, -1.8058],
        [-2.5347, -3.0101, -1.9635, -1.4618, -1.9995, -1.5821],
        [-1.3996, -1.1238, -2.8162, -1.1974, -1.3571, -1.5559],
        [-1.7418, -1.4605, -2.6761, -1.3657, -1.3092, -1.0791],
        [-2.5080, -1.3281, -2.0813, -1.3559, -1.4022, -1.1592],
        [-2.4191, -1.3653, -1.7820, -1.1347, -1.2858, -1.1337],
        [-2.6663, -1.3097, -1.1545, -1.0628, -2.2867, -2.8852],
        [-4.2323, -2.2939, -1.7591, -1.0998, -2.0870, -1.2243],
        [-3.5806, -4.0549, -2.7689, -1.4140, -2.9537, -1.6754],
        [-1.2642, -1.2186, -1.2212, -1.0359, -1.6974, -1.2145],
        [-1.6899, -1.3240, -3.0175, -1.1723, -1.3689, -1.8229]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.0693, -10.1484, -10.7997, -16.3536, -15.6256,  -8.5208, -15.3521,
        -13.1883, -11.2739, -13.6265, -12.5517,  -9.4499,  -9.6325,  -9.8347,
         -9.1206, -11.3651, -12.6964, -16.4476,  -7.6518, -10.3954],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.1742e+00, -3.7116e-03, -2.2379e+00, -3.9970e-02, -3.0899e-01,
         -1.8755e-01],
        [-3.6526e+00, -8.8673e-02, -1.6711e+00, -1.6455e-01, -1.3495e-01,
         -4.4209e-01],
        [-2.6643e+00, -4.4202e-01, -3.3723e-02, -3.5157e-03, -1.1179e+00,
         -2.4245e+00],
        [-5.4989e+00, -1.8333e+00, -2.6965e+00, -1.3242e-01, -1.9190e+00,
         -1.3187e+00],
        [-3.5331e+00, -2.5893e+00, -2.4386e+00, -7.1211e-01, -2.8987e+00,
         -1.1970e+00],
        [-1.7568e+00, -1.2125e-01, -7.5781e-01, -5.7875e-01, -3.3080e-01,
         -5.2030e-01],
        [-4.7372e+00, -6.0456e-02, -3.6992e+00, -4.0756e-02, -4.8085e+00,
         -1.2809e+00],
        [-3.3438e+00, -2.9471e-01, -2.7366e+00, -1.1691e-03, -3.6231e+00,
         -7.4822e-02],
        [-2.5953e+00, -6.4402e-02, -1.9718e+00, -2.2817e+00, -1.4432e+00,
         -1.9551e-01],
        [-3.2357e+00, -2.3936e+00, -1.9276e+00, -9.0373e-01, -2.1112e+00,
         -1.5101e+00],
        [-1.2277e+00, -3.0738e+00, -1.7836e+00, -4.0211e-01, -2.2002e+00,
         -1.1514e+00],
        [-1.6544e+00, -7.4999e-02, -3.0768e+00, -3.2277e-01, -5.1178e-01,
         -1.2307e+00],
        [-1.6466e+00, -1.5630e-01, -3.6201e+00, -2.6920e-01, -3.3599e-01,
         -7.6965e-02],
        [-2.0875e+00, -3.3306e-01, -1.9282e+00, -4.6335e-01, -7.2523e-01,
         -4.5756e-02],
        [-2.1833e+00, -3.2729e-02, -1.1418e+00, -2.2792e-02, -2.0356e-01,
         -5.9987e-02],
        [-2.9539e+00, -6.0475e-01, -2.6240e-01, -4.6740e-02, -2.4588e+00,
         -2.8683e+00],
        [-2.7577e+00, -2.0443e+00, -1.6871e+00, -1.6451e-05, -2.4808e+00,
         -6.5370e-02],
        [-2.8677e+00, -4.3069e+00, -3.1209e+00, -5.1620e-01, -4.4883e+00,
         -1.4045e+00],
        [-9.5824e-01, -2.4659e-02, -3.2334e-01, -1.5823e-01, -1.2150e+00,
         -1.8693e-03],
        [-1.9831e+00, -3.0334e-02, -3.6672e+00, -4.4450e-02, -4.1653e-01,
         -1.1351e+00]], device='cuda:0')
tensor([ -5.9523,  -6.1539,  -6.6860, -13.3989, -13.3689,  -4.0657, -14.6269,
        -10.0742,  -8.5519, -12.0820,  -9.8388,  -6.8715,  -6.1052,  -5.5831,
         -3.6442,  -9.1949,  -9.0353, -16.7045,  -2.6814,  -7.2767],
       device='cuda:0')
tensor([-5.1170, -3.9945, -4.1137, -2.9547, -2.2566, -4.4550, -0.7252, -3.1141,
        -2.7220, -1.5445, -2.7129, -2.5785, -3.5273, -4.2517, -5.4764, -2.1702,
        -3.6611,  0.2568, -4.9704, -3.1188], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.8227, -2.5323, -1.8242, -1.3701, -2.1035, -1.8024],
        [-4.6614, -2.6789, -3.4150, -0.9013, -2.7866, -1.4633],
        [-1.0927, -1.1978, -3.1643, -1.3566, -1.2649, -1.0992],
        [-1.8361, -1.4406, -2.8132, -1.5233, -1.4368, -0.9674],
        [-2.8047, -1.3988, -3.3763, -1.3824, -1.6958, -1.3149],
        [-4.2131, -1.2000, -1.2433, -1.6756, -1.8711, -1.5040],
        [-3.6102, -2.1609, -2.3330, -1.1376, -2.1660, -1.5702],
        [-6.6780, -2.4062, -2.1641, -1.0102, -1.9015, -1.9323],
        [-1.2635, -1.2271, -2.8873, -1.3119, -1.5299, -1.3698],
        [-3.2139, -1.4371, -2.9264, -1.3909, -1.2433, -1.0014],
        [-1.6959, -1.6260, -2.5314, -1.1409, -1.5988, -1.3536],
        [-6.4025, -1.2459, -2.9095, -1.3284, -1.7024, -1.2964],
        [-2.3043, -1.2399, -1.1374, -1.2360, -1.6936, -3.5593],
        [-4.1022, -2.1383, -2.3509, -1.1129, -2.2164, -1.4298],
        [-3.5842, -3.1624, -2.5022, -1.1939, -3.2487, -1.6576],
        [-1.4214, -1.5070, -1.2725, -1.3524, -1.7996, -1.3168],
        [-3.6722, -1.4935, -2.8491, -1.5859, -2.4794, -2.6292],
        [-5.7340, -1.2518, -1.8542, -1.3145, -2.5240, -1.4148],
        [-1.3031, -1.3031, -1.1224, -2.4004, -1.5034, -1.5961],
        [-4.5269, -2.0228, -1.9446, -1.7663, -2.6880, -1.7082]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.4552, -15.9063,  -9.1756, -10.0175, -11.9729, -11.7072, -12.9780,
        -16.0922,  -9.5896, -11.2131,  -9.9466, -14.8850, -11.1706, -13.3504,
        -15.3490,  -8.6697, -14.7093, -14.0932,  -9.2285, -14.6567],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.3216e+00, -2.7547e+00, -1.7070e+00, -1.0662e+00, -2.4572e+00,
         -1.7263e+00],
        [-3.3834e+00, -2.5111e+00, -3.3644e+00, -1.7178e-03, -4.3615e+00,
         -5.4227e-01],
        [-7.4258e-01, -5.3316e-01, -3.5205e+00, -8.0663e-01, -2.2711e-01,
         -4.9198e-04],
        [-1.9224e+00, -7.5070e-02, -4.0092e+00, -5.5945e-01, -6.6443e-01,
         -2.7195e-03],
        [-2.2206e+00, -2.5368e-04, -4.3277e+00, -5.7445e-01, -2.0636e+00,
         -6.8559e-01],
        [-4.0095e+00, -8.8384e-02, -4.3342e-01, -1.1257e+00, -2.0781e+00,
         -1.2058e+00],
        [-2.3335e+00, -1.8213e+00, -3.7377e+00, -3.1295e-01, -2.6759e+00,
         -7.6741e-01],
        [-4.9106e+00, -2.0739e+00, -2.4848e+00, -3.3740e-02, -1.7802e+00,
         -2.3154e+00],
        [-1.3845e-02, -2.7840e-01, -3.4701e+00, -4.3851e-01, -4.4761e-01,
         -5.8283e-01],
        [-3.5406e+00, -9.8737e-02, -3.7592e+00, -2.9092e-01, -6.5625e-04,
         -4.8006e-01],
        [-1.3139e+00, -5.6733e-01, -3.2360e+00, -1.1815e-01, -1.7171e+00,
         -9.7226e-01],
        [-5.2009e+00, -8.3307e-01, -2.8329e+00, -3.2506e-01, -1.6481e+00,
         -4.4584e-05],
        [-2.4169e+00, -1.9890e-02, -1.8632e-01, -6.9934e-02, -1.5189e+00,
         -2.8639e+00],
        [-3.0002e+00, -1.7154e+00, -3.4502e+00, -1.0831e-01, -2.8618e+00,
         -4.5347e-01],
        [-2.6633e+00, -3.0679e+00, -2.8514e+00, -3.0477e-01, -4.6089e+00,
         -1.3532e+00],
        [-1.5319e+00, -1.3227e+00, -3.1864e-01, -2.9823e-01, -1.3982e+00,
         -8.1352e-03],
        [-4.0766e+00, -8.2762e-02, -3.9783e+00, -7.9394e-01, -3.1801e+00,
         -1.8073e+00],
        [-4.1665e+00, -4.3967e-01, -1.4537e+00, -4.4922e-01, -3.9089e+00,
         -9.5125e-01],
        [-8.4066e-01, -3.3051e-03, -3.5023e-01, -2.4528e+00, -1.2493e+00,
         -9.2288e-01],
        [-4.0633e+00, -1.4904e+00, -2.3093e+00, -2.1478e+00, -3.6796e+00,
         -1.3031e+00]], device='cuda:0')
tensor([-13.0331, -14.1644,  -5.8305,  -7.2332,  -9.8722,  -8.9409, -11.6487,
        -13.5986,  -5.2312,  -8.1702,  -7.9247, -10.8401,  -7.0758, -11.5894,
        -14.8494,  -4.8778, -13.9190, -11.3692,  -5.8192, -14.9935],
       device='cuda:0')
tensor([-0.4222, -1.7419, -3.3451, -2.7843, -2.1007, -2.7663, -1.3293, -2.4936,
        -4.3584, -3.0429, -2.0219, -4.0450, -4.0948, -1.7611, -0.4996, -3.7920,
        -0.7903, -2.7240, -3.4093,  0.3368], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.4243, -1.8813, -1.6325, -1.3863, -2.5336, -1.3456],
        [-3.8073, -3.3826, -2.6135, -2.1075, -2.3472, -1.6931],
        [-1.4384, -1.0981, -1.0596, -1.0663, -1.6841, -1.1123],
        [-1.9282, -1.2543, -3.0424, -1.3183, -3.1929, -1.7733],
        [-2.1380, -1.5008, -2.1440, -1.8431, -2.2364, -0.9933],
        [-0.9106, -2.2066, -2.0902, -3.3262, -1.4791, -1.3274],
        [-4.8310, -2.6441, -2.0120, -2.9758, -2.4478, -2.5032],
        [-9.5675, -3.1179, -3.0745, -1.6684, -1.3895, -1.4015],
        [-1.4024, -2.5400, -2.2554, -6.1786, -1.2036, -1.1957],
        [-4.3934, -1.9069, -2.0532, -1.3491, -1.3641, -1.4949],
        [-5.9434, -1.4956, -1.7324, -1.3887, -2.1193, -2.2641],
        [-1.4079, -1.5419, -1.6341, -1.0876, -1.3799, -1.4236],
        [-1.9235, -2.1725, -1.5559, -2.0734, -2.1238, -1.2196],
        [-2.1150, -5.8395, -3.1334, -2.9714, -1.5847, -1.5884],
        [-2.3616, -1.2773, -2.5109, -1.7813, -1.8475, -2.0471],
        [-2.0367, -1.0343, -1.3312, -1.0187, -1.7149, -1.3760],
        [-1.7482, -1.2448, -1.1058, -1.2023, -1.7172, -2.4510],
        [-3.8143, -2.0636, -1.4033, -1.0503, -2.8184, -1.8245],
        [-3.8530, -2.4214, -1.8264, -0.9825, -2.6374, -1.9560],
        [-1.1962, -1.4263, -1.1799, -1.1925, -1.9485, -2.9867]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.2035, -15.9512,  -7.4588, -12.5093, -10.8555, -11.3401, -17.4140,
        -20.2192, -14.7758, -12.5617, -14.9435,  -8.4750, -11.0687, -17.2323,
        -11.8255,  -8.5118,  -9.4693, -12.9744, -13.6768,  -9.9300],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.3890e+00, -9.1047e-01, -1.2702e+00, -6.6603e-01, -3.6133e+00,
         -9.0107e-01],
        [-2.9170e+00, -3.7542e+00, -3.3010e+00, -9.9669e-01, -3.3603e+00,
         -1.3319e+00],
        [-1.6904e+00, -1.3012e-03, -2.2662e-04, -2.9260e-01, -1.5415e+00,
         -8.1335e-03],
        [-2.2848e+00, -1.5436e-01, -3.3766e+00, -2.8290e-01, -4.2127e+00,
         -1.0759e+00],
        [-1.9817e+00, -9.6315e-03, -2.2423e+00, -1.5154e+00, -3.4815e+00,
         -2.9578e-02],
        [-1.8589e-01, -1.8133e+00, -2.3998e+00, -3.7504e+00, -1.1456e+00,
         -4.5638e-01],
        [-2.9407e+00, -3.2436e+00, -2.6932e+00, -3.3890e+00, -3.1465e+00,
         -3.2507e+00],
        [-4.6779e+00, -2.4052e+00, -3.0151e+00, -7.2110e-01, -1.9550e-01,
         -6.0913e-02],
        [-1.0605e-01, -3.5261e+00, -2.2438e+00, -4.5397e+00, -1.9131e-01,
         -6.4559e-01],
        [-3.9390e+00, -2.8811e+00, -2.0020e+00, -7.3166e-02, -1.2247e+00,
         -9.4041e-01],
        [-3.4569e+00, -9.5367e-06, -1.6000e+00, -6.0926e-01, -1.6265e+00,
         -2.1291e+00],
        [-2.6194e-01, -1.5621e+00, -1.3708e+00, -2.2186e-01, -4.7086e-01,
         -8.9677e-01],
        [-8.7165e-01, -3.3014e+00, -9.1785e-01, -1.4633e+00, -2.5730e+00,
         -1.0588e-02],
        [-2.0804e+00, -6.1944e+00, -3.3564e+00, -3.5039e+00, -1.2431e+00,
         -1.1669e+00],
        [-2.3578e+00, -7.4229e-02, -3.1739e+00, -1.8664e+00, -1.9020e+00,
         -1.7865e+00],
        [-2.0830e+00, -4.1411e-01, -4.3321e-04, -1.9695e-01, -1.6928e+00,
         -9.2550e-01],
        [-1.7445e+00, -5.1579e-01, -1.3087e-02, -2.6321e-01, -1.4815e+00,
         -2.6691e+00],
        [-2.9327e+00, -1.6565e+00, -7.6873e-01, -3.2560e-02, -3.7396e+00,
         -1.6636e+00],
        [-3.2617e+00, -2.2947e+00, -1.4257e+00, -5.5313e-05, -4.0768e+00,
         -2.1412e+00],
        [-1.0248e+00, -3.7786e-01, -3.9392e-01, -4.8981e-01, -1.9510e+00,
         -3.7980e+00]], device='cuda:0')
tensor([ -9.7500, -15.6611,  -3.5341, -11.3873,  -9.2601,  -9.7514, -18.6638,
        -11.0757, -11.2525, -11.0604,  -9.4219,  -4.7843,  -9.1378, -17.5452,
        -11.1609,  -5.3128,  -6.6871, -10.7937, -13.2003,  -8.0353],
       device='cuda:0')
tensor([-2.4535, -0.2901, -3.9247, -1.1220, -1.5955, -1.5887,  1.2498, -9.1435,
        -3.5233, -1.5013, -5.5216, -3.6907, -1.9309,  0.3129, -0.6646, -3.1990,
        -2.7821, -2.1807, -0.4765, -1.8947], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.3100, -1.1490, -1.4294, -1.0496, -1.6878, -1.3530],
        [-2.7616, -1.6012, -4.1103, -1.3447, -1.7531, -0.8735],
        [-4.8922, -1.3840, -4.5143, -1.1694, -2.2170, -1.3266],
        [-2.5350, -1.5854, -1.1537, -2.1287, -2.2950, -1.7582],
        [-5.2794, -2.8400, -1.6632, -1.3491, -3.3410, -1.4056],
        [-4.5881, -2.3652, -1.7863, -1.3442, -2.7877, -1.5950],
        [-1.2464, -1.2669, -1.2378, -1.1633, -1.8031, -1.3599],
        [-3.8997, -1.3528, -2.6711, -1.2712, -2.0868, -1.5950],
        [-6.4235, -1.4262, -3.0889, -1.2869, -1.8930, -1.8411],
        [-3.6519, -1.3809, -1.3032, -2.1987, -1.4227, -1.1041],
        [-4.9265, -1.5555, -2.4309, -1.1462, -2.2834, -1.3098],
        [-3.5567, -2.9494, -1.9916, -1.3178, -2.1886, -1.7691],
        [-1.6361, -1.3201, -2.1925, -1.0541, -1.6061, -1.2756],
        [-2.5149, -2.0689, -3.5402, -1.3849, -1.3972, -1.6035],
        [-2.1012, -1.2205, -1.5527, -1.1936, -1.4144, -1.3721],
        [-2.9242, -1.3382, -3.1379, -1.3162, -1.5160, -1.6221],
        [-1.6593, -1.1854, -1.3429, -1.2197, -2.1493, -3.1326],
        [-5.7883, -2.4672, -1.8824, -1.1151, -2.1456, -1.4226],
        [-5.2893, -5.2547, -2.5754, -1.0450, -2.1380, -2.2475],
        [-1.2839, -1.4401, -1.8681, -1.7884, -1.5362, -1.2992]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -7.9787, -12.4444, -15.5035, -11.4560, -15.8782, -14.4665,  -8.0775,
        -12.8766, -15.9595, -11.0613, -13.6523, -13.7732,  -9.0845, -12.5096,
         -8.8544, -11.8545, -10.6891, -14.8213, -18.5500,  -9.2159],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.3530e+00, -6.6271e-02, -7.1776e-01, -5.8736e-02, -1.3765e+00,
         -4.4359e-01],
        [-3.4886e+00, -3.8037e-01, -5.6702e+00, -6.0811e-02, -1.5460e+00,
         -1.1984e-01],
        [-4.0218e+00, -1.2450e-01, -5.6270e+00, -9.9350e-02, -3.3782e+00,
         -4.0447e-01],
        [-2.4077e+00, -4.1633e-01, -1.8322e-03, -1.9280e+00, -3.2150e+00,
         -1.4592e+00],
        [-3.7837e+00, -2.8505e+00, -1.5460e+00, -7.1367e-01, -4.8024e+00,
         -8.3540e-01],
        [-4.1418e+00, -2.1386e+00, -1.2679e+00, -1.1592e-01, -4.4606e+00,
         -1.1628e+00],
        [-9.2361e-01, -4.3927e-01, -1.6749e-01, -1.2130e-01, -1.5644e+00,
         -3.2334e-01],
        [-4.2723e+00, -1.2026e-03, -3.3002e+00, -7.6772e-02, -2.3826e+00,
         -7.7889e-01],
        [-4.2924e+00, -3.1145e-01, -4.1549e+00, -2.4779e-01, -2.6551e+00,
         -2.0529e+00],
        [-3.1130e+00, -3.3367e-02, -1.0161e-03, -2.0743e+00, -9.2751e-01,
         -1.6148e-02],
        [-4.1170e+00, -3.4159e-01, -3.2620e+00, -3.4339e-01, -2.8214e+00,
         -1.1395e-01],
        [-2.2343e+00, -3.5048e+00, -2.0335e+00, -1.6651e-01, -2.7838e+00,
         -1.7215e+00],
        [-2.3725e+00, -7.8929e-01, -2.1289e+00, -6.0748e-02, -1.0250e+00,
         -7.3159e-02],
        [-3.2665e+00, -1.4923e+00, -5.2419e+00, -1.1473e-01, -3.7353e-01,
         -1.6604e+00],
        [-1.8865e+00, -1.3787e-02, -9.8218e-01, -2.3341e-01, -1.0834e+00,
         -5.8889e-04],
        [-2.7208e+00, -4.5245e-01, -2.8820e+00, -3.8651e-01, -1.1551e+00,
         -1.1385e+00],
        [-2.0191e+00, -1.5061e-01, -1.1107e+00, -6.7434e-03, -2.5982e+00,
         -2.1818e+00],
        [-4.2405e+00, -2.4426e+00, -1.9833e+00, -1.3501e-02, -2.7390e+00,
         -2.1888e-01],
        [-4.7592e+00, -5.0474e+00, -2.8689e+00, -1.3798e-01, -2.6827e+00,
         -2.5764e+00],
        [-1.1485e+00, -1.1268e+00, -1.7991e+00, -1.1987e+00, -2.9730e-01,
         -7.4713e-03]], device='cuda:0')
tensor([ -4.0159, -11.2659, -13.6553,  -9.4280, -14.5317, -13.2876,  -3.5394,
        -10.8119, -13.7146,  -6.1654, -10.9994, -12.4444,  -6.4496, -12.1493,
         -4.1998,  -8.7355,  -8.0671, -11.6378, -18.0726,  -5.5779],
       device='cuda:0')
tensor([-3.9628, -1.1785, -1.8482, -2.0279, -1.3465, -1.1789, -4.5381, -2.0647,
        -2.2449, -4.8959, -2.6529, -1.3288, -2.6349, -0.3602, -4.6546, -3.1190,
        -2.6220, -3.1835, -0.4774, -3.6380], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.3623, -1.0375, -1.5495, -1.1132, -2.7530, -1.8789],
        [-1.9556, -1.3115, -1.5321, -1.0430, -2.7386, -2.2463],
        [-2.2365, -2.6820, -1.8806, -1.1103, -1.9133, -1.9004],
        [-2.9018, -3.3048, -3.1875, -1.6929, -2.9161, -1.3559],
        [-1.5268, -1.2747, -1.2624, -1.2047, -1.7597, -1.2513],
        [-1.6954, -1.2367, -2.1682, -1.1996, -5.0107, -3.1709],
        [-7.1765, -1.4185, -2.8862, -1.3013, -1.7879, -1.6904],
        [-3.1199, -1.3503, -2.5166, -3.9435, -1.7173, -1.5020],
        [-5.4426, -1.9918, -2.1314, -2.2051, -2.4152, -2.1964],
        [-3.0385, -2.9625, -3.6849, -0.9856, -1.6406, -1.4888],
        [-1.2324, -1.2052, -1.7659, -1.6655, -1.1324, -1.0638],
        [-1.7773, -1.5438, -3.4636, -1.6476, -1.3726, -1.4975],
        [-2.6565, -1.2137, -1.2008, -1.2538, -1.5418, -1.6934],
        [-2.4314, -1.3988, -1.3621, -1.2202, -1.3172, -1.6023],
        [-3.2322, -1.5250, -2.0192, -1.3878, -1.3443, -1.8365],
        [-2.2702, -1.7455, -3.6215, -1.8148, -1.1737, -1.3692],
        [-0.9912, -1.3242, -1.5122, -2.1874, -1.5491, -1.1939],
        [-5.2415, -1.7471, -2.8052, -2.8600, -2.2329, -2.8772],
        [-3.3956, -1.0881, -2.6440, -1.6116, -1.9892, -1.7594],
        [-3.5703, -2.2670, -2.0737, -1.9736, -2.5810, -3.0146]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.6944, -10.8270, -11.7231, -15.3590,  -8.2796, -14.4815, -16.2608,
        -14.1496, -16.3826, -13.8010,  -8.0652, -11.3025,  -9.5600,  -9.3319,
        -11.3449, -11.9948,  -8.7582, -17.7640, -12.4880, -15.4803],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.7794e+00, -3.7647e-01, -9.1492e-01, -3.5763e-05, -3.0608e+00,
         -1.7917e+00],
        [-2.2031e+00, -7.3978e-02, -1.2607e+00, -1.7458e-02, -3.6008e+00,
         -2.2037e+00],
        [-9.8228e-01, -3.0377e+00, -1.7898e+00, -1.5382e-01, -1.8715e+00,
         -1.8342e+00],
        [-1.9493e+00, -2.9025e+00, -3.4637e+00, -1.1388e+00, -3.6294e+00,
         -5.6928e-01],
        [-1.9315e+00, -1.3686e-01, -1.2973e-01, -4.9703e-02, -1.2166e+00,
         -5.6609e-03],
        [-1.8336e+00, -9.3865e-02, -2.3745e+00, -5.3438e-02, -6.0571e+00,
         -2.5181e+00],
        [-4.8461e+00, -3.0075e-01, -3.6110e+00, -3.8568e-01, -2.3486e+00,
         -1.5530e+00],
        [-2.6077e+00, -9.3217e-02, -3.0772e+00, -4.1596e+00, -1.9542e+00,
         -6.2053e-01],
        [-4.7298e+00, -1.3107e+00, -2.9781e+00, -2.9413e+00, -3.1901e+00,
         -2.6576e+00],
        [-1.4537e+00, -2.4934e+00, -2.7825e+00, -2.2457e-03, -7.3736e-01,
         -5.3156e-01],
        [-9.1135e-01, -2.1013e-03, -1.3140e+00, -1.0175e+00, -2.1047e-01,
         -7.5930e-03],
        [-1.6505e+00, -4.1180e-02, -5.2302e+00, -1.0288e+00, -4.5408e-01,
         -1.3360e+00],
        [-1.7148e+00, -4.7321e-02, -1.2774e-01, -2.1219e-02, -5.6092e-02,
         -8.7942e-01],
        [-1.7010e+00, -9.1440e-01, -9.1495e-01, -6.8808e-04, -4.1191e-01,
         -1.5408e+00],
        [-2.2409e+00, -1.1657e+00, -2.5442e+00, -6.3546e-01, -1.6703e-01,
         -2.0247e+00],
        [-1.8120e+00, -1.6564e+00, -3.8312e+00, -1.5422e+00, -4.3327e-02,
         -6.2320e-01],
        [-8.6901e-03, -8.5729e-01, -1.1424e+00, -2.2391e+00, -1.1225e+00,
         -5.3074e-01],
        [-4.9807e+00, -9.5870e-01, -3.4077e+00, -3.5141e+00, -2.5017e+00,
         -2.4697e+00],
        [-3.9777e+00, -6.9670e-01, -3.3635e+00, -9.2320e-01, -1.9863e+00,
         -1.9337e+00],
        [-2.2897e+00, -1.3848e+00, -2.4295e+00, -1.8914e+00, -3.7953e+00,
         -3.1715e+00]], device='cuda:0')
tensor([ -8.9233,  -9.3597,  -9.6693, -13.6530,  -3.4701, -12.9307, -13.0451,
        -12.5124, -17.8075,  -8.0008,  -3.4630,  -9.7408,  -2.8465,  -5.4837,
         -8.7781,  -9.5084,  -5.9007, -17.8326, -12.8812, -14.9623],
       device='cuda:0')
tensor([-1.7711, -1.4673, -2.0538, -1.7060, -4.8095, -1.5508, -3.2156, -1.6372,
         1.4250, -5.8002, -4.6023, -1.5617, -6.7135, -3.8481, -2.5668, -2.4864,
        -2.8575,  0.0686,  0.3931, -0.5180], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-6.1101, -2.1658, -2.6829, -1.5969, -1.8376, -1.5924],
        [-5.8429, -2.6999, -1.9775, -2.7362, -2.0574, -1.4849],
        [-1.3017, -2.5465, -3.8498, -1.0528, -1.4877, -1.2136],
        [-1.3071, -1.5729, -1.2234, -1.2952, -2.4857, -2.2374],
        [-3.1494, -1.6913, -1.7683, -1.2632, -1.7618, -1.2234],
        [-4.0753, -0.9778, -1.2753, -1.7639, -2.1190, -1.4632],
        [-2.8253, -1.2519, -1.1196, -1.2267, -2.0864, -1.9633],
        [-4.8998, -3.6415, -2.6267, -1.0791, -2.5012, -1.4484],
        [-5.1055, -2.5624, -2.4696, -2.3286, -1.7131, -1.4078],
        [-1.1973, -1.2925, -2.1428, -0.9018, -1.8059, -1.0474],
        [-3.5858, -1.4848, -3.1421, -1.2851, -1.8685, -1.1885],
        [-8.3765, -1.2748, -3.6379, -1.2643, -1.8519, -1.3688],
        [-2.8180, -1.1687, -1.1552, -1.7863, -1.4622, -1.5429],
        [-5.8779, -2.0009, -1.7361, -1.2025, -1.7536, -1.2571],
        [-6.3830, -3.2914, -1.8910, -1.0958, -1.9039, -1.6580],
        [-1.4541, -1.5570, -2.3752, -1.8532, -1.4464, -1.4468],
        [-2.0456, -1.5938, -3.0483, -1.3572, -1.5702, -0.8974],
        [-4.0293, -1.1548, -3.1941, -1.1532, -1.5499, -1.3376],
        [-3.5159, -1.3389, -1.4232, -1.1814, -2.2446, -1.9872],
        [-2.2165, -1.2465, -1.5366, -1.1873, -2.0629, -4.7638]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-15.9857, -16.7988, -11.4520, -10.1218, -10.8574, -11.6746, -10.4731,
        -16.1966, -15.5871,  -8.3878, -12.5547, -17.7743,  -9.9333, -13.8281,
        -16.2232, -10.1327, -10.5125, -12.4189, -11.6912, -13.0136],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.4998e+00, -1.1536e+00, -3.5981e+00, -9.8552e-01, -1.5676e+00,
         -1.2280e+00],
        [-4.5689e+00, -2.7853e+00, -1.8537e+00, -2.1873e+00, -2.0572e+00,
         -8.3310e-01],
        [-2.7134e-02, -3.4196e+00, -5.0575e+00, -4.6575e-03, -7.8326e-01,
         -3.1165e-01],
        [-1.2502e+00, -2.0458e+00, -7.7571e-02, -1.3153e-01, -3.2368e+00,
         -1.5937e+00],
        [-2.9832e+00, -1.2480e-01, -9.0721e-01, -9.7374e-03, -2.0150e+00,
         -4.2693e-01],
        [-4.0204e+00, -2.0896e-02, -1.0186e-01, -1.3453e+00, -2.2805e+00,
         -2.7780e-01],
        [-3.0940e+00, -1.2418e-01, -1.1677e-01, -3.4627e-01, -2.4855e+00,
         -2.0062e+00],
        [-3.6319e+00, -4.3890e+00, -3.6155e+00, -2.0249e-01, -3.3113e+00,
         -9.4100e-01],
        [-4.7399e+00, -2.4536e+00, -3.1917e+00, -1.3158e+00, -1.1999e+00,
         -7.6265e-01],
        [-8.3508e-01, -6.9293e-01, -2.1679e+00, -1.6603e-02, -1.7300e+00,
         -2.6892e-03],
        [-4.2110e+00, -9.1765e-02, -4.2941e+00, -3.6532e-03, -2.0315e+00,
         -2.5326e-01],
        [-5.3227e+00, -1.8862e-02, -4.5235e+00, -3.6581e-01, -2.4349e+00,
         -7.4611e-01],
        [-2.5348e+00, -4.4930e-01, -2.0515e-01, -1.3731e+00, -1.1253e+00,
         -1.2649e+00],
        [-4.2431e+00, -1.1568e+00, -1.4708e+00, -1.8579e-01, -1.2135e+00,
         -4.5146e-02],
        [-4.6971e+00, -3.6682e+00, -1.5752e+00, -1.1222e-01, -1.9204e+00,
         -1.2090e+00],
        [-1.6984e+00, -1.4020e+00, -2.6318e+00, -1.2510e+00, -6.8584e-02,
         -7.6931e-01],
        [-2.3080e+00, -2.6441e-01, -4.4753e+00, -1.2137e-01, -1.1435e+00,
         -1.4107e-02],
        [-3.5904e+00, -2.4113e-02, -3.9887e+00, -8.6295e-02, -1.3732e+00,
         -1.5141e-01],
        [-3.2069e+00, -1.0760e+00, -4.7674e-01, -5.5724e-02, -3.0135e+00,
         -1.9449e+00],
        [-2.9333e+00, -5.8611e-02, -1.4926e+00, -2.6882e-02, -2.3051e+00,
         -3.2874e+00]], device='cuda:0')
tensor([-13.0326, -14.2856,  -9.6037,  -8.3356,  -6.4670,  -8.0467,  -8.1729,
        -16.0911, -13.6635,  -5.4451, -10.8853, -13.4118,  -6.9525,  -8.3150,
        -13.1821,  -7.8211,  -8.3266,  -9.2140,  -9.7739, -10.1038],
       device='cuda:0')
tensor([-2.9531, -2.5133, -1.8483, -1.7862, -4.3904, -3.6279, -2.3002, -0.1055,
        -1.9235, -2.9427, -1.6694, -4.3624, -2.9807, -5.5131, -3.0410, -2.3116,
        -2.1859, -3.2048, -1.9173, -2.9098], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.4204, -1.5682, -2.5767, -1.1725, -1.5669, -1.4503],
        [-3.6050, -1.3256, -1.2202, -1.1927, -2.9329, -1.8230],
        [-5.5410, -2.3651, -1.8301, -1.1418, -1.9503, -1.4127],
        [-2.8783, -3.8877, -2.1347, -1.6431, -2.2785, -1.3443],
        [-2.6930, -1.2104, -1.2768, -1.1397, -1.7940, -1.4152],
        [-2.4612, -1.5906, -2.8323, -1.2460, -1.9562, -1.2890],
        [-4.5629, -1.4270, -2.2324, -1.2086, -1.7103, -1.1946],
        [-3.6920, -1.2556, -1.6322, -1.3201, -2.2731, -2.8438],
        [-2.8843, -1.1635, -1.4181, -1.2361, -2.3669, -2.0126],
        [-5.0306, -2.2179, -1.6576, -1.1176, -2.0047, -1.4303],
        [-4.9268, -3.3755, -2.2986, -1.7521, -3.1256, -1.7369],
        [-2.3183, -1.2926, -1.2437, -1.2539, -1.9647, -1.3260],
        [-2.8620, -1.6853, -3.0370, -1.3530, -1.3798, -0.8654],
        [-3.9713, -1.2950, -1.9380, -1.2550, -1.6024, -1.2939],
        [-6.7436, -1.2018, -2.7692, -1.2355, -1.8332, -1.4251],
        [-2.5489, -1.1329, -1.0494, -1.2360, -1.6089, -5.3459],
        [-4.0340, -2.1344, -1.8792, -1.1456, -2.4604, -1.2510],
        [-4.2793, -3.1468, -2.6066, -1.0891, -2.1115, -2.2409],
        [-1.6758, -1.2333, -1.2211, -1.0494, -2.1463, -1.2671],
        [-3.6549, -1.3297, -2.1092, -1.4713, -1.5245, -1.1545]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.7550, -12.0995, -14.2410, -14.1665,  -9.5291, -11.3752, -12.3357,
        -13.0169, -11.0815, -13.4588, -17.2156,  -9.3992, -11.1826, -11.3555,
        -15.2083, -12.9220, -12.9046, -15.4742,  -8.5929, -11.2439],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.0341e+00, -1.7952e+00, -2.3011e+00, -4.9039e-02, -1.4181e+00,
         -8.8573e-01],
        [-3.6986e+00, -9.0398e-01, -7.2581e-01, -2.4346e-02, -3.6728e+00,
         -1.5083e+00],
        [-4.0533e+00, -2.0976e+00, -1.7361e+00, -4.6100e-02, -2.0088e+00,
         -6.1081e-01],
        [-2.1547e+00, -4.0903e+00, -2.0141e+00, -6.9696e-01, -3.1745e+00,
         -3.4921e-01],
        [-3.7105e+00, -1.5715e-01, -4.7952e-01, -2.1553e-04, -1.4598e+00,
         -6.8988e-01],
        [-2.9199e+00, -4.8370e-01, -3.8933e+00, -2.3522e-02, -1.9685e+00,
         -4.9533e-01],
        [-3.4018e+00, -8.3718e-02, -2.3171e+00, -1.7210e-01, -2.0422e+00,
         -9.7548e-02],
        [-3.3522e+00, -5.9704e-01, -7.5784e-01, -3.5711e-01, -3.2139e+00,
         -3.4928e+00],
        [-3.2436e+00, -1.4033e-01, -1.0930e+00, -7.4114e-02, -2.9062e+00,
         -1.5453e+00],
        [-3.4552e+00, -1.9091e+00, -2.5144e-01, -1.7234e-01, -2.2210e+00,
         -4.5187e-01],
        [-4.9246e+00, -3.4939e+00, -2.3599e+00, -8.5411e-01, -4.8848e+00,
         -1.6000e+00],
        [-3.4236e+00, -5.2582e-01, -2.9364e-01, -2.7866e-01, -2.0443e+00,
         -2.8195e-01],
        [-3.6811e+00, -4.3380e-01, -4.4491e+00, -2.9997e-02, -2.0474e-01,
         -1.1697e-02],
        [-3.3283e+00, -4.4986e-02, -1.8446e+00, -5.4439e-01, -1.6554e+00,
         -2.9635e-01],
        [-5.0969e+00, -1.7113e-01, -2.3312e+00, -1.6409e-01, -2.1218e+00,
         -7.1488e-01],
        [-2.8464e+00, -1.4700e-01, -1.3540e-01, -1.2673e-01, -1.1870e+00,
         -4.6072e+00],
        [-3.0527e+00, -1.4760e+00, -1.9664e+00, -5.4142e-02, -3.6003e+00,
         -3.2042e-02],
        [-3.5411e+00, -3.2845e+00, -3.1962e+00, -9.2127e-02, -2.6073e+00,
         -2.8424e+00],
        [-2.3836e+00, -1.4305e-02, -1.3641e-01, -6.8669e-02, -2.5989e+00,
         -1.6400e-01],
        [-4.2020e+00, -2.7337e-02, -2.3338e+00, -6.0381e-01, -1.0249e+00,
         -3.0674e-01]], device='cuda:0')
tensor([ -9.4833, -10.5338, -10.5528, -12.4797,  -6.4971,  -9.7842,  -8.1145,
        -11.7708,  -9.0025,  -8.4610, -18.1173,  -6.8479,  -8.8104,  -7.7140,
        -10.6000,  -9.0497, -10.1816, -15.5636,  -5.3659,  -8.4985],
       device='cuda:0')
tensor([-2.2717, -1.5657, -3.6882, -1.6869, -3.0320, -1.5911, -4.2212, -1.2461,
        -2.0790, -4.9978,  0.9018, -2.5512, -2.3721, -3.6415, -4.6083, -3.8723,
        -2.7230,  0.0895, -3.2270, -2.7453], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.9546, -1.3433, -3.3101, -1.3386, -2.9222, -0.8459],
        [-7.3035, -1.4219, -3.7203, -1.2151, -1.7521, -1.2648],
        [-3.5621, -1.3125, -2.8348, -1.3213, -2.1814, -1.8584],
        [-3.1525, -1.1818, -2.0748, -1.2810, -2.3130, -1.3196],
        [-4.8797, -4.0670, -2.1282, -1.1815, -2.6545, -1.3379],
        [-4.0479, -5.2721, -2.3991, -1.0140, -2.0039, -1.6534],
        [-1.5478, -1.7044, -2.0229, -1.3971, -1.5269, -1.2686],
        [-4.6949, -1.7736, -1.9339, -1.5355, -1.3166, -0.9871],
        [-2.5573, -1.2479, -2.9612, -1.0742, -1.5632, -1.7400],
        [-3.9685, -1.1922, -1.6926, -1.5989, -2.2625, -1.7428],
        [-1.7866, -2.5702, -1.5636, -1.2538, -1.4028, -1.0989],
        [-6.9998, -3.0675, -1.9169, -0.9952, -2.5470, -1.6629],
        [-1.1874, -1.2109, -1.6115, -1.0759, -1.5513, -1.3458],
        [-2.3201, -1.2762, -3.0826, -1.2005, -1.1834, -0.9867],
        [-1.4948, -2.1387, -1.8409, -2.0957, -1.9090, -1.2310],
        [-2.0218, -2.4885, -1.3440, -3.7399, -1.2964, -1.4106],
        [-2.3956, -4.2003, -1.3109, -1.6236, -2.3911, -1.6256],
        [-3.4359, -3.6198, -1.8545, -2.7298, -2.2133, -1.1990],
        [-2.4301, -1.4748, -1.6705, -2.0263, -1.2845, -1.3780],
        [-3.7528, -1.3033, -4.2248, -1.3094, -1.6701, -3.7452]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.7148, -16.6776, -13.0705, -11.3227, -16.2488, -16.3904,  -9.4678,
        -12.2415, -11.1437, -12.4575,  -9.6759, -17.1893,  -7.9829, -10.0495,
        -10.7102, -12.3012, -13.5472, -15.0523, -10.2642, -16.0057],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.3526e+00, -2.8497e-02, -4.2131e+00, -2.5825e-01, -3.7195e+00,
         -2.1652e-03],
        [-4.9039e+00, -6.4130e-02, -4.8134e+00, -1.4761e-01, -2.1265e+00,
         -5.9810e-01],
        [-3.5036e+00, -3.8848e-01, -2.8335e+00, -4.3453e-01, -2.8547e+00,
         -1.5460e+00],
        [-3.2422e+00, -2.7914e-01, -2.4443e+00, -3.2124e-01, -2.8659e+00,
         -6.4567e-01],
        [-3.7385e+00, -5.2632e+00, -2.9446e+00, -5.2614e-02, -4.5377e+00,
         -1.0137e-01],
        [-3.6959e+00, -5.2030e+00, -2.6448e+00, -1.6326e-02, -2.4301e+00,
         -1.4107e+00],
        [-1.9842e+00, -1.9643e+00, -2.1028e+00, -4.4708e-01, -6.1678e-01,
         -2.0485e-02],
        [-4.4988e+00, -6.4698e-01, -1.8395e+00, -5.5809e-01, -6.1800e-03,
         -3.6333e-02],
        [-2.2600e+00, -1.5255e-02, -3.9906e+00, -2.7169e-02, -1.6403e+00,
         -2.0206e+00],
        [-3.5283e+00, -2.5584e-01, -7.8486e-01, -1.0279e+00, -3.1449e+00,
         -1.6182e+00],
        [-8.6582e-01, -2.2439e+00, -1.3678e+00, -2.7047e-01, -2.7291e-01,
         -1.2773e-01],
        [-5.0871e+00, -3.9512e+00, -1.8238e+00, -5.0161e-03, -3.9176e+00,
         -8.1455e-01],
        [-7.5965e-01, -2.0133e-01, -1.1623e+00, -2.1132e-02, -3.7794e-01,
         -2.1471e-01],
        [-2.6368e+00, -7.1735e-01, -3.9627e+00, -3.7877e-02, -4.8267e-02,
         -3.3312e-03],
        [-1.3635e+00, -1.7813e+00, -1.3992e+00, -1.9123e+00, -2.6267e+00,
         -4.4145e-01],
        [-1.6817e+00, -1.7205e+00, -1.9187e-03, -4.7565e+00, -2.9398e-01,
         -7.4207e-02],
        [-1.4618e+00, -5.1047e+00, -3.1326e-01, -1.3132e+00, -2.8985e+00,
         -1.2405e+00],
        [-2.0395e+00, -4.3353e+00, -1.8088e+00, -1.9821e+00, -2.3534e+00,
         -1.6686e-03],
        [-1.4879e+00, -1.3092e+00, -1.0908e+00, -1.2551e+00, -2.6514e-01,
         -8.1167e-01],
        [-4.2217e+00, -1.4424e-05, -4.7417e+00, -1.3347e-03, -1.4256e+00,
         -2.3631e+00]], device='cuda:0')
tensor([-10.5742, -12.6536, -11.5608,  -9.7985, -16.6379, -15.4009,  -7.1357,
         -7.5859,  -9.9539, -10.3601,  -5.1486, -15.5992,  -2.7370,  -7.4063,
         -9.5244,  -8.5288, -12.3320, -12.5209,  -6.2199, -12.7534],
       device='cuda:0')
tensor([-1.1406, -4.0239, -1.5097, -1.5242,  0.3891, -0.9895, -2.3321, -4.6556,
        -1.1898, -2.0974, -4.5273, -1.5901, -5.2458, -2.6432, -1.1858, -3.7724,
        -1.2152, -2.5315, -4.0443, -3.2522], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.5078, -1.6389, -1.7063, -2.3522, -1.9010, -1.2600],
        [-5.6440, -2.5990, -2.0513, -1.2008, -1.7570, -1.5129],
        [-3.8827, -1.9471, -2.3661, -1.9484, -2.7657, -1.3638],
        [-1.2932, -1.2235, -1.2144, -0.8735, -1.1892, -1.8287],
        [-1.7447, -1.3469, -2.4787, -1.2330, -1.6913, -1.8959],
        [-3.6546, -1.3763, -1.7241, -1.4201, -2.2622, -1.6571],
        [-4.5545, -1.2465, -1.2763, -1.6215, -1.4329, -1.1489],
        [-4.7943, -2.1694, -1.8387, -2.7960, -2.4596, -2.2106],
        [-5.5655, -2.1776, -3.0262, -0.9436, -2.2553, -1.3483],
        [-1.7158, -1.7901, -4.5520, -0.9744, -1.1399, -1.0823],
        [-2.2303, -1.6073, -2.4295, -1.4626, -1.5245, -1.1445],
        [-2.9348, -1.0927, -2.5032, -1.2690, -1.8753, -1.4599],
        [-1.4478, -1.3402, -1.2482, -1.8668, -1.4877, -1.5723],
        [-3.1927, -1.4945, -1.7727, -1.0974, -1.7594, -1.1692],
        [-3.1199, -2.3162, -1.8885, -2.3402, -2.0070, -1.4494],
        [-1.4914, -1.4659, -1.6257, -1.1849, -1.4369, -1.1495],
        [-1.5329, -1.6229, -1.5281, -1.2708, -1.3162, -0.9307],
        [-6.3491, -1.2691, -2.4707, -1.7371, -1.2970, -1.2895],
        [-1.2625, -2.7215, -1.8315, -2.9783, -1.3814, -1.6900],
        [-3.2763, -2.3948, -2.0775, -2.5582, -2.3858, -1.2228]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.3661, -14.7650, -14.2739,  -7.6225, -10.3905, -12.0945, -11.2807,
        -16.2686, -15.3164, -11.2545, -10.3987, -11.1349,  -8.9630, -10.4859,
        -13.1213,  -8.3543,  -8.2016, -14.4126, -11.8651, -13.9153],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.2052, -0.6829, -1.7367, -2.3174, -2.3376, -0.0655],
        [-3.9879, -2.9266, -2.6545, -0.5371, -1.1874, -0.3757],
        [-2.9015, -1.2928, -2.7494, -1.0798, -3.6916, -0.6243],
        [-1.2409, -0.5527, -0.0909, -0.0497, -0.0135, -1.6662],
        [-1.6962, -0.0487, -2.8950, -0.0093, -1.7633, -1.8038],
        [-3.3192, -0.6961, -0.9742, -0.6671, -3.2711, -1.4292],
        [-3.6590, -0.0448, -0.6020, -0.9761, -1.1130, -0.0661],
        [-3.8896, -1.8424, -1.8806, -3.5392, -3.3469, -2.7527],
        [-4.1462, -1.6646, -2.7080, -0.0280, -3.0558, -0.0123],
        [-2.4716, -2.2397, -5.2156, -0.1953, -0.1230, -0.0375],
        [-2.2844, -0.2311, -3.2720, -0.3761, -1.1210, -0.5605],
        [-2.8930, -0.1587, -3.1266, -0.1309, -2.4566, -0.4863],
        [-1.3316, -0.0220, -0.0370, -1.4942, -1.1055, -1.0356],
        [-2.2474, -0.1902, -1.7580, -0.1968, -1.3701, -0.0392],
        [-1.8970, -2.2345, -1.6938, -1.1303, -2.0775, -0.0233],
        [-1.8996, -1.2338, -1.1558, -0.2979, -0.5073, -0.0563],
        [-1.5331, -0.4842, -0.7074, -0.0415, -0.2798, -0.0069],
        [-5.1127, -0.0536, -2.9316, -1.1515, -0.8804, -0.0270],
        [-0.9802, -2.4947, -2.1269, -3.5040, -0.4036, -1.5626],
        [-1.5220, -2.3894, -2.9971, -2.7665, -3.3147, -0.0562]],
       device='cuda:0')
tensor([ -9.3452, -11.6692, -12.3393,  -3.6139,  -8.2162, -10.3569,  -6.4610,
        -17.2514, -11.6149, -10.2827,  -7.8452,  -9.2520,  -5.0260,  -5.8017,
         -9.0563,  -5.1507,  -3.0530, -10.1568, -11.0719, -13.0459],
       device='cuda:0')
tensor([-2.0209, -3.0958, -1.9345, -4.0086, -2.1743, -1.7375, -4.8197,  0.9827,
        -3.7016, -0.9718, -2.5535, -1.8829, -3.9371, -4.6842, -4.0649, -3.2036,
        -5.1487, -4.2558, -0.7932, -0.8693], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.3172, -1.2070, -1.8072, -1.1362, -1.4953, -1.1857],
        [-1.5260, -1.5249, -2.2271, -1.2910, -1.4284, -0.9765],
        [-7.2390, -1.2698, -3.3252, -1.1118, -1.3312, -1.6269],
        [-4.3570, -1.3154, -2.3744, -1.2490, -1.7512, -1.3332],
        [-2.2786, -1.1583, -1.8919, -1.1112, -2.7458, -1.8917],
        [-2.9521, -2.2745, -2.3561, -1.1190, -1.7926, -1.5662],
        [-2.9019, -3.7371, -2.2214, -1.0387, -3.0068, -2.6634],
        [-1.7643, -1.2100, -1.7322, -1.1405, -1.9754, -1.2217],
        [-2.8386, -1.3803, -1.8971, -1.2315, -1.8985, -1.0150],
        [-3.8220, -1.3825, -4.6789, -1.2406, -1.3921, -1.5011],
        [-3.2785, -1.2682, -1.2155, -1.8693, -1.8503, -1.5948],
        [-6.9933, -2.2190, -1.7319, -1.2342, -1.8628, -1.3486],
        [-4.3347, -3.6409, -2.5083, -1.4165, -1.9871, -1.3955],
        [-1.2266, -1.3741, -2.5512, -1.8117, -1.3862, -2.0260],
        [-1.6445, -1.5473, -1.9631, -1.3113, -1.6186, -1.0146],
        [-6.0212, -1.5788, -2.5404, -1.1833, -1.3008, -1.3197],
        [-4.1409, -1.3343, -1.3046, -2.3896, -1.7417, -2.0655],
        [-2.3945, -2.6075, -2.3007, -1.1041, -2.3193, -1.2024],
        [-3.7858, -2.2036, -2.0124, -1.5516, -1.7327, -1.4450],
        [-1.2604, -1.2794, -1.7506, -1.2347, -1.5201, -1.2525]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.1485,  -8.9740, -15.9041, -12.3802, -11.0775, -12.0606, -15.5692,
         -9.0441, -10.2611, -14.0172, -11.0766, -15.3896, -15.2830, -10.3758,
         -9.0995, -13.9442, -12.9765, -11.9284, -12.7311,  -8.2977],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.2525e+00, -3.8687e-01, -1.5381e+00, -2.3172e-01, -7.4968e-01,
         -1.6684e-01],
        [-1.5274e+00, -1.6410e-01, -2.6563e+00, -1.9679e-03, -6.7309e-01,
         -1.6656e-02],
        [-4.8288e+00, -2.5412e-02, -4.0719e+00, -2.1955e-03, -5.9430e-01,
         -1.6289e+00],
        [-3.7946e+00, -1.0285e-02, -2.2340e+00, -1.5302e-01, -1.9505e+00,
         -5.3579e-01],
        [-2.5225e+00, -3.2896e-02, -1.9228e+00, -8.0904e-03, -3.7520e+00,
         -1.9055e+00],
        [-1.6224e+00, -1.9603e+00, -3.9429e+00, -3.2198e-02, -1.2024e+00,
         -4.9025e-01],
        [-1.9080e+00, -3.4763e+00, -2.1416e+00, -1.7934e-01, -3.8522e+00,
         -3.1503e+00],
        [-2.5547e+00, -1.4268e-01, -1.5360e+00, -1.3163e-01, -2.0954e+00,
         -4.0034e-02],
        [-3.2131e+00, -1.5403e-03, -1.8452e+00, -1.4186e-03, -1.8687e+00,
         -6.7225e-02],
        [-2.9353e+00, -6.8013e-02, -5.8875e+00, -2.0483e-01, -6.8984e-01,
         -1.2791e+00],
        [-2.9918e+00, -3.1579e-04, -2.0098e-01, -1.5262e+00, -2.1866e+00,
         -1.2139e+00],
        [-4.8862e+00, -1.9233e+00, -1.6175e+00, -4.9087e-01, -1.7182e+00,
         -1.8105e-01],
        [-3.3514e+00, -4.2843e+00, -3.2328e+00, -3.0552e-01, -2.1045e+00,
         -1.1994e-01],
        [-9.2533e-01, -1.0064e+00, -2.7739e+00, -1.2396e+00, -1.5360e-01,
         -2.1783e+00],
        [-1.6030e+00, -2.1752e-01, -1.9371e+00, -3.6787e-02, -1.3750e+00,
         -5.1134e-02],
        [-4.4892e+00, -4.4383e-01, -2.8855e+00, -1.3216e-01, -5.5084e-01,
         -2.9045e-03],
        [-4.0978e+00, -2.2027e-01, -3.1320e-01, -2.2018e+00, -1.7630e+00,
         -2.3422e+00],
        [-1.4533e+00, -2.4946e+00, -2.8219e+00, -7.8273e-02, -2.8291e+00,
         -2.6215e-01],
        [-2.4243e+00, -1.8174e+00, -2.1772e+00, -4.0789e-01, -9.2841e-01,
         -7.2257e-02],
        [-3.6008e-01, -5.0451e-01, -1.5552e+00, -1.9901e-01, -2.1622e-01,
         -1.8996e-01]], device='cuda:0')
tensor([ -4.3257,  -5.0395, -11.1514,  -8.6782, -10.1439,  -9.2504, -14.7078,
         -6.5005,  -6.9972, -11.0646,  -8.1198, -10.8171, -13.3984,  -8.2770,
         -5.2206,  -8.5044, -10.9383,  -9.9394,  -7.8275,  -3.0250],
       device='cuda:0')
tensor([-3.8229, -3.9345, -4.7527, -3.7020, -0.9336, -2.8102, -0.8614, -2.5436,
        -3.2639, -2.9526, -2.9569, -4.5725, -1.8846, -2.0988, -3.8789, -5.4398,
        -2.0383, -1.9890, -4.9036, -5.2727], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.4359, -3.1497, -1.9298, -1.0791, -1.9065, -1.5185],
        [-2.3448, -3.4109, -2.8205, -1.1042, -1.9800, -2.6254],
        [-1.6261, -1.5815, -1.1958, -1.3391, -1.4662, -1.3033],
        [-3.8950, -1.5492, -2.1702, -1.5194, -1.5281, -1.0609],
        [-3.1528, -1.1747, -3.7184, -1.1731, -2.5633, -1.5061],
        [-2.3774, -1.3199, -1.5567, -1.7418, -1.6411, -1.1369],
        [-5.7531, -1.9605, -2.3347, -1.2914, -1.9314, -1.7073],
        [-3.9195, -2.8216, -2.8267, -2.0761, -1.9912, -1.7381],
        [-1.8170, -1.1339, -1.6505, -0.8474, -1.7980, -1.3667],
        [-2.0633, -1.5317, -2.2592, -1.3857, -1.4430, -1.1666],
        [-3.0375, -1.4691, -2.3107, -1.2734, -1.5404, -1.0714],
        [-3.2127, -1.9321, -1.0889, -1.9747, -1.3412, -1.5749],
        [-1.0451, -1.6297, -1.5235, -1.0048, -1.5090, -1.6370],
        [-7.5628, -3.6537, -2.0972, -1.2187, -1.8578, -1.5339],
        [-3.8828, -3.6115, -2.2758, -1.6363, -2.0105, -1.2810],
        [-1.1488, -2.1780, -3.0958, -1.9596, -1.4040, -1.1305],
        [-2.6702, -1.4815, -2.0068, -1.4013, -1.4717, -1.0670],
        [-3.6175, -1.3316, -2.3011, -1.1069, -1.4096, -1.2396],
        [-3.8256, -1.3762, -1.4005, -1.1185, -2.0262, -1.3991],
        [-2.6195, -1.3715, -1.5619, -1.1247, -2.0351, -1.9153]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.0195, -14.2857,  -8.5120, -11.7228, -13.2883,  -9.7739, -14.9785,
        -15.3731,  -8.6137,  -9.8494, -10.7024, -11.1245,  -8.3492, -17.9241,
        -14.6980, -10.9168, -10.0984, -11.0064, -11.1463, -10.6279],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.2899e+00, -3.7746e+00, -2.1683e+00, -1.3042e-01, -1.8815e+00,
         -6.7327e-01],
        [-1.3171e+00, -3.2222e+00, -3.1750e+00, -3.0006e-01, -2.3032e+00,
         -3.1814e+00],
        [-2.0270e+00, -1.5839e+00, -1.6410e-01, -2.6364e-01, -2.8286e-01,
         -1.4409e-02],
        [-4.0084e+00, -2.8581e-01, -2.5398e+00, -5.9361e-01, -1.2134e+00,
         -1.7341e-01],
        [-2.7508e+00, -6.7033e-02, -4.6727e+00, -1.5708e-01, -3.8271e+00,
         -1.1869e+00],
        [-2.4321e+00, -4.8723e-02, -1.5785e+00, -1.1772e+00, -1.6133e+00,
         -7.7890e-02],
        [-4.5104e+00, -1.0004e+00, -3.7117e+00, -4.9032e-01, -1.8841e+00,
         -1.6147e+00],
        [-3.1888e+00, -2.9041e+00, -3.4210e+00, -1.1215e+00, -2.1687e+00,
         -1.6186e+00],
        [-2.4921e+00, -4.9059e-01, -1.0718e+00, -1.9761e-02, -1.7505e+00,
         -6.8093e-01],
        [-2.9326e+00, -1.4278e-02, -2.5887e+00, -3.1505e-02, -1.5116e-04,
         -9.1655e-02],
        [-2.7085e+00, -1.0524e-01, -2.8256e+00, -1.6250e-01, -1.6045e+00,
         -1.0836e-02],
        [-2.9095e+00, -1.1338e+00, -8.7023e-06, -1.8846e+00, -3.8707e-01,
         -1.3563e+00],
        [-6.2325e-01, -6.3526e-01, -1.0431e+00, -2.8839e-02, -8.4538e-01,
         -1.2758e+00],
        [-4.8026e+00, -5.2613e+00, -2.8217e+00, -6.4717e-01, -1.5387e+00,
         -7.8643e-01],
        [-2.6758e+00, -3.2164e+00, -2.2299e+00, -1.0362e+00, -1.9014e+00,
         -5.0795e-01],
        [-2.2930e-01, -2.9508e+00, -3.6954e+00, -1.3416e+00, -1.9101e-01,
         -9.5606e-05],
        [-2.5289e+00, -9.9826e-01, -2.0904e+00, -2.7679e-01, -1.2004e+00,
         -4.5629e-02],
        [-2.8574e+00, -6.7041e-02, -2.7645e+00, -2.4302e-02, -1.2127e+00,
         -4.8576e-02],
        [-4.2706e+00, -1.3675e+00, -5.6745e-01, -5.8857e-03, -2.0664e+00,
         -5.7503e-01],
        [-3.1673e+00, -9.7239e-04, -1.3475e+00, -4.5507e-03, -2.4220e+00,
         -1.5176e+00]], device='cuda:0')
tensor([-10.9180, -13.4990,  -4.3359,  -8.8145, -12.6616,  -6.9277, -13.2115,
        -14.4226,  -6.5057,  -5.6589,  -7.4172,  -7.6713,  -4.4516, -15.8578,
        -11.5677,  -8.4083,  -7.1405,  -6.9745,  -8.8529,  -8.4599],
       device='cuda:0')
tensor([-2.1015, -0.7867, -4.1761, -2.9083, -0.6267, -2.8462, -1.7670, -0.9506,
        -2.1079, -4.1905, -3.2852, -3.4532, -3.8975, -2.0664, -3.1303, -2.5085,
        -2.9579, -4.0319, -2.2934, -2.1680], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.6147, -1.3793, -2.3091, -1.5333, -2.3915, -3.0344],
        [-1.9572, -1.6810, -3.1188, -1.2364, -2.4872, -2.6764],
        [-6.6545, -1.4333, -1.5695, -1.5226, -2.3918, -1.1535],
        [-3.5205, -3.2820, -2.1386, -1.6180, -2.1126, -2.5123],
        [-2.6166, -3.8202, -3.4083, -0.9347, -1.9630, -1.5634],
        [-1.2419, -1.4460, -1.8498, -1.1519, -1.3703, -1.2260],
        [-3.1006, -1.6791, -2.0493, -1.9402, -1.6180, -3.7998],
        [-2.3696, -2.2378, -2.0260, -1.3144, -1.2401, -1.8070],
        [-3.2703, -1.3362, -3.1376, -1.3197, -1.5858, -2.1378],
        [-2.6979, -1.2127, -1.2893, -1.1197, -1.7834, -2.6113],
        [-1.0261, -2.2461, -2.9276, -1.1366, -1.3363, -2.7134],
        [-5.9163, -4.2845, -2.4516, -0.9693, -2.4210, -1.6099],
        [-2.7122, -2.2912, -2.4863, -1.5940, -2.5294, -1.1534],
        [-1.2345, -2.3219, -2.6247, -1.8852, -1.3179, -1.6115],
        [-2.2404, -1.3342, -2.0964, -1.4707, -1.4187, -0.9590],
        [-1.1255, -1.0827, -1.9489, -0.9837, -2.2250, -1.3705],
        [-2.9326, -1.2018, -1.3706, -1.2572, -2.3985, -1.5932],
        [-2.4884, -1.9578, -1.8371, -1.1281, -1.4091, -1.0966],
        [-6.0162, -3.0582, -1.9754, -0.9407, -1.9147, -1.6884],
        [-1.2766, -1.3163, -1.0720, -1.0214, -1.9217, -1.3721]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.2622, -13.1570, -14.7251, -15.1839, -14.3061,  -8.2859, -14.1869,
        -10.9948, -12.7873, -10.7143, -11.3860, -17.6527, -12.7665, -10.9957,
         -9.5195,  -8.7363, -10.7539,  -9.9171, -15.5937,  -7.9802],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.9640e+00, -6.1357e-02, -2.5000e+00, -7.9421e-01, -3.0631e+00,
         -2.4124e+00],
        [-1.8048e+00, -2.0361e+00, -4.2687e+00, -1.1413e-01, -3.8111e+00,
         -3.0881e+00],
        [-4.8437e+00, -1.2600e-02, -1.3974e+00, -8.0709e-01, -3.6901e+00,
         -6.4787e-02],
        [-2.7289e+00, -4.6735e+00, -3.1090e+00, -1.7174e+00, -2.3996e+00,
         -3.4451e+00],
        [-1.1725e+00, -3.2643e+00, -3.0751e+00, -2.6842e-02, -2.0170e+00,
         -8.9555e-01],
        [-1.6737e+00, -2.5859e-01, -1.5843e+00, -9.6921e-02, -1.9604e-01,
         -2.4869e-01],
        [-3.2824e+00, -8.1630e-01, -2.3174e+00, -1.6260e+00, -1.0502e+00,
         -4.1956e+00],
        [-2.3322e+00, -1.3103e+00, -2.4337e+00, -2.7224e-01, -4.4538e-01,
         -1.1210e+00],
        [-2.8328e+00, -5.5034e-02, -2.9929e+00, -3.6122e-01, -1.4780e+00,
         -2.2383e+00],
        [-3.6813e+00, -3.4288e-01, -1.3194e-01, -1.7684e-02, -1.5787e+00,
         -1.3981e+00],
        [-1.6487e-02, -1.9973e+00, -4.1031e+00, -3.6675e-02, -5.0586e-03,
         -2.7633e+00],
        [-4.4137e+00, -5.1475e+00, -2.9828e+00, -4.2956e-02, -3.2013e+00,
         -9.1433e-01],
        [-1.7619e+00, -1.7323e+00, -2.4242e+00, -1.0601e+00, -2.8441e+00,
         -1.6471e-01],
        [-7.8955e-01, -3.0287e+00, -2.4576e+00, -1.8295e+00, -3.2285e-01,
         -1.3307e+00],
        [-2.3528e+00, -1.3830e-01, -2.2900e+00, -4.4297e-01, -6.6491e-01,
         -1.3990e-01],
        [-4.6545e-01, -1.7760e-01, -2.0034e+00, -1.8906e-01, -2.9186e+00,
         -8.5067e-01],
        [-3.4445e+00, -1.1321e+00, -4.8561e-01, -2.8958e-01, -2.6842e+00,
         -1.1606e+00],
        [-1.5875e+00, -1.3297e+00, -1.9455e+00, -8.6663e-02, -2.1524e-01,
         -1.3522e-03],
        [-4.3320e+00, -3.8852e+00, -1.9054e+00, -3.6201e-03, -1.7723e+00,
         -9.3304e-01],
        [-1.3052e+00, -1.8850e-01, -4.9355e-02, -9.6162e-02, -1.8636e+00,
         -3.9918e-01]], device='cuda:0')
tensor([-11.7951, -15.1229, -10.8157, -18.0735, -10.4513,  -4.0582, -13.2878,
         -7.9149,  -9.9583,  -7.1505,  -8.9219, -16.7028,  -9.9872,  -9.7589,
         -6.0289,  -6.6048,  -9.1966,  -5.1660, -12.8315,  -3.9021],
       device='cuda:0')
tensor([-1.4671,  1.9660, -3.9094,  2.8896, -3.8549, -4.2277, -0.8992, -3.0800,
        -2.8290, -3.5638, -2.4641, -0.9499, -2.7793, -1.2367, -3.4905, -2.1316,
        -1.5573, -4.7511, -2.7622, -4.0781], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.4016, -0.9494, -1.8662, -1.3769, -3.3025, -3.1978],
        [-5.5071, -4.3424, -2.1327, -1.5147, -2.3563, -1.5957],
        [-3.4327, -3.5319, -1.5730, -2.8561, -2.3183, -2.0048],
        [-1.2964, -1.8516, -2.1647, -1.1110, -1.4761, -1.2066],
        [-2.2423, -1.3597, -3.8612, -1.5101, -2.1767, -2.7554],
        [-7.7609, -1.4227, -2.9350, -1.2649, -1.7316, -1.1333],
        [-4.2201, -1.1972, -2.3329, -2.3940, -1.9213, -1.1895],
        [-5.0512, -2.8348, -1.9565, -1.2296, -2.1922, -1.4571],
        [-3.3976, -3.1279, -2.6615, -0.9246, -2.3432, -1.7317],
        [-2.0267, -2.1966, -2.4076, -2.3046, -1.3898, -1.2793],
        [-2.7796, -1.6066, -2.1489, -1.7455, -1.3157, -0.9896],
        [-2.2075, -1.2944, -1.4081, -1.0941, -1.1979, -1.2519],
        [-2.7157, -1.1389, -1.2088, -1.2333, -1.1008, -1.9953],
        [-1.1520, -1.6284, -1.4653, -1.8032, -1.5529, -1.4925],
        [-1.7868, -1.2596, -1.9743, -3.3648, -2.8916, -1.6202],
        [-3.1533, -1.6834, -1.2659, -0.9522, -3.8713, -1.8816],
        [-6.7178, -2.2145, -1.9568, -2.6013, -2.4775, -1.2468],
        [-2.5176, -4.0916, -2.4020, -1.2582, -2.6589, -1.9345],
        [-1.9494, -1.4362, -2.1636, -1.0178, -1.4371, -1.3292],
        [-1.6760, -1.8008, -1.1523, -1.0091, -1.1182, -1.2760]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.0945, -17.4487, -15.7169,  -9.1064, -13.9053, -16.2483, -13.2550,
        -14.7214, -14.1865, -11.6046, -10.5859,  -8.4538,  -9.3928,  -9.0943,
        -12.8972, -12.8077, -17.2146, -14.8627,  -9.3334,  -8.0324],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.8227e+00, -1.1067e-01, -1.7223e+00, -5.0471e-01, -3.8054e+00,
         -3.7229e+00],
        [-3.8443e+00, -3.8630e+00, -2.5723e+00, -6.9868e-01, -3.1510e+00,
         -1.3620e+00],
        [-2.5808e+00, -4.0599e+00, -6.7080e-01, -1.7180e+00, -3.2458e+00,
         -2.3078e+00],
        [-1.1913e+00, -2.0290e+00, -2.2633e+00, -5.2546e-02, -6.6991e-01,
         -1.4914e-01],
        [-2.8163e+00, -6.0665e-02, -5.2142e+00, -5.4959e-01, -2.7137e+00,
         -1.7114e+00],
        [-4.7121e+00, -5.1413e-01, -3.5599e+00, -2.2229e-01, -2.1557e+00,
         -2.4048e-01],
        [-3.7423e+00, -2.0576e-04, -2.6860e+00, -2.1234e+00, -2.3742e+00,
         -5.7582e-02],
        [-3.5446e+00, -3.7071e+00, -2.2242e+00, -5.3292e-01, -2.7957e+00,
         -1.8448e-01],
        [-2.4552e+00, -3.2115e+00, -3.4201e+00, -4.3584e-02, -3.1631e+00,
         -1.6297e+00],
        [-3.0650e+00, -2.9507e+00, -2.5252e+00, -1.9667e+00, -7.4148e-02,
         -8.4865e-02],
        [-2.8129e+00, -1.5553e-01, -2.5570e+00, -1.1288e+00, -1.0591e-01,
         -1.8410e-02],
        [-1.5581e+00, -4.4104e-02, -3.0624e-01, -5.6835e-03, -2.7714e-02,
         -1.0466e-01],
        [-1.6244e+00, -6.5626e-01, -7.6321e-01, -2.9802e-06, -1.8368e-01,
         -1.9949e+00],
        [-1.9462e-02, -6.9450e-01, -2.7332e-01, -9.9743e-01, -7.2614e-01,
         -6.5955e-01],
        [-1.9137e+00, -3.5212e-01, -1.9732e+00, -3.5240e+00, -2.7510e+00,
         -1.2271e+00],
        [-3.9582e+00, -9.0772e-01, -3.6133e-01, -8.2320e-03, -4.2809e+00,
         -2.0340e+00],
        [-4.3170e+00, -1.9762e+00, -2.0811e+00, -2.9923e+00, -3.2548e+00,
         -5.9056e-01],
        [-1.6852e+00, -4.9677e+00, -3.5905e+00, -2.9935e-02, -3.6161e+00,
         -2.0542e+00],
        [-7.5850e-01, -5.0361e-01, -1.9786e+00, -1.4931e-02, -1.0292e-01,
         -2.8412e-01],
        [-1.6223e+00, -2.0644e+00, -5.9920e-02, -6.9441e-03, -2.9187e-02,
         -7.7810e-01]], device='cuda:0')
tensor([-12.6886, -15.4912, -14.5830,  -6.3551, -13.0659, -11.4046, -10.9836,
        -12.9890, -13.9233, -10.6666,  -6.7785,  -2.0464,  -5.2225,  -3.3704,
        -11.7410, -11.5504, -15.2119, -15.9435,  -3.6427,  -4.5609],
       device='cuda:0')
tensor([-0.4059, -1.9575, -1.1338, -2.7514, -0.8395, -4.8437, -2.2714, -1.7324,
        -0.2632, -0.9380, -3.8074, -6.4073, -4.1703, -5.7239, -1.1562, -1.2573,
        -2.0027,  1.0808, -5.6907, -3.4715], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.3089, -1.6654, -2.1920, -2.8209, -1.8944, -1.8535],
        [-4.4385, -2.3287, -1.7283, -1.9076, -2.2722, -1.3194],
        [-3.3910, -4.1630, -3.0404, -1.5475, -2.6341, -2.2386],
        [-1.2737, -1.7217, -2.0877, -2.0608, -1.5156, -1.1728],
        [-3.5952, -1.3730, -2.4196, -1.4085, -1.9332, -0.9048],
        [-3.1247, -1.2431, -2.6205, -1.2082, -1.5015, -1.7215],
        [-1.2611, -1.3899, -1.8686, -1.9093, -2.6340, -1.9037],
        [-2.5808, -2.3110, -1.8248, -1.1293, -2.2916, -1.4516],
        [-2.7374, -2.9600, -2.3332, -1.1754, -2.0668, -1.3539],
        [-1.1810, -1.1817, -2.1991, -1.2666, -1.3818, -1.1626],
        [-4.0783, -1.3769, -2.1065, -1.3262, -1.4734, -1.3356],
        [-4.5382, -1.3784, -2.2200, -1.1925, -1.3380, -1.2117],
        [-4.5908, -1.4148, -1.1262, -1.5902, -1.9391, -1.9999],
        [-2.8629, -2.0450, -2.1979, -1.1003, -2.0480, -1.2986],
        [-5.9988, -2.6212, -2.0919, -2.0308, -2.2438, -1.3640],
        [-1.3096, -1.3027, -1.6926, -1.7100, -1.5336, -1.2118],
        [-2.8819, -1.5962, -4.0544, -1.3135, -2.2015, -2.0851],
        [-2.2067, -1.2374, -2.8512, -1.3075, -1.6426, -1.1812],
        [-2.5703, -1.6111, -1.3556, -2.3808, -2.1583, -1.1525],
        [-6.7224, -2.4563, -1.8631, -2.2692, -2.5006, -2.7128]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.7351, -13.9946, -17.0145,  -9.8323, -11.6342, -11.4195, -10.9667,
        -11.5891, -12.6267,  -8.3726, -11.6968, -11.8788, -12.6609, -11.5526,
        -16.3505,  -8.7602, -14.1325, -10.4266, -11.2286, -18.5244],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-8.3796e-01, -7.3119e-01, -2.3986e+00, -2.9008e+00, -2.2471e+00,
         -2.1474e+00],
        [-2.5998e+00, -2.2489e+00, -1.1241e+00, -1.8121e+00, -2.9399e+00,
         -1.4206e-01],
        [-1.8927e+00, -3.6704e+00, -3.8407e+00, -7.7154e-01, -2.9832e+00,
         -2.5442e+00],
        [-7.5710e-03, -1.9622e+00, -2.2461e+00, -1.1975e+00, -4.3080e-01,
         -2.0961e-02],
        [-3.3142e+00, -9.7247e-03, -2.3739e+00, -3.5288e-01, -2.5910e+00,
         -2.1590e-02],
        [-2.5930e+00, -2.0033e-01, -3.3180e+00, -2.2802e-01, -1.4252e+00,
         -1.8771e+00],
        [-9.9137e-01, -2.4100e-01, -1.9783e+00, -1.3771e+00, -3.0745e+00,
         -1.3575e+00],
        [-1.3859e+00, -2.1290e+00, -1.6546e+00, -2.8641e-01, -3.1013e+00,
         -1.2820e-01],
        [-1.7034e+00, -3.0106e+00, -2.7132e+00, -1.0658e-01, -2.3711e+00,
         -5.4320e-01],
        [-2.9406e-01, -3.3221e-01, -2.2592e+00, -4.4728e-01, -3.2684e-01,
         -9.3051e-02],
        [-4.1600e+00, -1.1481e-03, -2.1664e+00, -1.2071e-01, -8.7845e-01,
         -4.4679e-01],
        [-3.3562e+00, -1.1683e-05, -2.2848e+00, -1.0011e-01, -5.6099e-01,
         -9.5725e-04],
        [-3.9618e+00, -2.9779e-01, -2.0817e-01, -9.7559e-01, -2.3316e+00,
         -2.2865e+00],
        [-1.8058e+00, -1.5461e+00, -3.1375e+00, -2.0011e-01, -2.3500e+00,
         -1.4458e-01],
        [-4.4042e+00, -2.7221e+00, -2.4672e+00, -9.6194e-01, -2.7263e+00,
         -1.4945e-01],
        [-2.9721e-02, -5.5960e-01, -1.3824e+00, -8.2047e-01, -3.3260e-01,
         -6.9340e-02],
        [-3.4061e+00, -4.3554e-01, -5.0897e+00, -1.2695e-01, -2.6746e+00,
         -1.2319e+00],
        [-1.9537e+00, -9.9927e-02, -3.1683e+00, -4.5479e-01, -1.8214e+00,
         -1.0513e-02],
        [-2.2442e+00, -5.8390e-01, -7.5258e-01, -2.3108e+00, -2.9672e+00,
         -1.8559e-01],
        [-4.8448e+00, -2.6961e+00, -2.0107e+00, -2.7390e+00, -3.4222e+00,
         -3.8230e+00]], device='cuda:0')
tensor([-11.2632, -10.8669, -15.7026,  -5.8652,  -8.6633,  -9.6417,  -9.0197,
         -8.6855, -10.4481,  -3.7526,  -7.7735,  -6.3031, -10.0614,  -9.1840,
        -13.4312,  -3.1941, -12.9648,  -7.5086,  -9.0442, -19.5357],
       device='cuda:0')
tensor([-0.4719, -3.1277, -1.3119, -3.9671, -2.9709, -1.7778, -1.9470, -2.9036,
        -2.1786, -4.6200, -3.9233, -5.5758, -2.5995, -2.3686, -2.9194, -5.5661,
        -1.1677, -2.9180, -2.1844,  1.0113], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.0474, -1.0963, -1.7171, -1.1079, -1.3915, -2.9762],
        [-1.1788, -1.5572, -1.3250, -1.2031, -1.7912, -3.3485],
        [-7.4341, -3.2279, -1.6532, -1.0494, -1.7889, -2.2674],
        [-3.1991, -3.2614, -2.7981, -1.8086, -2.5618, -1.2311],
        [-1.7841, -1.9121, -1.6686, -3.1349, -1.3545, -1.3294],
        [-1.9798, -1.2826, -2.5778, -1.3275, -1.9047, -4.4213],
        [-5.5851, -1.3326, -1.8939, -1.3021, -3.1136, -1.0858],
        [-4.1195, -1.4637, -1.5975, -3.7224, -1.5642, -1.8519],
        [-3.3615, -2.3734, -1.6050, -3.2727, -2.4280, -2.2950],
        [-4.9996, -4.1156, -4.9281, -1.0243, -1.5893, -1.6238],
        [-1.2945, -1.1555, -2.1719, -2.4863, -1.0599, -1.1150],
        [-1.9431, -1.5490, -2.9431, -1.7362, -1.3119, -1.3568],
        [-1.8976, -1.3177, -1.5299, -1.2350, -1.4857, -1.4324],
        [-2.2770, -1.2797, -1.1852, -1.2086, -1.3337, -1.3637],
        [-3.4442, -1.9538, -2.1548, -1.8140, -1.8227, -1.1091],
        [-1.0808, -1.3600, -1.5266, -2.9195, -1.3476, -1.2384],
        [-2.9175, -1.7282, -2.5401, -2.1250, -3.4804, -1.8652],
        [-3.9253, -2.3468, -1.3400, -1.2382, -2.5869, -1.8946],
        [-4.0224, -2.7973, -2.2250, -1.7899, -2.1161, -3.2105],
        [-2.6493, -2.2528, -1.6224, -1.5900, -1.9972, -2.3098]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.3364, -10.4038, -17.4209, -14.8600, -11.1836, -13.4937, -14.3132,
        -14.3193, -15.3356, -18.2807,  -9.2830, -10.8402,  -8.8983,  -8.6478,
        -12.2984,  -9.4727, -14.6564, -13.3318, -16.1612, -12.4215],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.1873e+00, -8.7099e-02, -1.1612e+00, -4.3183e-02, -6.0841e-01,
         -3.4351e+00],
        [-8.7333e-01, -7.1956e-01, -8.3977e-01, -1.5310e-02, -1.7808e+00,
         -3.4579e+00],
        [-5.5026e+00, -3.8546e+00, -1.3162e+00, -1.9312e-04, -1.5520e+00,
         -2.8656e+00],
        [-2.1222e+00, -2.8362e+00, -3.1002e+00, -1.3245e+00, -2.8017e+00,
         -4.4679e-01],
        [-7.1748e-01, -2.3873e+00, -1.3953e+00, -2.5380e+00, -4.9850e-02,
         -5.7269e-01],
        [-2.4987e+00, -1.0559e-01, -2.8175e+00, -2.8013e-03, -2.0458e+00,
         -2.7984e+00],
        [-4.1461e+00, -3.6916e-01, -1.5858e+00, -4.7298e-01, -4.6872e+00,
         -2.3694e-01],
        [-3.4714e+00, -3.7927e-01, -1.4674e+00, -3.8979e+00, -1.5066e+00,
         -1.3970e+00],
        [-2.9639e+00, -2.4597e+00, -7.4932e-01, -4.4209e+00, -3.0350e+00,
         -2.7880e+00],
        [-3.4265e+00, -3.9210e+00, -4.1741e+00, -4.6122e-03, -3.2827e-01,
         -9.9156e-01],
        [-1.0114e+00, -5.5722e-01, -2.1076e+00, -1.9793e+00, -1.0227e-02,
         -4.8402e-01],
        [-1.7701e+00, -3.3528e-03, -4.3660e+00, -1.2878e+00, -4.2796e-01,
         -1.0247e+00],
        [-1.1963e+00, -5.0672e-02, -1.0347e+00, -7.1082e-02, -7.3042e-02,
         -1.1952e-01],
        [-1.1699e+00, -6.2809e-01, -1.1023e-01, -3.8697e-01, -4.9694e-01,
         -1.1738e+00],
        [-2.5665e+00, -2.0310e+00, -1.7267e+00, -1.3516e+00, -1.7650e+00,
         -9.1351e-02],
        [-7.8137e-02, -8.0338e-01, -1.0265e+00, -3.0872e+00, -2.5266e-01,
         -3.8817e-02],
        [-3.7971e+00, -8.1381e-01, -3.1349e+00, -2.3517e+00, -3.7344e+00,
         -9.6438e-01],
        [-4.4073e+00, -1.4969e+00, -6.1221e-01, -3.2018e-01, -3.2208e+00,
         -2.3456e+00],
        [-2.9688e+00, -1.9511e+00, -2.3544e+00, -1.4849e+00, -2.1477e+00,
         -2.9532e+00],
        [-2.0293e+00, -2.0837e+00, -4.3328e-01, -2.1216e-01, -1.7228e+00,
         -3.0052e+00]], device='cuda:0')
tensor([ -8.5223,  -7.6866, -15.0912, -12.6316,  -7.6607, -10.2688, -11.4981,
        -12.1196, -16.4168, -12.8461,  -6.1498,  -8.8798,  -2.5453,  -3.9659,
         -9.5322,  -5.2867, -14.7963, -12.4029, -13.8601,  -9.4864],
       device='cuda:0')
tensor([-2.8140, -2.7171, -2.3296, -2.2284, -3.5229, -3.2248, -2.8150, -2.1997,
         1.0812, -5.4346, -3.1332, -1.9603, -6.3529, -4.6820, -2.7662, -4.1860,
         0.1400, -0.9289, -2.3011, -2.9351], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.1217, -1.2830, -1.1478, -1.2164, -1.8995, -3.2112],
        [-4.6870, -2.6905, -2.0183, -1.2500, -2.1981, -1.3499],
        [-2.6496, -4.6324, -2.1938, -1.8638, -2.0578, -1.4493],
        [-1.1461, -1.9171, -1.8163, -1.2740, -1.4091, -1.3472],
        [-3.4705, -1.3536, -4.4557, -1.5637, -2.4770, -1.5224],
        [-2.5886, -1.3203, -1.5395, -1.2085, -1.7444, -1.3498],
        [-2.6488, -1.1810, -2.5633, -1.5937, -2.5725, -1.4718],
        [-3.7155, -2.6122, -2.3789, -1.8265, -2.1294, -2.3358],
        [-4.3807, -2.3288, -2.1687, -0.9337, -2.2102, -1.4649],
        [-1.3655, -1.5022, -2.3074, -0.9592, -1.6843, -1.1144],
        [-5.2887, -1.7084, -3.2802, -1.5029, -1.3391, -2.4938],
        [-4.9292, -1.3428, -1.9317, -1.1415, -1.9359, -1.4193],
        [-2.8777, -1.2134, -1.8551, -1.4053, -1.9431, -1.6629],
        [-1.2958, -1.1556, -1.1971, -1.1693, -2.2404, -1.7874],
        [-9.6493, -2.4871, -1.9227, -1.1607, -2.4699, -1.3629],
        [-3.8682, -2.5961, -2.8924, -1.7545, -2.1982, -1.3917],
        [-1.1374, -1.2595, -2.2422, -1.3231, -1.2791, -1.1424],
        [-3.0625, -1.3459, -3.6515, -1.4753, -1.3962, -0.8683],
        [-3.4779, -1.3874, -1.6144, -1.1830, -1.3826, -1.0339],
        [-1.6583, -1.2251, -1.1744, -1.1687, -1.5517, -1.2917]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.8796, -14.1937, -14.8466,  -8.9097, -14.8430,  -9.7512, -12.0312,
        -14.9982, -13.4870,  -8.9330, -15.6132, -12.7003, -10.9574,  -8.8457,
        -19.0527, -14.7010,  -8.3836, -11.7997, -10.0793,  -8.0699],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.5893e+00, -9.9643e-03, -4.6603e-01, -4.6792e-02, -2.0114e+00,
         -2.0330e+00],
        [-3.3149e+00, -3.1158e+00, -2.5520e+00, -5.5082e-01, -2.8621e+00,
         -2.2876e-04],
        [-1.6409e+00, -4.6880e+00, -2.1981e+00, -1.0551e+00, -2.4304e+00,
         -7.5821e-01],
        [-3.9949e-01, -2.4097e+00, -1.7762e+00, -2.1430e-01, -2.9156e-02,
         -3.1334e-01],
        [-3.7091e+00, -3.6299e-04, -5.8842e+00, -7.0273e-01, -3.6137e+00,
         -7.5212e-01],
        [-2.2510e+00, -7.6419e-01, -5.3517e-01, -2.5215e-01, -2.2063e+00,
         -2.7971e-01],
        [-2.3029e+00, -1.0293e-01, -2.9667e+00, -9.5361e-01, -3.4993e+00,
         -7.3486e-01],
        [-2.8628e+00, -3.1187e+00, -3.9012e+00, -2.0901e+00, -2.5500e+00,
         -3.1966e+00],
        [-3.0834e+00, -1.8991e+00, -1.7270e+00, -8.0651e-03, -2.8239e+00,
         -5.7772e-01],
        [-1.6534e+00, -1.5680e+00, -2.2846e+00, -7.2566e-02, -1.4346e+00,
         -5.0006e-02],
        [-5.1952e+00, -5.2243e-01, -5.0418e+00, -4.6260e-01, -1.7084e-01,
         -1.6926e+00],
        [-3.7576e+00, -2.4889e-02, -1.7252e+00, -1.4721e-02, -2.6233e+00,
         -1.0036e+00],
        [-2.4157e+00, -8.0217e-02, -1.1202e+00, -6.3211e-01, -2.4783e+00,
         -1.4934e+00],
        [-1.1144e+00, -1.3189e-01, -4.5116e-01, -5.4001e-02, -2.6562e+00,
         -1.4418e+00],
        [-6.3803e+00, -2.6359e+00, -2.0997e+00, -3.2355e-01, -3.5340e+00,
         -4.5479e-01],
        [-3.5122e+00, -2.4658e+00, -3.5030e+00, -8.2464e-01, -2.8538e+00,
         -5.6813e-01],
        [-1.1570e-01, -5.6816e-01, -2.4100e+00, -4.8072e-01, -1.1640e-02,
         -7.2177e-02],
        [-3.0381e+00, -2.4725e-02, -4.4774e+00, -5.6934e-01, -2.9175e-03,
         -7.4313e-03],
        [-3.2279e+00, -8.2167e-01, -7.7589e-01, -1.3709e-01, -1.0041e+00,
         -5.4204e-02],
        [-1.3744e+00, -2.4104e-01, -6.5339e-04, -6.9763e-02, -1.4004e+00,
         -4.8880e-01]], device='cuda:0')
tensor([ -7.1564, -12.3959, -12.7708,  -5.1422, -14.6622,  -6.2884, -10.5603,
        -17.7195, -10.1191,  -7.0631, -13.0855,  -9.1493,  -8.2200,  -5.8494,
        -15.4282, -13.7275,  -3.6584,  -8.1199,  -6.0209,  -3.5750],
       device='cuda:0')
tensor([-3.7232, -1.7978, -2.0758, -3.7675, -0.1808, -3.4628, -1.4709,  2.7212,
        -3.3679, -1.8699, -2.5277, -3.5510, -2.7374, -2.9963, -3.6245, -0.9736,
        -4.7252, -3.6798, -4.0584, -4.4949], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.5742, -1.3026, -1.1521, -1.3981, -1.5595, -1.4059],
        [-0.7447, -1.9496, -1.2936, -2.0566, -1.5910, -1.0536],
        [-5.3449, -1.6119, -1.6153, -1.3541, -1.8467, -1.0632],
        [-2.5525, -2.6041, -1.9750, -1.3412, -1.4615, -2.2428],
        [-1.4715, -1.8944, -2.0449, -1.1680, -1.3348, -1.2143],
        [-2.3571, -1.4241, -2.1897, -1.2017, -1.2527, -1.3353],
        [-1.8479, -1.1061, -1.3460, -1.2521, -1.1982, -1.3232],
        [-1.4985, -2.5405, -2.1160, -1.4869, -1.5122, -1.2780],
        [-1.1330, -1.7571, -1.2027, -1.8869, -1.2850, -1.5166],
        [-1.8633, -2.2859, -2.0741, -1.8207, -1.4733, -2.5885],
        [-2.7150, -0.9319, -2.2490, -1.7692, -2.2168, -2.1649],
        [-5.0586, -4.3992, -2.3682, -1.4842, -1.6841, -0.9033],
        [-4.5014, -2.8649, -1.6385, -2.1151, -2.5609, -2.0557],
        [-1.2140, -1.4361, -1.5417, -0.9944, -1.2101, -1.2022],
        [-2.4818, -1.1892, -2.5673, -1.5103, -1.3980, -1.7784],
        [-3.1237, -1.4205, -1.9517, -1.5512, -1.7882, -1.1612],
        [-0.9539, -1.9666, -1.2642, -3.0169, -1.1493, -1.1983],
        [-2.7552, -2.8150, -1.5210, -2.6077, -1.9144, -1.2001],
        [-4.0167, -2.4241, -1.9150, -1.1157, -2.2773, -2.0651],
        [-1.1197, -2.3976, -2.2771, -1.1735, -1.2272, -1.1266]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.3923,  -8.6891, -12.8361, -12.1771,  -9.1279,  -9.7607,  -8.0735,
        -10.4322,  -8.7813, -12.1059, -12.0468, -15.8976, -15.7365,  -7.5986,
        -10.9249, -10.9965,  -9.5491, -12.8134, -13.8140,  -9.3217],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.2837e+00, -3.9223e-01, -2.6728e-02, -5.7784e-01, -1.0722e+00,
         -9.3072e-01],
        [-2.9645e-02, -1.3696e+00, -7.1371e-01, -2.0757e+00, -1.0830e+00,
         -1.2529e-01],
        [-4.1486e+00, -7.4016e-01, -1.4360e+00, -8.5830e-01, -1.5247e+00,
         -1.3173e-03],
        [-1.6048e+00, -2.3343e+00, -1.9435e+00, -3.7054e-01, -7.2044e-01,
         -2.6945e+00],
        [-1.6046e+00, -2.1011e+00, -1.9957e+00, -2.4245e-01, -2.5625e-01,
         -3.9065e-01],
        [-2.6458e+00, -7.6181e-01, -2.4709e+00, -4.1491e-02, -8.6788e-02,
         -1.2440e+00],
        [-1.2333e+00, -7.1866e-02, -5.2698e-01, -2.6709e-01, -5.3775e-04,
         -7.2435e-02],
        [-6.9373e-01, -2.7664e+00, -2.0094e+00, -7.8999e-01, -1.0602e+00,
         -5.0929e-01],
        [-6.9167e-01, -1.8214e+00, -4.6864e-01, -1.3666e+00, -6.1648e-01,
         -8.1640e-01],
        [-1.9051e+00, -2.1350e+00, -2.0850e+00, -1.7773e+00, -8.4245e-01,
         -2.3860e+00],
        [-3.0379e+00, -3.0923e-03, -2.2518e+00, -1.1607e+00, -2.1692e+00,
         -2.5038e+00],
        [-3.5130e+00, -3.6546e+00, -3.0678e+00, -6.6175e-01, -9.9251e-01,
         -4.9644e-03],
        [-2.9802e+00, -2.6422e+00, -9.8178e-01, -1.2707e+00, -3.2647e+00,
         -2.2880e+00],
        [-7.0032e-01, -1.0654e+00, -9.1146e-01, -1.6531e-01, -8.7866e-02,
         -4.2547e-01],
        [-2.5684e+00, -1.4789e-02, -2.6875e+00, -6.8012e-01, -6.3547e-01,
         -1.3438e+00],
        [-2.7699e+00, -2.5965e-02, -1.5491e+00, -1.0492e+00, -1.9409e+00,
         -4.0314e-01],
        [-8.9920e-02, -1.2326e+00, -1.7363e-01, -2.7382e+00, -3.1913e-01,
         -1.6982e-02],
        [-2.0856e+00, -3.2892e+00, -1.0511e+00, -2.8902e+00, -1.7504e+00,
         -1.9453e-02],
        [-2.8825e+00, -2.3550e+00, -2.0576e+00, -5.4526e-02, -2.7172e+00,
         -2.6027e+00],
        [-4.8247e-01, -2.8686e+00, -2.3712e+00, -2.6938e-01, -1.5824e-03,
         -1.3068e-01]], device='cuda:0')
tensor([ -4.2834,  -5.3970,  -8.7091,  -9.6681,  -6.5907,  -7.2508,  -2.1722,
         -7.8290,  -5.7812, -11.1309, -11.1265, -11.8947, -13.4276,  -3.3558,
         -7.9300,  -7.7382,  -4.5704, -11.0860, -12.6696,  -6.1239],
       device='cuda:0')
tensor([-4.1089, -3.2922, -4.1270, -2.5090, -2.5372, -2.5098, -5.9013, -2.6031,
        -3.0001, -0.9751, -0.9203, -4.0030, -2.3089, -4.2428, -2.9949, -3.2583,
        -4.9787, -1.7274, -1.1444, -3.1978], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-7.7259, -1.4419, -2.4402, -1.8551, -1.6270, -1.8953],
        [-1.1481, -2.6941, -4.0558, -2.5741, -1.4411, -1.2150],
        [-6.3019, -2.6068, -2.1535, -2.1245, -2.7227, -2.5938],
        [-5.9656, -3.1549, -2.7609, -1.9773, -1.7790, -1.3417],
        [-1.2146, -2.0338, -2.1332, -2.9081, -1.1295, -1.4011],
        [-2.5997, -1.5376, -1.5757, -1.4496, -1.4005, -1.0022],
        [-7.1002, -1.3967, -1.4965, -1.1852, -1.9658, -2.0929],
        [-1.2854, -2.8450, -1.2663, -3.2265, -2.1471, -1.1417],
        [-3.2294, -3.0989, -2.1539, -4.5732, -1.5074, -1.0873],
        [-3.7699, -1.0748, -3.3247, -3.7771, -1.5769, -2.2079],
        [-3.3143, -1.0879, -1.3441, -0.9564, -2.1992, -2.7963],
        [-2.0229, -2.1704, -1.5534, -1.2577, -2.3536, -2.1216],
        [-2.5930, -2.7536, -2.2678, -1.1933, -3.4673, -1.7677],
        [-1.1415, -1.1907, -1.1626, -1.0514, -1.6554, -1.4030],
        [-4.4796, -1.2185, -3.0758, -1.1960, -1.5951, -0.9450],
        [-3.6368, -1.3935, -3.1832, -1.2660, -1.5075, -1.5484],
        [-8.9081, -1.4312, -3.9768, -1.2447, -1.4270, -1.1777],
        [-4.2018, -1.2679, -1.4834, -1.2839, -2.0086, -1.2573],
        [-3.9895, -1.9347, -2.3722, -1.1421, -2.1262, -1.3070],
        [-2.4650, -2.7948, -3.2720, -1.1236, -2.2393, -1.3282]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-16.9855, -13.1281, -18.5033, -16.9793, -10.8204,  -9.5653, -15.2372,
        -11.9120, -15.6501, -15.7313, -11.6983, -11.4797, -14.0427,  -7.6046,
        -12.5100, -12.5354, -18.1655, -11.5028, -12.8716, -13.2229],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-5.4176e+00, -4.6253e-05, -2.6840e+00, -1.7340e+00, -1.6481e+00,
         -1.6275e+00],
        [-4.9741e-01, -2.2507e+00, -4.8734e+00, -2.4789e+00, -1.2252e+00,
         -1.0498e-02],
        [-4.6265e+00, -3.1879e+00, -3.1877e+00, -2.5347e+00, -3.8675e+00,
         -3.6272e+00],
        [-3.6061e+00, -2.8203e+00, -2.5374e+00, -1.4070e+00, -1.2096e+00,
         -1.7816e-03],
        [-2.3327e-01, -2.7315e+00, -2.0234e+00, -2.2662e+00, -9.9820e-02,
         -1.1242e+00],
        [-2.3827e+00, -9.9736e-01, -6.6716e-01, -5.9113e-01, -7.5602e-01,
         -8.4598e-03],
        [-4.0425e+00, -3.6608e-01, -3.0276e-01, -5.7154e-02, -1.2245e+00,
         -1.6383e+00],
        [-1.6409e-01, -4.2421e+00, -4.5229e-01, -3.0410e+00, -2.3239e+00,
         -1.4032e-01],
        [-3.8723e+00, -4.4259e+00, -2.2396e+00, -3.8655e+00, -8.2061e-01,
         -6.1579e-03],
        [-3.8768e+00, -2.3009e-02, -3.9466e+00, -4.4496e+00, -1.1947e+00,
         -2.3325e+00],
        [-4.1004e+00, -6.5105e-01, -3.6315e-01, -1.4402e-02, -2.2868e+00,
         -3.1902e+00],
        [-7.8975e-01, -1.6143e+00, -1.2237e+00, -2.5749e-01, -3.1729e+00,
         -2.2538e+00],
        [-1.4509e+00, -3.0006e+00, -2.5607e+00, -8.6058e-02, -6.3118e+00,
         -1.5501e+00],
        [-2.5727e-01, -1.8763e-02, -2.2354e-01, -2.8782e-01, -9.3234e-01,
         -5.0198e-01],
        [-4.8987e+00, -7.0029e-02, -3.4481e+00, -2.0826e-04, -1.0923e+00,
         -2.7053e-01],
        [-3.0128e+00, -4.4697e-02, -3.9370e+00, -3.6637e-02, -1.2041e+00,
         -1.2736e+00],
        [-6.3850e+00, -1.1746e+00, -4.8955e+00, -1.9833e-01, -9.4886e-01,
         -5.5683e-02],
        [-3.7322e+00, -8.2425e-02, -1.3172e+00, -2.7232e-01, -2.5222e+00,
         -5.2431e-01],
        [-2.7679e+00, -1.0348e+00, -3.5977e+00, -5.5003e-04, -2.6513e+00,
         -8.6176e-02],
        [-1.3800e+00, -2.6650e+00, -4.2921e+00, -9.6910e-02, -2.9823e+00,
         -1.3497e-01]], device='cuda:0')
tensor([-13.1112, -11.3361, -21.0315, -11.5822,  -8.4785,  -5.4028,  -7.6313,
        -10.3637, -15.2301, -15.8232, -10.6059,  -9.3119, -14.9601,  -2.2217,
         -9.7799,  -9.5088, -13.6580,  -8.4507, -10.1385, -11.5513],
       device='cuda:0')
tensor([-3.8743, -1.7920,  2.5282, -5.3971, -2.3419, -4.1625, -7.6059, -1.5483,
        -0.4200,  0.0919, -1.0923, -2.1678,  0.9175, -5.3829, -2.7301, -3.0266,
        -4.5075, -3.0521, -2.7331, -1.6716], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.2058, -1.0934, -2.9371, -3.5938, -1.0715, -1.0489],
        [-5.1755, -1.5601, -3.4841, -1.4004, -1.3277, -1.6078],
        [-6.3858, -1.5512, -1.5151, -1.4481, -1.7140, -1.6375],
        [-1.3967, -1.2801, -1.0893, -0.9101, -1.3493, -1.1020],
        [-1.8791, -1.1940, -1.1312, -1.0131, -2.0913, -0.9051],
        [-3.5369, -2.0544, -1.9519, -1.6624, -1.3117, -1.2147],
        [-0.9782, -1.5393, -1.6510, -1.6832, -1.3707, -1.5883],
        [-3.2034, -1.7038, -3.3685, -2.1958, -1.4494, -1.4702],
        [-1.9274, -1.1903, -1.7083, -1.2181, -2.5285, -1.6190],
        [-4.3486, -3.6708, -2.5668, -1.5380, -2.1750, -1.5216],
        [-4.4050, -2.5613, -2.1504, -2.2254, -2.0067, -1.8971],
        [-1.4407, -1.1658, -1.3717, -1.1654, -1.8729, -1.3542],
        [-3.0870, -1.3856, -2.1596, -1.3756, -1.5072, -3.0662],
        [-2.9141, -1.6650, -2.3566, -1.8125, -1.5825, -1.1532],
        [-0.9651, -2.2269, -1.1884, -2.1523, -2.0292, -1.4567],
        [-3.5035, -3.7211, -1.8919, -2.4238, -2.7374, -1.2838],
        [-3.4211, -4.7641, -1.9152, -1.3071, -1.9839, -1.4203],
        [-1.4198, -1.2165, -1.6852, -1.1208, -1.5392, -1.2072],
        [-1.5068, -1.3114, -3.3372, -1.3239, -1.6343, -0.9970],
        [-4.1358, -1.4631, -1.6185, -1.2794, -1.5316, -1.3263]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.9505, -14.5556, -14.2517,  -7.1275,  -8.2138, -11.7319,  -8.8108,
        -13.3909, -10.1916, -15.8208, -15.2460,  -8.3707, -12.5813, -11.4838,
        -10.0186, -15.5616, -14.8118,  -8.1887, -10.1105, -11.3546],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.4755e-01, -2.9896e-01, -3.2938e+00, -3.0175e+00, -4.3671e-03,
         -2.3978e-01],
        [-4.3534e+00, -5.0979e-01, -5.4112e+00, -5.5571e-01, -6.4709e-01,
         -1.5181e+00],
        [-4.0769e+00, -3.2379e-01, -1.0129e+00, -6.2026e-01, -6.4717e-01,
         -6.6501e-01],
        [-3.1162e-01, -7.5337e-01, -3.4171e-03, -1.3512e-02, -5.1194e-01,
         -3.8225e-02],
        [-7.9122e-01, -3.3460e-02, -1.2757e-03, -1.1954e-02, -2.1535e+00,
         -3.1894e-03],
        [-2.7743e+00, -2.4875e+00, -1.9545e+00, -1.2866e+00, -3.2549e-01,
         -1.4987e-01],
        [-3.0917e-02, -1.3020e+00, -1.5879e+00, -1.2855e+00, -4.6168e-01,
         -1.0449e+00],
        [-3.3161e+00, -8.4004e-01, -4.2710e+00, -2.6604e+00, -8.6908e-01,
         -7.0404e-01],
        [-1.9460e+00, -9.3970e-01, -1.4706e+00, -3.0504e-01, -2.9746e+00,
         -1.6522e+00],
        [-2.8149e+00, -3.3439e+00, -3.4246e+00, -6.6874e-01, -2.7238e+00,
         -1.3036e+00],
        [-3.4845e+00, -2.5793e+00, -2.3226e+00, -1.0651e+00, -2.0695e+00,
         -2.0163e+00],
        [-1.8002e+00, -1.2437e-02, -4.4483e-01, -7.6582e-01, -1.9013e+00,
         -4.0887e-01],
        [-3.8238e+00, -4.1536e-02, -2.2295e+00, -1.6764e-01, -7.3474e-01,
         -1.9050e+00],
        [-2.7422e+00, -4.0264e-01, -2.7073e+00, -1.4815e+00, -1.5299e+00,
         -4.1255e-01],
        [-2.1541e-01, -1.7278e+00, -4.1003e-01, -2.1093e+00, -2.6939e+00,
         -9.9369e-01],
        [-2.2001e+00, -5.0908e+00, -2.2423e+00, -2.5920e+00, -3.7450e+00,
         -3.0037e-01],
        [-2.6216e+00, -5.4404e+00, -1.8725e+00, -1.6225e-01, -1.9039e+00,
         -8.0277e-01],
        [-1.6407e+00, -4.4522e-01, -1.3486e+00, -1.2683e-01, -8.8385e-01,
         -2.4419e-01],
        [-1.5105e+00, -2.2871e-01, -4.5499e+00, -1.1451e-01, -1.2872e+00,
         -3.3947e-01],
        [-3.2125e+00, -1.0220e-01, -1.0557e+00, -2.2819e-01, -1.3760e+00,
         -6.5687e-01]], device='cuda:0')
tensor([ -7.0019, -12.9952,  -7.3460,  -1.6321,  -2.9946,  -8.9783,  -5.7129,
        -12.6608,  -9.2882, -14.2794, -13.5373,  -5.3335,  -8.9022,  -9.2761,
         -8.1502, -16.1706, -12.8035,  -4.6894,  -8.0303,  -6.6315],
       device='cuda:0')
tensor([-3.9485, -1.5604, -6.9057, -5.4954, -5.2192, -2.7536, -3.0979, -0.7302,
        -0.9034, -1.5413, -1.7087, -3.0372, -3.6791, -2.2077, -1.8685,  0.6090,
        -2.0083, -3.4994, -2.0802, -4.7232], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.2068, -1.1718, -1.3635, -1.2466, -1.6196, -1.5864],
        [-1.9370, -1.1746, -4.0136, -1.4288, -1.8936, -2.5324],
        [-4.1541, -1.4535, -1.5399, -1.4496, -3.0019, -1.6440],
        [-2.4982, -1.5272, -1.5640, -3.2033, -1.4861, -1.3674],
        [-4.8695, -1.7754, -1.6874, -3.7933, -2.8969, -2.0341],
        [-7.2859, -3.5050, -4.3082, -1.0583, -2.3643, -1.3726],
        [-1.3111, -1.1797, -1.4111, -2.1239, -1.1129, -1.1155],
        [-1.7752, -1.5097, -2.3555, -1.7233, -1.2201, -1.4450],
        [-2.0019, -1.2418, -1.6415, -1.2301, -2.0689, -1.5259],
        [-1.3083, -1.0050, -0.9773, -1.1148, -1.3414, -0.9953],
        [-1.0171, -1.6366, -1.5465, -1.3482, -1.3637, -1.4884],
        [-3.6137, -1.1030, -1.2036, -1.2203, -1.4093, -2.1817],
        [-3.1700, -1.3155, -1.7018, -1.4894, -1.3427, -1.4780],
        [-0.9370, -1.3835, -1.5703, -3.2442, -1.4666, -1.1557],
        [-3.4189, -2.0088, -2.2594, -2.5823, -3.6608, -3.6660],
        [-2.3370, -0.9771, -3.2592, -2.5762, -2.7232, -3.2343],
        [-7.1249, -2.1062, -2.0496, -1.5498, -2.1233, -4.2992],
        [-6.0351, -3.6438, -2.4158, -1.3740, -1.8142, -1.9884],
        [-1.2878, -1.2835, -1.8898, -1.0784, -1.3867, -1.0659],
        [-2.5713, -1.3402, -2.3932, -1.4731, -2.1026, -0.8915]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.1946, -12.9800, -13.2431, -11.6461, -17.0567, -19.8942,  -8.2543,
        -10.0288,  -9.7101,  -6.7421,  -8.4004, -10.7316, -10.4974,  -9.7572,
        -17.5962, -15.1069, -19.2531, -17.2712,  -7.9920, -10.7720],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-6.4108e-01, -1.3400e-01, -3.4516e-01, -9.8038e-01, -1.1864e+00,
         -1.2851e+00],
        [-2.2960e+00, -2.0188e-01, -4.6818e+00, -4.8490e-01, -1.9269e+00,
         -1.9252e+00],
        [-3.6520e+00, -1.5332e-01, -4.3535e-01, -7.0155e-01, -5.0079e+00,
         -1.2419e+00],
        [-1.9799e+00, -3.9875e-01, -1.4183e+00, -3.3654e+00, -1.1891e+00,
         -3.2722e-01],
        [-4.1292e+00, -9.0077e-01, -1.4126e+00, -4.6728e+00, -3.8285e+00,
         -2.0699e+00],
        [-4.8092e+00, -3.2055e+00, -3.6169e+00, -9.1581e-03, -2.9948e+00,
         -9.7435e-02],
        [-1.2118e+00, -4.5723e-01, -1.7630e-01, -1.5588e+00, -3.3531e-01,
         -4.8141e-01],
        [-1.4374e+00, -6.4688e-02, -2.8791e+00, -1.2049e+00, -9.3823e-02,
         -1.0255e+00],
        [-1.3764e+00, -5.1939e-03, -1.1660e+00, -6.4025e-02, -1.7549e+00,
         -4.0845e-01],
        [-2.5737e-01, -1.4727e-02, -6.8383e-02, -1.6648e-01, -6.2829e-01,
         -4.2038e-01],
        [-8.2921e-01, -2.2721e-03, -2.7096e-02, -6.2585e-01, -4.1469e-02,
         -5.2806e-01],
        [-3.2079e+00, -1.6288e-01, -1.8872e-02, -4.5400e-01, -6.4266e-01,
         -2.0554e+00],
        [-1.9464e+00, -1.2286e+00, -1.6326e+00, -8.9854e-01, -5.7294e-01,
         -5.3100e-01],
        [-7.6795e-02, -9.4871e-01, -1.3175e+00, -3.4354e+00, -7.5959e-01,
         -2.0074e-02],
        [-4.0586e+00, -1.7329e+00, -2.6707e+00, -2.9535e+00, -4.2828e+00,
         -3.5270e+00],
        [-3.0083e+00, -7.4727e-02, -4.1965e+00, -2.0255e+00, -3.0181e+00,
         -3.8398e+00],
        [-5.0276e+00, -1.0084e+00, -2.3676e+00, -8.8377e-01, -2.5834e+00,
         -4.1592e+00],
        [-3.4722e+00, -3.2973e+00, -1.9067e+00, -2.1545e-01, -1.1938e+00,
         -2.0163e+00],
        [-6.7106e-02, -6.8453e-01, -1.6188e+00, -8.4297e-01, -5.3143e-01,
         -8.9648e-03],
        [-3.0576e+00, -2.4554e-02, -2.6835e+00, -5.4702e-01, -2.3847e+00,
         -1.2846e-02]], device='cuda:0')
tensor([ -4.5721, -11.5168, -11.1921,  -8.6787, -17.0138, -14.7329,  -4.2209,
         -6.7054,  -4.7750,  -1.5556,  -2.0540,  -6.5417,  -6.8101,  -6.5580,
        -19.2254, -16.1630, -16.0300, -12.1017,  -3.7538,  -8.7102],
       device='cuda:0')
tensor([-3.6224, -1.4632, -2.0510, -2.9674, -0.0429, -5.1613, -4.0334, -3.3233,
        -4.9351, -5.1865, -6.3465, -4.1899, -3.6873, -3.1992,  1.6292,  1.0560,
        -3.2231, -5.1695, -4.2382, -2.0618], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.7874, -4.6081, -1.7556, -1.4195, -1.8811, -1.4262],
        [-1.1708, -1.4791, -1.6986, -2.0091, -1.7533, -1.1544],
        [-2.2729, -1.4056, -2.1253, -1.3129, -1.6646, -0.9049],
        [-2.5742, -1.1912, -3.3518, -1.1971, -1.3925, -1.3174],
        [-1.7214, -1.3880, -1.6199, -1.2896, -1.4824, -2.5481],
        [-1.8894, -1.3044, -1.2321, -1.0249, -2.2639, -1.8711],
        [-5.5006, -2.6433, -1.8217, -1.1435, -1.9889, -1.4288],
        [-2.3575, -3.2549, -2.9144, -1.8029, -2.1123, -2.1298],
        [-1.1334, -1.2821, -1.8465, -1.0710, -1.3780, -1.3117],
        [-5.3685, -1.1998, -3.3246, -1.2917, -1.4133, -1.7337],
        [-3.9555, -1.5043, -2.9740, -1.2687, -1.5463, -1.8836],
        [-2.6188, -1.2160, -1.4542, -1.9836, -1.8774, -1.5659],
        [-4.2695, -2.3205, -1.9559, -1.2599, -2.2950, -1.3644],
        [-5.6047, -3.2097, -1.9768, -1.0473, -2.3935, -1.7634],
        [-1.2433, -1.2323, -2.8584, -2.2399, -1.5656, -1.2704],
        [-3.2741, -1.4152, -2.3191, -1.4704, -1.3092, -0.9671],
        [-1.1577, -1.2466, -2.4367, -1.5024, -1.3549, -1.2511],
        [-3.1075, -1.3725, -1.1867, -1.8079, -1.6332, -1.6118],
        [-4.4972, -3.1567, -2.4345, -1.1420, -1.6801, -1.7327],
        [-3.3507, -2.8492, -1.8035, -1.3780, -2.1835, -1.5123]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-15.8780,  -9.2652,  -9.6862, -11.0242, -10.0492,  -9.5859, -14.5267,
        -14.5717,  -8.0227, -14.3315, -13.1325, -10.7158, -13.4651, -15.9954,
        -10.4099, -10.7551,  -8.9493, -10.7197, -14.6432, -13.0773],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.2705e+00, -4.0128e+00, -1.2360e+00, -7.4395e-01, -1.7221e+00,
         -7.3266e-01],
        [-1.5220e-01, -1.2221e+00, -1.4406e+00, -1.3825e+00, -1.3885e+00,
         -2.7231e-02],
        [-2.4608e+00, -2.9993e-04, -2.2895e+00, -8.3351e-02, -1.5565e+00,
         -1.8392e-02],
        [-2.2665e+00, -9.0805e-02, -3.8239e+00, -1.9992e-02, -4.1092e-01,
         -8.1963e-02],
        [-1.5699e+00, -1.7024e-01, -8.8440e-01, -3.0264e-01, -1.2051e+00,
         -2.8072e+00],
        [-2.1068e+00, -1.8765e-01, -4.6526e-01, -1.7621e-02, -2.6916e+00,
         -1.7867e+00],
        [-3.8853e+00, -2.9668e+00, -1.7303e+00, -3.8650e-01, -2.1703e+00,
         -2.2392e-01],
        [-9.9742e-01, -2.8738e+00, -3.3217e+00, -1.1292e+00, -2.2015e+00,
         -2.3013e+00],
        [-2.4251e-01, -4.7211e-01, -1.7628e+00, -4.5037e-02, -1.6310e-02,
         -5.4549e-01],
        [-4.9676e+00, -1.4561e-01, -3.7870e+00, -1.1495e-01, -5.6608e-01,
         -1.1315e+00],
        [-3.3562e+00, -1.4013e-01, -4.0757e+00, -2.0373e-01, -1.3053e+00,
         -1.5305e+00],
        [-2.3343e+00, -2.9078e-02, -1.0536e+00, -1.6121e+00, -2.2269e+00,
         -1.4061e+00],
        [-2.9096e+00, -2.2187e+00, -2.4317e+00, -5.6719e-01, -3.0692e+00,
         -1.1228e-01],
        [-3.7973e+00, -3.1145e+00, -1.7515e+00, -5.6122e-02, -3.0397e+00,
         -1.6147e+00],
        [-8.0059e-01, -4.1439e-01, -3.2611e+00, -1.7566e+00, -8.0121e-01,
         -3.1224e-01],
        [-3.3488e+00, -2.7994e-03, -2.8663e+00, -5.0483e-01, -3.8784e-01,
         -8.6402e-02],
        [-7.2417e-01, -7.6008e-03, -2.7252e+00, -8.2111e-01, -6.1345e-01,
         -5.6767e-02],
        [-2.9905e+00, -2.6299e-01, -5.2578e-02, -1.4428e+00, -1.5581e+00,
         -1.3732e+00],
        [-3.5100e+00, -3.7199e+00, -3.2099e+00, -3.5165e-01, -1.2559e+00,
         -1.4739e+00],
        [-1.9125e+00, -3.4718e+00, -1.3006e+00, -3.5764e-01, -2.6164e+00,
         -7.5859e-01]], device='cuda:0')
tensor([-11.7179,  -5.6131,  -6.4089,  -6.6941,  -6.9395,  -7.2557, -11.3632,
        -12.8250,  -3.0843, -10.7127, -10.6115,  -8.6621, -11.3088, -13.3739,
         -7.3462,  -7.1970,  -4.9483,  -7.6801, -13.5213, -10.4176],
       device='cuda:0')
tensor([-4.1601, -3.6521, -3.2773, -4.3301, -3.1098, -2.3302, -3.1635, -1.7468,
        -4.9384, -3.6188, -2.5209, -2.0537, -2.1564, -2.6214, -3.0637, -3.5581,
        -4.0010, -3.0396, -1.1219, -2.6597], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.9140, -1.1477, -1.1346, -1.0865, -1.3491, -1.4225],
        [-1.2827, -1.4480, -1.2501, -1.0895, -1.5370, -1.7746],
        [-7.1535, -2.3772, -2.0546, -1.0145, -2.1388, -2.0037],
        [-3.1286, -4.0769, -1.9229, -1.1115, -3.0923, -1.5231],
        [-2.4237, -1.2840, -1.5467, -3.0905, -1.4768, -1.2038],
        [-1.9022, -2.3661, -1.2440, -1.2160, -2.0921, -1.7268],
        [-3.1779, -1.5916, -1.7767, -1.3893, -1.9398, -1.5736],
        [-2.2269, -1.5964, -2.7601, -5.4974, -2.2268, -1.4819],
        [-3.8035, -1.9728, -2.1316, -4.2367, -2.1761, -2.0282],
        [-5.1788, -2.5141, -2.4893, -0.9889, -1.6640, -1.7176],
        [-1.3686, -1.1487, -1.9241, -2.4233, -1.1769, -1.4155],
        [-1.6604, -1.4919, -2.8901, -1.4173, -1.2871, -1.0806],
        [-4.0157, -1.4348, -1.4860, -1.5983, -1.5469, -1.9362],
        [-1.6591, -1.5765, -1.0139, -1.1011, -1.3446, -1.3454],
        [-2.9104, -1.3053, -1.3317, -1.7994, -1.4100, -1.1105],
        [-0.9650, -1.8290, -1.3371, -1.8387, -1.2816, -1.1022],
        [-4.2234, -1.6187, -2.2171, -2.3788, -2.4252, -2.1468],
        [-2.6770, -1.0479, -1.3810, -1.3132, -1.9165, -2.2744],
        [-8.4232, -2.3228, -3.0361, -1.5719, -2.3905, -1.5030],
        [-3.0386, -2.4182, -2.1115, -2.2175, -2.5149, -2.6686]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.0543,  -8.3819, -16.7423, -14.8553, -11.0254, -10.5471, -11.4490,
        -15.7895, -16.3489, -14.5528,  -9.4570,  -9.8275, -12.0177,  -8.0406,
         -9.8674,  -8.3536, -15.0101, -10.6101, -19.2474, -14.9693],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.9866e+00, -7.3756e-02, -5.3560e-02, -1.2784e-01, -5.6130e-01,
         -1.0122e+00],
        [-1.0528e+00, -3.7510e-01, -2.2450e-01, -9.9423e-02, -9.0047e-01,
         -1.6915e+00],
        [-5.4920e+00, -2.3604e+00, -2.3140e+00, -8.1265e-02, -2.4080e+00,
         -2.0621e+00],
        [-2.1024e+00, -3.5948e+00, -1.5843e+00, -3.2069e-01, -3.5847e+00,
         -1.2118e+00],
        [-1.5806e+00, -2.5117e-04, -1.1265e+00, -2.5573e+00, -3.7211e-01,
         -2.1896e-01],
        [-2.1096e+00, -3.1509e+00, -2.5137e-01, -1.2692e-01, -2.2148e+00,
         -1.0204e+00],
        [-2.5698e+00, -2.8843e-01, -1.0004e+00, -2.7336e-01, -2.5012e+00,
         -1.1348e+00],
        [-2.0373e+00, -5.3558e-01, -4.1983e+00, -4.9686e+00, -2.8609e+00,
         -2.6166e-01],
        [-3.1979e+00, -1.4611e+00, -2.5040e+00, -5.2173e+00, -2.4798e+00,
         -2.0106e+00],
        [-3.4744e+00, -2.0127e+00, -1.9382e+00, -3.3976e-03, -5.3346e-01,
         -1.2907e+00],
        [-1.5940e+00, -1.3120e-01, -1.6898e+00, -1.7427e+00, -2.6503e-02,
         -1.0804e+00],
        [-1.3989e+00, -1.4993e-01, -4.2791e+00, -4.6385e-01, -5.1813e-02,
         -1.7227e-01],
        [-2.3478e+00, -1.1784e-02, -1.0310e+00, -1.0899e+00, -2.8987e-01,
         -1.4670e+00],
        [-7.1910e-01, -1.5171e+00, -8.6215e-02, -6.8976e-02, -4.8091e-01,
         -7.1380e-01],
        [-2.3100e+00, -6.7372e-01, -7.2603e-01, -1.2052e+00, -8.3608e-01,
         -1.4704e-01],
        [-1.8057e-02, -1.9181e+00, -7.2450e-01, -1.4697e+00, -4.1176e-01,
         -8.2142e-02],
        [-4.5035e+00, -7.7476e-01, -2.5519e+00, -2.9361e+00, -2.6425e+00,
         -1.7416e+00],
        [-3.0714e+00, -5.3350e-01, -4.5542e-01, -3.9915e-01, -1.9754e+00,
         -2.7366e+00],
        [-5.4436e+00, -1.5421e+00, -3.9945e+00, -7.8768e-01, -3.1932e+00,
         -1.2689e+00],
        [-2.0159e+00, -2.3352e+00, -2.3192e+00, -1.0257e+00, -3.9422e+00,
         -3.9194e+00]], device='cuda:0')
tensor([ -3.8153,  -4.3437, -14.7178, -12.3986,  -5.8557,  -8.8739,  -7.7680,
        -14.8623, -16.8707,  -9.2528,  -6.2647,  -6.5159,  -6.2373,  -3.5861,
         -5.8981,  -4.6242, -15.1504,  -9.1714, -16.2300, -15.5576],
       device='cuda:0')
tensor([-4.2390, -4.0382, -2.0245, -2.4567, -5.1697, -1.6732, -3.6810, -0.9272,
         0.5218, -5.3001, -3.1923, -3.3116, -5.7804, -4.4545, -3.9693, -3.7294,
         0.1404, -1.4386, -3.0174,  0.5883], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[ -5.4362,  -4.4747,  -2.6992,  -3.5375,  -1.7729,  -1.8255],
        [ -1.2424,  -5.4904,  -5.3111,  -1.8033,  -1.4784,  -1.0800],
        [ -0.7077,  -2.3334,  -1.1716,  -1.3160,  -1.9836,  -1.8981],
        [ -3.4496,  -1.7080,  -1.8452,  -1.4169,  -1.6159,  -0.9980],
        [ -4.4827,  -1.5445,  -1.1752,  -0.9218,  -3.9466,  -3.5606],
        [ -1.5757,  -1.2383,  -1.5754,  -1.1186,  -1.8699,  -2.0024],
        [-11.0240,  -2.6416,  -1.5650,  -1.0987,  -1.8245,  -1.4057],
        [ -5.0726,  -3.2048,  -2.2565,  -1.0626,  -3.3450,  -2.0717],
        [ -1.2009,  -1.2870,  -2.7343,  -1.4134,  -1.3962,  -1.6794],
        [ -2.8216,  -1.3526,  -2.6883,  -1.2687,  -1.6650,  -0.8868],
        [ -2.7041,  -1.2522,  -2.4047,  -1.4120,  -1.6762,  -1.3616],
        [ -2.7609,  -1.6760,  -1.3233,  -1.9893,  -1.9336,  -1.2855],
        [ -5.9379,  -2.2985,  -1.7010,  -1.2212,  -1.8520,  -1.3975],
        [ -4.6962,  -3.5278,  -2.3908,  -1.1494,  -1.8662,  -1.4134],
        [ -1.5138,  -1.2840,  -1.4369,  -1.2418,  -1.7481,  -1.2415],
        [ -2.0280,  -1.6857,  -3.7024,  -1.3625,  -1.8208,  -0.9864],
        [ -4.8368,  -1.1060,  -2.5733,  -1.2347,  -1.7015,  -1.8952],
        [ -3.9667,  -1.4836,  -1.2434,  -2.5422,  -1.7213,  -1.8187],
        [ -5.2443,  -2.2304,  -1.7689,  -1.3240,  -2.0265,  -1.2933],
        [ -2.8735,  -2.7289,  -2.1768,  -1.4640,  -2.7666,  -1.4546]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-19.7460, -16.4057,  -9.4104, -11.0335, -15.6312,  -9.3803, -19.5595,
        -17.0131,  -9.7113, -10.6830, -10.8107, -10.9687, -14.4081, -15.0437,
         -8.4662, -11.5858, -13.3475, -12.7759, -13.8875, -13.4645],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.8144e+00, -4.6650e+00, -3.5221e+00, -2.8961e+00, -1.1614e+00,
         -1.8692e+00],
        [-1.6519e-02, -6.1000e+00, -5.3459e+00, -1.4043e+00, -6.8306e-01,
         -1.9350e-03],
        [-4.5120e-01, -4.0065e+00, -3.4955e-01, -1.1757e-02, -2.6667e+00,
         -1.3738e+00],
        [-2.8620e+00, -1.6483e-01, -1.0150e+00, -4.1853e-01, -1.1134e-02,
         -3.1599e-02],
        [-5.3200e+00, -1.5989e+00, -1.6602e-01, -3.4128e-02, -4.4418e+00,
         -3.6893e+00],
        [-1.6136e+00, -2.6428e-03, -1.3413e+00, -1.0006e-01, -1.9846e+00,
         -1.7993e+00],
        [-7.1653e+00, -2.7438e+00, -5.8457e-01, -1.7756e-01, -1.4330e+00,
         -2.4692e-02],
        [-3.9717e+00, -3.0582e+00, -2.1620e+00, -1.3035e-01, -4.5033e+00,
         -2.2103e+00],
        [-5.9762e-02, -5.8251e-01, -3.1954e+00, -5.5903e-01, -2.0392e-03,
         -1.3595e+00],
        [-2.9723e+00, -7.4480e-03, -3.3535e+00, -1.0405e-02, -1.6607e+00,
         -9.2322e-02],
        [-2.2252e+00, -3.1269e-04, -2.6133e+00, -6.5388e-01, -1.9256e+00,
         -5.6600e-01],
        [-2.8713e+00, -7.2337e-01, -5.2711e-01, -1.6853e+00, -2.1615e+00,
         -5.3335e-01],
        [-4.2938e+00, -2.1295e+00, -1.2840e+00, -5.2902e-01, -1.7176e+00,
         -1.7491e-01],
        [-3.4136e+00, -3.9470e+00, -3.1800e+00, -5.7970e-02, -1.5658e+00,
         -6.6808e-01],
        [-1.9914e+00, -6.1896e-01, -7.8747e-01, -3.5623e-01, -1.4988e+00,
         -2.6336e-01],
        [-2.1337e+00, -5.3230e-01, -5.6906e+00, -1.4863e-01, -1.8495e+00,
         -5.1168e-03],
        [-4.2913e+00, -2.1958e-01, -2.9241e+00, -1.6517e-01, -1.9611e+00,
         -2.3478e+00],
        [-3.4882e+00, -6.5354e-02, -7.0704e-02, -2.7347e+00, -1.8823e+00,
         -1.3282e+00],
        [-3.9823e+00, -1.9787e+00, -1.7200e+00, -7.9507e-01, -2.2662e+00,
         -1.3071e-01],
        [-1.5619e+00, -2.8885e+00, -2.5069e+00, -3.2942e-01, -4.3749e+00,
         -3.7123e-01]], device='cuda:0')
tensor([-18.9282, -13.5517,  -8.8596,  -4.5031, -15.2502,  -6.8414, -12.1289,
        -16.0357,  -5.7583,  -8.0967,  -7.9843,  -8.5020, -10.1288, -12.8325,
         -5.5162, -10.3598, -11.9092,  -9.5695, -10.8730, -12.0329],
       device='cuda:0')
tensor([-0.8178, -2.8540, -0.5508, -6.5304, -0.3810, -2.5389, -7.4306, -0.9774,
        -3.9531, -2.5863, -2.8263, -2.4667, -4.2792, -2.2113, -2.9500, -1.2260,
        -1.4384, -3.2064, -3.0145, -1.4316], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.0825, -1.5944, -1.5807, -0.8708, -1.1007, -1.0688],
        [-2.4911, -1.6266, -5.2932, -1.5778, -1.7545, -2.1528],
        [-2.5528, -1.1597, -1.6218, -2.0223, -1.8028, -1.2468],
        [-1.8288, -1.5306, -1.9858, -2.7203, -2.6591, -1.1980],
        [-2.9526, -2.3434, -1.9225, -1.6482, -2.1013, -1.5539],
        [-3.1847, -2.0267, -2.1787, -3.8233, -2.2978, -1.4082],
        [-1.1895, -1.7311, -2.3077, -0.8297, -0.9821, -0.9743],
        [-2.4529, -1.3210, -2.9546, -1.2325, -2.0204, -2.0289],
        [-1.3650, -1.1749, -3.0285, -1.0591, -1.2475, -1.2757],
        [-0.9405, -1.1578, -2.4222, -1.2143, -2.2928, -1.6776],
        [-4.3446, -1.1497, -1.2995, -1.2617, -2.7591, -1.1963],
        [-4.6625, -3.2917, -1.9607, -1.2207, -1.8363, -1.5121],
        [-3.9278, -3.2962, -1.7749, -1.5166, -1.8289, -1.2914],
        [-1.3905, -1.2013, -1.5848, -1.7187, -1.5615, -1.0713],
        [-2.7300, -1.3244, -2.2540, -1.1550, -1.6526, -0.9523],
        [-3.0774, -1.5142, -2.1737, -2.7045, -2.1132, -1.2606],
        [-2.6796, -1.2841, -2.8215, -3.6272, -1.8285, -1.7018],
        [-3.4720, -2.2198, -1.9880, -4.2422, -1.9908, -2.5370],
        [-3.6469, -3.6876, -5.0159, -0.9768, -1.8766, -1.3846],
        [-1.2522, -1.2471, -1.8144, -2.4175, -1.0965, -1.1637]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -7.2979, -14.8960, -10.4063, -11.9225, -12.5218, -14.9194,  -8.0145,
        -12.0104,  -9.1507,  -9.7051, -12.0108, -14.4840, -13.6357,  -8.5280,
        -10.0682, -12.8436, -13.9426, -16.4497, -16.5884,  -8.9914],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.9532e-01, -1.6409e+00, -9.7314e-01, -6.7490e-02, -2.8363e-02,
         -7.1030e-03],
        [-2.8869e+00, -3.7101e-01, -6.5039e+00, -7.7530e-01, -1.7461e+00,
         -1.1793e+00],
        [-2.1632e+00, -9.0044e-02, -8.8659e-01, -1.8048e+00, -2.2716e+00,
         -8.0028e-02],
        [-1.5836e+00, -5.0777e-01, -1.9156e+00, -2.3533e+00, -3.2241e+00,
         -1.6792e-01],
        [-2.1602e+00, -2.3883e+00, -2.1283e+00, -1.7735e+00, -2.4302e+00,
         -7.2203e-01],
        [-2.0819e+00, -1.5110e+00, -2.3582e+00, -2.8825e+00, -2.8750e+00,
         -7.3298e-01],
        [-3.0502e-01, -1.7948e+00, -2.0435e+00, -4.9422e-02, -1.0508e-03,
         -4.3442e-02],
        [-2.9920e+00, -1.4360e-03, -3.6745e+00, -4.4570e-02, -2.3144e+00,
         -1.4719e+00],
        [-9.7406e-01, -5.8977e-02, -4.2741e+00, -7.6954e-02, -8.3526e-01,
         -5.1263e-01],
        [-3.4205e-01, -4.5479e-01, -2.0087e+00, -1.5984e-01, -2.9558e+00,
         -1.3517e+00],
        [-4.2188e+00, -1.4442e-02, -8.6825e-01, -3.9743e-01, -3.4136e+00,
         -4.1250e-01],
        [-3.6494e+00, -3.3259e+00, -2.0431e+00, -2.3083e-02, -1.4892e+00,
         -6.1082e-01],
        [-3.5211e+00, -3.3775e+00, -1.3177e+00, -4.7317e-01, -1.7591e+00,
         -4.1960e-01],
        [-1.4329e+00, -1.0285e-01, -1.2006e+00, -1.4510e+00, -1.1310e+00,
         -1.1921e-07],
        [-3.0274e+00, -1.2501e+00, -2.2899e+00, -1.0412e-01, -1.7276e+00,
         -2.6259e-01],
        [-2.9933e+00, -1.4882e-02, -2.4509e+00, -2.8404e+00, -3.0993e+00,
         -4.6602e-01],
        [-2.1322e+00, -3.4301e-02, -3.6880e+00, -3.8491e+00, -2.2021e+00,
         -1.1038e+00],
        [-2.9370e+00, -2.1029e+00, -2.6029e+00, -5.0904e+00, -2.0064e+00,
         -3.2796e+00],
        [-2.1364e+00, -3.3985e+00, -3.9673e+00, -4.3298e-03, -1.7046e+00,
         -3.4013e-01],
        [-8.2949e-03, -8.3939e-01, -1.4673e+00, -1.9729e+00, -2.7636e-02,
         -1.5511e-01]], device='cuda:0')
tensor([ -3.1123, -13.4625,  -7.2963,  -9.7523, -11.6025, -12.4414,  -4.2372,
        -10.4988,  -6.7320,  -7.2729,  -9.3250, -11.1415, -10.8682,  -5.3184,
         -8.6617, -11.8648, -13.0096, -18.0191, -11.5512,  -4.4706],
       device='cuda:0')
tensor([-4.1856, -1.4335, -3.1100, -2.1702, -0.9193, -2.4779, -3.7772, -1.5116,
        -2.4187, -2.4323, -2.6858, -3.3425, -2.7676, -3.2096, -1.4064, -0.9787,
        -0.9330,  1.5694, -5.0372, -4.5208], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.5959, -2.1095, -2.2765, -1.0885, -2.0453, -2.0358],
        [-1.3601, -1.2326, -1.3933, -1.0609, -1.7181, -1.2614],
        [-4.1695, -1.3561, -2.1942, -1.2682, -1.3012, -1.1319],
        [-5.3197, -1.4098, -2.5863, -1.1940, -2.0657, -1.2476],
        [-4.1295, -1.2593, -1.5301, -1.7124, -2.0699, -1.8420],
        [-2.0687, -1.1490, -1.4593, -1.2550, -2.4196, -1.8506],
        [-4.6183, -2.5907, -2.9342, -1.0798, -2.1443, -1.3882],
        [-2.6282, -2.4425, -2.3220, -1.1433, -1.7801, -1.2737],
        [-1.2999, -1.3467, -1.9215, -1.1009, -1.3409, -1.3060],
        [-1.5598, -1.4951, -2.7958, -1.4634, -1.5405, -0.9779],
        [-1.3839, -1.2423, -3.0333, -1.2239, -1.5115, -1.8590],
        [-1.9190, -1.1646, -1.7577, -1.9827, -2.0328, -2.8584],
        [-5.9602, -2.4616, -1.9989, -1.2098, -1.8624, -1.0618],
        [-3.5640, -3.0118, -1.7719, -1.9224, -2.0477, -1.5997],
        [-1.2064, -1.3943, -1.3903, -1.3050, -1.6806, -1.2871],
        [-4.2731, -1.6946, -3.0727, -1.4325, -2.6363, -1.4415],
        [-4.6792, -1.2689, -2.6853, -1.2247, -1.7204, -1.2884],
        [-2.8988, -1.4924, -1.4885, -2.4582, -1.6381, -1.4442],
        [-6.0835, -2.2073, -3.6830, -2.3964, -2.7869, -2.2146],
        [-4.2705, -3.3170, -4.5291, -0.9674, -2.3362, -1.5257]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.1515,  -8.0263, -11.4211, -13.8232, -12.5431, -10.2021, -14.7554,
        -11.5898,  -8.3158,  -9.8324, -10.2538, -11.7152, -14.5547, -13.9175,
         -8.2638, -14.5507, -12.8668, -11.4201, -19.3717, -16.9459],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.2844e+00, -1.6327e+00, -2.7812e+00, -6.2455e-02, -2.3101e+00,
         -2.6086e+00],
        [-1.4701e+00, -3.1656e-01, -7.2073e-01, -1.0512e-03, -1.2992e+00,
         -1.0300e-01],
        [-4.3857e+00, -7.0525e-03, -2.5118e+00, -5.3004e-02, -2.6755e-01,
         -2.5870e-01],
        [-3.9726e+00, -2.5010e-04, -3.2476e+00, -5.1684e-02, -2.9358e+00,
         -6.7503e-01],
        [-3.4480e+00, -3.2783e-04, -5.1097e-01, -1.2029e+00, -2.7544e+00,
         -1.8786e+00],
        [-2.1931e+00, -2.3818e-02, -1.0463e+00, -4.1530e-01, -3.1744e+00,
         -1.8265e+00],
        [-3.2335e+00, -2.5999e+00, -4.9893e+00, -6.2845e-02, -2.7113e+00,
         -4.8043e-01],
        [-1.7489e+00, -2.2231e+00, -2.7028e+00, -3.6541e-02, -1.5846e+00,
         -3.6078e-01],
        [-1.1666e+00, -9.0207e-01, -1.7524e+00, -2.5006e-01, -3.0062e-01,
         -5.6479e-01],
        [-1.4761e+00, -5.9180e-02, -3.9962e+00, -4.0899e-01, -1.1579e+00,
         -1.6362e-01],
        [-1.0362e+00, -1.1563e-04, -3.5473e+00, -1.3277e-01, -1.2372e+00,
         -2.2577e+00],
        [-2.0940e+00, -3.1800e-01, -1.0195e+00, -1.7749e+00, -2.3116e+00,
         -3.2022e+00],
        [-4.4981e+00, -2.2943e+00, -2.2651e+00, -2.9214e-01, -1.6982e+00,
         -1.0954e-01],
        [-2.1885e+00, -3.9255e+00, -1.0429e+00, -6.5939e-01, -2.2585e+00,
         -5.7889e-01],
        [-5.3338e-01, -9.0119e-01, -5.3601e-01, -1.9897e-01, -1.0298e+00,
         -4.0046e-02],
        [-4.7747e+00, -6.0633e-01, -3.9322e+00, -4.3290e-01, -3.5175e+00,
         -5.2910e-01],
        [-3.6694e+00, -7.7096e-01, -2.9464e+00, -9.4152e-02, -2.0669e+00,
         -5.2897e-01],
        [-2.6783e+00, -7.7857e-02, -1.1710e+00, -2.4094e+00, -1.7600e+00,
         -1.0725e+00],
        [-5.2031e+00, -1.9786e+00, -6.3336e+00, -3.2217e+00, -3.9303e+00,
         -2.6820e+00],
        [-2.8426e+00, -3.1678e+00, -3.8717e+00, -1.5145e-02, -3.0925e+00,
         -8.1374e-01]], device='cuda:0')
tensor([-11.6795,  -3.9107,  -7.4838, -10.8831,  -9.7953,  -8.6794, -14.0773,
         -8.6567,  -4.9366,  -7.2620,  -8.2113, -10.7201, -11.1573, -10.6537,
         -3.2394, -13.7928, -10.0768,  -9.1691, -23.3494, -13.8034],
       device='cuda:0')
tensor([-1.4720, -4.1156, -3.9373, -2.9401, -2.7479, -1.5227, -0.6781, -2.9330,
        -3.3792, -2.5704, -2.0425, -0.9950, -3.3974, -3.2638, -5.0244, -0.7578,
        -2.7900, -2.2510,  3.9777, -3.1425], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-7.3036, -2.5603, -1.9286, -1.2223, -2.2361, -1.2231],
        [-4.2470, -3.2505, -2.0808, -1.9111, -2.6214, -1.8275],
        [-1.5429, -1.6370, -2.0305, -1.6753, -1.4775, -1.2799],
        [-1.6005, -1.5061, -2.3993, -1.4829, -1.4358, -1.1295],
        [-4.5397, -1.2977, -1.9697, -1.4411, -1.4783, -1.3562],
        [-2.6212, -1.2837, -1.1460, -1.6441, -1.3738, -2.2437],
        [-3.9086, -3.0258, -2.0953, -1.1520, -2.1477, -1.6992],
        [-3.1327, -3.0649, -1.9361, -1.7628, -2.5414, -1.4181],
        [-1.2782, -1.6706, -2.0863, -1.6552, -1.4847, -1.2204],
        [-3.3374, -2.1618, -3.0200, -1.3147, -2.0737, -0.9716],
        [-6.1132, -1.0845, -2.4203, -1.1135, -1.5096, -1.5079],
        [-2.5274, -1.4157, -1.2874, -1.5238, -1.4533, -1.2115],
        [-5.3658, -2.2629, -2.3249, -1.2641, -2.7197, -1.7501],
        [-3.9959, -2.8410, -2.5401, -1.4318, -2.2996, -1.6114],
        [-1.1601, -1.1829, -1.6020, -1.3953, -1.4222, -1.2878],
        [-4.3271, -1.3719, -2.4226, -1.6551, -1.2861, -1.0175],
        [-2.0581, -1.5158, -2.3836, -1.2331, -1.3868, -1.7558],
        [-3.1187, -1.4639, -1.7046, -1.5407, -1.4710, -1.4830],
        [-1.3588, -1.3558, -2.0424, -1.4893, -2.5806, -1.8317],
        [-3.6579, -2.8361, -2.0765, -1.1122, -2.0289, -2.1017]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-16.4739, -15.9383,  -9.6430,  -9.5541, -12.0827, -10.3124, -14.0286,
        -13.8559,  -9.3954, -12.8793, -13.7490,  -9.4191, -15.6876, -14.7199,
         -8.0504, -12.0803, -10.3332, -10.7820, -10.6587, -13.8133],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.9552e+00, -2.8513e+00, -1.5619e+00, -2.0128e-01, -2.7782e+00,
         -3.7185e-01],
        [-3.5878e+00, -3.5930e+00, -1.8734e+00, -9.5421e-01, -4.1142e+00,
         -1.8752e+00],
        [-1.8858e+00, -1.6620e+00, -2.1078e+00, -9.1068e-01, -4.0837e-01,
         -9.8962e-02],
        [-1.9989e+00, -4.9385e-03, -3.1228e+00, -3.3734e-01, -1.8474e-01,
         -9.6512e-01],
        [-4.0160e+00, -3.5219e-02, -2.1691e+00, -5.3667e-01, -1.3390e+00,
         -3.2915e-02],
        [-2.5518e+00, -1.8216e-01, -8.4284e-02, -1.1398e+00, -8.2367e-01,
         -2.8863e+00],
        [-2.7125e+00, -3.3190e+00, -2.6730e+00, -3.5714e-01, -2.4967e+00,
         -9.2401e-01],
        [-1.7172e+00, -4.1135e+00, -1.9651e+00, -6.9328e-01, -3.5840e+00,
         -3.3779e-02],
        [-7.2756e-01, -1.7966e+00, -2.1987e+00, -7.8109e-01, -1.1229e-01,
         -1.0912e-01],
        [-3.7929e+00, -1.5606e+00, -4.1761e+00, -4.7537e-02, -2.4637e+00,
         -9.2550e-02],
        [-4.6539e+00, -2.5477e-01, -2.7180e+00, -2.4009e-04, -1.4801e+00,
         -1.1806e+00],
        [-2.6009e+00, -3.2586e-01, -1.3192e-02, -7.6162e-01, -8.6167e-01,
         -2.7326e-01],
        [-4.0745e+00, -1.9739e+00, -3.2241e+00, -4.5641e-01, -3.7140e+00,
         -1.6847e+00],
        [-2.9004e+00, -3.2289e+00, -3.1388e+00, -2.7469e-01, -3.2934e+00,
         -7.5694e-01],
        [-4.5713e-01, -1.5196e-01, -1.0701e+00, -5.8132e-01, -2.6879e-01,
         -1.4255e-01],
        [-4.4577e+00, -8.1742e-04, -2.9224e+00, -9.8323e-01, -6.6787e-02,
         -4.8230e-01],
        [-1.7617e+00, -2.4336e-01, -2.9859e+00, -1.9407e-02, -7.9959e-01,
         -2.0788e+00],
        [-3.3264e+00, -3.1245e-01, -1.0155e+00, -7.9513e-01, -1.0210e+00,
         -8.7943e-01],
        [-1.1588e+00, -2.1355e-01, -2.0947e+00, -1.0730e+00, -3.3469e+00,
         -1.7362e+00],
        [-2.4048e+00, -3.4533e+00, -2.8179e+00, -4.4360e-02, -2.2668e+00,
         -2.1426e+00]], device='cuda:0')
tensor([-12.7197, -15.9978,  -7.0737,  -6.6138,  -8.1288,  -7.6679, -12.4822,
        -12.1068,  -5.7253, -12.1333, -10.2876,  -4.8365, -15.1275, -13.5931,
         -2.6719,  -8.9133,  -7.8887,  -7.3498,  -9.6231, -13.1297],
       device='cuda:0')
tensor([-3.7543,  0.0595, -2.5694, -2.9403, -3.9539, -2.6445, -1.5464, -1.7491,
        -3.6701, -0.7459, -3.4614, -4.5826, -0.5600, -1.1268, -5.3785, -3.1670,
        -2.4445, -3.4321, -1.0355, -0.6837], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.8439, -1.3480, -2.3287, -2.7530, -2.3460, -1.7891],
        [-2.9333, -1.0075, -1.4914, -1.2938, -2.4650, -1.4624],
        [-6.1225, -1.8750, -2.3434, -1.3757, -2.1495, -2.0986],
        [-3.1139, -3.6754, -2.1207, -2.8731, -2.2780, -2.0768],
        [-1.1415, -1.2499, -1.8538, -1.0663, -1.3833, -1.9682],
        [-5.3992, -1.3373, -2.9787, -1.2967, -2.0878, -1.8704],
        [-3.8508, -1.4109, -4.1539, -1.2560, -1.6026, -1.2706],
        [-0.9938, -2.1763, -1.2307, -1.6396, -1.4476, -1.4962],
        [-4.2363, -3.0665, -1.5388, -1.5766, -2.0632, -1.3930],
        [-3.9867, -3.5640, -2.2912, -1.2141, -2.1166, -1.3359],
        [-1.2008, -1.2605, -2.3309, -1.1710, -1.3403, -1.5469],
        [-4.1502, -1.2902, -2.6300, -1.4967, -2.2523, -1.4677],
        [-4.5306, -1.5167, -4.0081, -1.2402, -1.8578, -1.1157],
        [-2.7972, -1.0653, -1.3011, -1.3893, -2.7266, -1.4931],
        [-8.4549, -1.8097, -1.8487, -1.5723, -2.0748, -1.5134],
        [-3.7511, -3.0008, -2.2748, -1.9270, -2.0788, -1.5514],
        [-1.9579, -1.2183, -1.1861, -1.0645, -1.8083, -1.3510],
        [-2.6965, -1.4212, -1.8145, -1.3747, -1.8382, -1.4522],
        [-6.1785, -1.6473, -2.0097, -1.9068, -1.5214, -1.3417],
        [-1.8491, -1.8432, -1.1999, -3.5617, -1.6959, -1.2701]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-14.4086, -10.6533, -15.9647, -16.1380,  -8.6630, -14.9702, -13.5449,
         -8.9843, -13.8744, -14.5085,  -8.8504, -13.2872, -14.2692, -10.7727,
        -17.2738, -14.5839,  -8.5861, -10.5974, -14.6054, -11.4199],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.8676e+00, -2.3763e-01, -2.7738e+00, -3.1849e+00, -2.5467e+00,
         -1.5322e+00],
        [-3.2205e+00, -3.0915e-01, -8.0479e-01, -3.5453e-01, -2.8314e+00,
         -1.3862e+00],
        [-4.4824e+00, -8.2141e-01, -3.0314e+00, -3.7188e-01, -2.6425e+00,
         -2.2065e+00],
        [-2.1873e+00, -4.4066e+00, -2.3801e+00, -1.6323e+00, -3.1423e+00,
         -2.5999e+00],
        [-2.6857e-01, -8.0287e-01, -1.3229e+00, -3.2090e-01, -1.0572e+00,
         -1.8405e+00],
        [-5.6628e+00, -2.1144e-02, -3.4449e+00, -3.8060e-02, -2.3917e+00,
         -1.0628e+00],
        [-3.1839e+00, -1.9636e-02, -5.5537e+00, -2.3654e-01, -1.7436e+00,
         -7.0846e-01],
        [-4.5473e-01, -1.5578e+00, -3.8301e-01, -1.0134e+00, -1.1138e+00,
         -9.7431e-01],
        [-2.8056e+00, -4.0150e+00, -8.0003e-01, -1.4011e+00, -2.2118e+00,
         -3.9285e-01],
        [-3.3512e+00, -3.8655e+00, -2.8215e+00, -6.2236e-02, -2.3831e+00,
         -2.3853e-01],
        [-2.0903e-01, -6.8402e-01, -2.4294e+00, -3.3025e-01, -1.7830e-01,
         -1.1178e+00],
        [-4.3697e+00, -8.7498e-02, -3.1556e+00, -6.0409e-01, -2.8033e+00,
         -6.0650e-01],
        [-3.2990e+00, -7.0194e-02, -5.2273e+00, -8.5911e-02, -2.5777e+00,
         -3.3692e-02],
        [-2.7695e+00, -7.0324e-02, -2.9620e-01, -5.9258e-01, -3.5117e+00,
         -1.0340e+00],
        [-5.7680e+00, -1.0721e+00, -2.1048e+00, -1.2051e+00, -2.1464e+00,
         -1.1899e+00],
        [-3.1372e+00, -3.7011e+00, -3.1486e+00, -6.1231e-01, -2.3118e+00,
         -5.4096e-01],
        [-3.1774e+00, -3.0019e-01, -2.7227e-04, -4.5411e-01, -1.7001e+00,
         -2.8848e-01],
        [-3.4707e+00, -1.8561e-03, -1.3903e+00, -3.5001e-01, -1.8646e+00,
         -5.1332e-01],
        [-4.5294e+00, -4.7894e-01, -2.0099e+00, -1.6282e+00, -1.4112e+00,
         -8.0011e-01],
        [-1.4315e+00, -1.0459e+00, -4.9678e-01, -4.1287e+00, -1.8140e+00,
         -1.1654e-02]], device='cuda:0')
tensor([-14.1428,  -8.9066, -13.5561, -16.3485,  -5.6130, -12.6214, -11.4458,
         -5.4970, -11.6263, -12.7221,  -4.9488, -11.6267, -11.2938,  -8.2744,
        -13.4864, -13.4520,  -5.9206,  -7.5908, -10.8578,  -8.9284],
       device='cuda:0')
tensor([-0.2658, -1.7467, -2.4087,  0.2105, -3.0500, -2.3488, -2.0990, -3.4873,
        -2.2481, -1.7864, -3.9016, -1.6605, -2.9753, -2.4983, -3.7874, -1.1319,
        -2.6655, -3.0066, -3.7476, -2.4914], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-6.1183, -3.0756, -1.8605, -1.6657, -2.8575, -1.1701],
        [-2.2859, -2.4016, -2.2282, -2.1290, -2.9385, -1.4683],
        [-1.2188, -1.6905, -2.1530, -2.0234, -1.4503, -1.1805],
        [-3.2556, -1.4426, -2.5614, -1.3662, -1.5405, -0.9818],
        [-2.7752, -1.3497, -2.9246, -1.1191, -1.4859, -1.7821],
        [-1.0150, -1.3739, -1.3359, -1.4310, -1.3872, -1.2055],
        [-7.5147, -2.6939, -2.0171, -1.1914, -2.3530, -1.3422],
        [-3.7903, -3.4616, -3.5748, -2.0408, -2.4166, -1.4222],
        [-1.4857, -1.6023, -1.2245, -1.2728, -1.8282, -1.2111],
        [-2.2562, -1.7922, -1.8550, -1.4460, -1.2974, -0.9615],
        [-3.2788, -1.1572, -2.9983, -1.1940, -1.7159, -1.3371],
        [-4.5332, -1.3129, -3.0293, -1.2411, -1.5077, -2.1965],
        [-2.7988, -1.1714, -1.5977, -1.1465, -2.4396, -2.2393],
        [-5.0929, -2.7541, -1.9433, -1.1192, -1.7728, -1.4202],
        [-3.5779, -3.9358, -1.8604, -1.4435, -1.7036, -1.4286],
        [-1.2188, -1.2116, -1.3582, -1.0425, -2.2105, -1.4555],
        [-3.4451, -1.3389, -3.6057, -1.2983, -2.0159, -1.3863],
        [-4.9192, -1.4440, -2.5623, -1.7986, -1.5881, -1.1561],
        [-2.8867, -1.6558, -1.8820, -3.0884, -1.6727, -1.7832],
        [-8.5336, -2.9756, -2.0790, -2.6873, -2.4927, -2.0251]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-16.7477, -13.4516,  -9.7165, -11.1480, -11.4366,  -7.7486, -17.1124,
        -16.7064,  -8.6246,  -9.6082, -11.6813, -13.8207, -11.3932, -14.1025,
        -13.9499,  -8.4970, -13.0901, -13.4682, -12.9688, -20.7934],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.8823e+00, -3.5089e+00, -2.2315e+00, -1.3296e+00, -4.2088e+00,
         -1.1900e-01],
        [-7.5660e-01, -2.2817e+00, -3.0952e+00, -1.1191e+00, -4.2732e+00,
         -9.0069e-01],
        [-1.3551e-01, -1.8744e+00, -2.3755e+00, -1.1822e+00, -5.2265e-02,
         -2.2564e-02],
        [-3.2391e+00, -7.7006e-02, -2.8611e+00, -2.2116e-01, -1.0775e+00,
         -1.3281e-01],
        [-2.3955e+00, -9.5519e-01, -3.8093e+00, -1.6860e-02, -1.4066e+00,
         -1.9551e+00],
        [-5.2247e-01, -3.6787e-01, -2.5427e-01, -5.7959e-01, -8.2839e-01,
         -3.0516e-01],
        [-5.4524e+00, -3.1214e+00, -2.6267e+00, -3.8686e-01, -3.0903e+00,
         -3.8708e-01],
        [-2.8122e+00, -3.9904e+00, -4.7180e+00, -9.5995e-01, -3.6813e+00,
         -7.4804e-04],
        [-1.9456e+00, -1.7838e+00, -2.6467e-01, -4.1620e-01, -1.5212e+00,
         -1.3239e-01],
        [-2.3756e+00, -8.2883e-01, -1.6443e+00, -3.6870e-01, -3.5506e-01,
         -1.1921e-07],
        [-2.9466e+00, -1.1392e-01, -3.8796e+00, -1.6082e-01, -2.0019e+00,
         -6.0439e-01],
        [-3.8371e+00, -2.4155e-01, -2.9301e+00, -5.9760e-02, -1.1914e+00,
         -1.9053e+00],
        [-3.0346e+00, -1.5570e-03, -1.4104e+00, -8.8409e-02, -2.8937e+00,
         -2.1389e+00],
        [-3.8185e+00, -2.8715e+00, -2.1163e+00, -5.3755e-02, -1.2929e+00,
         -2.1696e-05],
        [-2.8772e+00, -3.9676e+00, -1.4753e+00, -5.4943e-01, -1.4627e+00,
         -7.5128e-01],
        [-7.1113e-01, -2.1796e-01, -6.5158e-01, -1.0516e-02, -2.7254e+00,
         -7.9579e-01],
        [-3.8867e+00, -8.0335e-03, -5.0864e+00, -5.9665e-02, -2.1527e+00,
         -6.0109e-01],
        [-3.6244e+00, -5.9580e-03, -3.0837e+00, -1.5535e+00, -1.7751e+00,
         -1.5803e-01],
        [-2.6010e+00, -6.5298e-01, -2.1861e+00, -3.4261e+00, -1.7185e+00,
         -1.9676e+00],
        [-5.1835e+00, -4.0417e+00, -2.9265e+00, -3.2424e+00, -3.4983e+00,
         -2.4002e+00]], device='cuda:0')
tensor([-15.2801, -12.4264,  -5.6425,  -7.6087, -10.5386,  -2.8577, -15.0648,
        -16.1627,  -6.0639,  -5.5725,  -9.7072, -10.1651,  -9.5675, -10.1529,
        -11.0835,  -5.1123, -11.7946, -10.2007, -12.5524, -21.2925],
       device='cuda:0')
tensor([-1.4676, -1.0252, -4.0739, -3.5393, -0.8980, -4.8908, -2.0476, -0.5437,
        -2.5607, -4.0357, -1.9742, -3.6556, -1.8257, -3.9497, -2.8664, -3.3847,
        -1.2956, -3.2675, -0.4163,  0.4991], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.7024, -1.1806, -3.2257, -1.6029, -1.4509, -1.3411],
        [-1.8712, -1.5467, -1.8175, -2.4819, -2.2322, -1.7110],
        [-4.5155, -3.3571, -1.9276, -1.1496, -2.0970, -1.4968],
        [-4.2311, -4.0877, -1.9659, -0.9839, -2.1512, -2.2487],
        [-1.2242, -1.6524, -1.5375, -2.4397, -1.4183, -1.3791],
        [-1.1672, -1.5064, -2.8197, -1.4905, -1.3522, -1.4009],
        [-3.3664, -1.1774, -2.5032, -1.1961, -1.3996, -1.3220],
        [-2.1200, -1.2303, -1.1820, -1.3827, -1.2419, -3.2854],
        [-4.3380, -2.8168, -2.4208, -1.1057, -1.5633, -1.8205],
        [-9.0552, -2.0832, -2.0393, -1.4048, -2.2781, -1.5917],
        [-1.3734, -1.5600, -2.6167, -1.8286, -1.6257, -1.2650],
        [-4.6069, -2.1335, -2.0018, -1.2611, -1.8111, -1.1752],
        [-5.6997, -1.1837, -2.6223, -1.3120, -2.9881, -1.6880],
        [-2.2036, -1.4227, -1.6293, -1.7656, -1.5680, -2.0750],
        [-5.2129, -2.1267, -1.8655, -1.2160, -2.3384, -1.6358],
        [-2.8745, -2.6371, -1.9147, -1.2601, -1.7944, -1.3272],
        [-1.9408, -1.1100, -1.2465, -0.9109, -1.5779, -1.1546],
        [-2.2183, -1.4302, -2.7219, -1.3493, -1.8856, -1.5776],
        [-5.0110, -1.5257, -1.7941, -2.6079, -1.5732, -1.2695],
        [-1.8288, -2.7741, -2.6354, -4.0781, -1.8192, -1.5023]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.5036, -11.6606, -14.5436, -15.6686,  -9.6512,  -9.7369, -10.9647,
        -10.4423, -14.0651, -18.4524, -10.2694, -12.9896, -15.4938, -10.6641,
        -14.3952, -11.8079,  -7.9408, -11.1828, -13.7813, -14.6381],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.1753e+00, -2.4658e-01, -4.2512e+00, -9.6851e-01, -1.2577e+00,
         -5.3120e-01],
        [-2.0346e+00, -5.0488e-01, -2.0707e+00, -2.6167e+00, -2.8726e+00,
         -1.2104e+00],
        [-3.0281e+00, -4.4553e+00, -2.1582e+00, -4.0700e-01, -2.4738e+00,
         -1.3185e-01],
        [-3.1521e+00, -4.0413e+00, -1.7456e+00, -2.0035e-02, -2.4757e+00,
         -2.7192e+00],
        [-5.9780e-02, -1.6855e+00, -1.0629e+00, -1.9136e+00, -8.1632e-02,
         -6.1763e-01],
        [-9.0270e-01, -3.8981e-05, -3.8531e+00, -5.8888e-01, -1.3647e-01,
         -1.1708e+00],
        [-2.9974e+00, -4.6115e-03, -3.1109e+00, -7.8722e-02, -1.0709e+00,
         -1.0513e-01],
        [-2.2038e+00, -8.5200e-02, -6.7006e-03, -4.6315e-01, -4.6137e-01,
         -4.0133e+00],
        [-2.8565e+00, -3.0913e+00, -3.8391e+00, -1.9711e-01, -7.5844e-01,
         -1.1218e+00],
        [-5.9480e+00, -1.6115e+00, -2.2926e+00, -3.6549e-01, -3.0007e+00,
         -9.0225e-01],
        [-1.3612e+00, -1.4619e+00, -3.1702e+00, -1.0577e+00, -8.5756e-01,
         -1.6764e-01],
        [-4.5883e+00, -1.4219e+00, -1.8915e+00, -2.7675e-02, -1.9664e+00,
         -4.9832e-01],
        [-4.4343e+00, -4.2250e-02, -3.4037e+00, -3.6307e-01, -4.1309e+00,
         -1.7528e+00],
        [-2.3085e+00, -4.6524e-02, -1.7146e+00, -1.3375e+00, -1.2941e+00,
         -1.5406e+00],
        [-3.7280e+00, -1.4972e+00, -1.7462e+00, -3.2539e-01, -3.3107e+00,
         -1.4844e+00],
        [-1.9289e+00, -2.6822e+00, -1.6733e+00, -1.2341e-01, -1.5175e+00,
         -2.8865e-01],
        [-2.9445e+00, -5.5093e-02, -2.9880e-01, -9.6288e-03, -1.1930e+00,
         -1.6270e-01],
        [-2.9511e+00, -1.9899e-02, -3.6090e+00, -9.5481e-02, -1.9787e+00,
         -5.6959e-01],
        [-3.6750e+00, -1.2703e-01, -1.3766e+00, -2.7406e+00, -1.7464e+00,
         -5.3878e-01],
        [-1.4890e+00, -2.3060e+00, -3.2745e+00, -4.6439e+00, -2.1297e+00,
         -1.1634e+00]], device='cuda:0')
tensor([-10.4306, -11.3099, -12.6542, -14.1540,  -5.4211,  -6.6520,  -7.3677,
         -7.2335, -11.8643, -14.1206,  -8.0762, -10.3941, -14.1271,  -8.2418,
        -12.0919,  -8.2140,  -4.6637,  -9.2238, -10.2044, -15.0066],
       device='cuda:0')
tensor([-2.0730, -0.3507, -1.8894, -1.5146, -4.2301, -3.0849, -3.5970, -3.2088,
        -2.2009, -4.3318, -2.1932, -2.5955, -1.3667, -2.4223, -2.3033, -3.5940,
        -3.2771, -1.9590, -3.5769,  0.3685], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.2511, -1.2437, -1.6729, -1.2325, -1.7060, -2.9169],
        [-5.1218, -2.2942, -1.5577, -1.1419, -2.4936, -1.8051],
        [-4.3144, -4.1589, -3.0475, -1.5726, -2.3019, -2.2252],
        [-1.3042, -1.2879, -1.6524, -1.5376, -1.5962, -1.6098],
        [-1.9099, -1.4896, -2.3910, -1.2920, -1.6049, -0.9709],
        [-3.9261, -1.2761, -5.8242, -1.1737, -1.5291, -1.7151],
        [-8.3951, -1.3395, -2.1093, -1.3552, -2.0878, -1.9996],
        [-3.1328, -1.1524, -1.2559, -1.1103, -2.3745, -1.9044],
        [-5.6947, -2.3575, -1.6947, -1.0803, -1.7329, -1.4086],
        [-3.3895, -5.9059, -2.4102, -1.2034, -2.1737, -1.5458],
        [-1.2532, -1.1921, -2.2625, -1.2241, -1.7927, -1.2781],
        [-2.0606, -1.3917, -2.5714, -1.3039, -1.3215, -1.4218],
        [-7.9318, -1.5387, -3.4211, -1.2078, -1.4207, -1.0900],
        [-6.9183, -1.1756, -2.2890, -1.1805, -2.3333, -1.6677],
        [-0.9152, -1.2536, -1.1929, -1.2201, -2.1327, -2.5392],
        [-6.2873, -2.3964, -1.7953, -1.0405, -1.8132, -1.7227],
        [-3.0706, -3.8577, -2.4964, -1.7927, -2.2802, -1.6984],
        [-1.9776, -1.2630, -1.2242, -1.1548, -1.9196, -1.1908],
        [-1.9826, -1.3891, -2.0059, -1.1035, -1.8398, -2.5843],
        [-1.6966, -1.3582, -3.6413, -1.1824, -1.4911, -1.4138]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.0232, -14.4143, -17.6205,  -8.9881,  -9.6584, -15.4443, -17.2865,
        -10.9304, -13.9686, -16.6285,  -9.0028, -10.0709, -16.6100, -15.5645,
         -9.2536, -15.0552, -15.1960,  -8.7300, -10.9052, -10.7834],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.1830e+00, -1.7659e-02, -1.8111e+00, -6.4013e-02, -1.4996e+00,
         -2.7158e+00],
        [-3.8732e+00, -1.8068e+00, -3.4895e-01, -6.3674e-03, -3.8877e+00,
         -1.7479e+00],
        [-3.3496e+00, -3.6399e+00, -3.5197e+00, -9.4201e-01, -2.6485e+00,
         -2.3729e+00],
        [-1.0171e+00, -4.3416e-01, -1.2923e+00, -6.5400e-01, -6.4216e-01,
         -1.1474e+00],
        [-2.2129e+00, -8.5556e-02, -3.0490e+00, -4.0607e-03, -1.2159e+00,
         -4.7707e-02],
        [-3.1295e+00, -3.6603e-01, -6.9903e+00, -1.2886e-01, -1.5293e+00,
         -1.8317e+00],
        [-5.8215e+00, -2.6959e-01, -1.6914e+00, -3.3087e-01, -2.8415e+00,
         -1.7107e+00],
        [-3.4910e+00, -8.3214e-02, -5.5202e-01, -2.5612e-03, -3.0242e+00,
         -1.8512e+00],
        [-3.9107e+00, -2.2707e+00, -7.7311e-01, -1.2579e-01, -1.2294e+00,
         -5.9685e-02],
        [-2.7790e+00, -5.9565e+00, -2.6068e+00, -2.1650e-01, -2.7767e+00,
         -1.0677e+00],
        [-9.0605e-01, -2.5155e-01, -2.4391e+00, -3.3278e-01, -1.6519e+00,
         -2.8742e-01],
        [-2.2487e+00, -1.4663e-02, -3.4273e+00, -5.3789e-02, -3.1169e-01,
         -5.8913e-01],
        [-5.0849e+00, -8.9398e-02, -4.6827e+00, -3.0294e-02, -9.6025e-01,
         -1.2897e-03],
        [-5.6668e+00, -7.3165e-01, -1.8909e+00, -8.2426e-02, -3.0031e+00,
         -1.5670e+00],
        [-1.6925e-01, -3.7705e-02, -6.3357e-01, -7.3586e-02, -2.4303e+00,
         -1.9764e+00],
        [-4.4802e+00, -2.0843e+00, -1.6635e+00, -6.0552e-02, -1.3149e+00,
         -1.4239e+00],
        [-2.1858e+00, -3.5760e+00, -2.7209e+00, -1.1564e+00, -2.7557e+00,
         -1.4744e+00],
        [-2.8629e+00, -2.9165e-01, -1.3780e-01, -2.3788e-02, -1.6546e+00,
         -1.8618e-01],
        [-2.1271e+00, -3.6179e-02, -2.1679e+00, -1.0850e-02, -2.0698e+00,
         -2.0548e+00],
        [-1.3499e+00, -1.1613e-01, -5.0324e+00, -6.8504e-03, -1.3043e+00,
         -9.5051e-01]], device='cuda:0')
tensor([ -7.2911, -11.6709, -16.4727,  -5.1870,  -6.6151, -13.9757, -12.6657,
         -9.0042,  -8.3694, -15.4032,  -5.8688,  -6.6453, -10.8488, -12.9418,
         -5.3208, -11.0274, -13.8692,  -5.1569,  -8.4667,  -8.7602],
       device='cuda:0')
tensor([-2.7321, -2.7434, -1.1478, -3.8011, -3.0433, -1.4687, -4.6208, -1.9262,
        -5.5993, -1.2253, -3.1340, -3.4256, -5.7612, -2.6226, -3.9328, -4.0279,
        -1.3268, -3.5731, -2.4386, -2.0233], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.0715, -1.2944, -1.9564, -1.2252, -1.7210, -1.4814],
        [-3.1934, -1.2317, -1.4643, -1.2835, -1.5581, -1.4937],
        [-6.1186, -3.5743, -1.6581, -1.1216, -1.6819, -1.4744],
        [-2.5874, -4.4913, -3.2272, -1.0994, -1.9116, -1.4579],
        [-1.6018, -1.3251, -1.5041, -1.0702, -1.2401, -1.6127],
        [-4.0978, -1.3914, -2.6426, -1.5731, -1.3384, -1.1895],
        [-3.7808, -1.2847, -3.0826, -1.2169, -1.9206, -1.1402],
        [-2.9524, -1.0055, -1.4400, -1.6902, -2.0556, -1.4160],
        [-4.5874, -2.7005, -2.1368, -1.6167, -2.4577, -1.6005],
        [-4.2199, -2.3760, -2.3142, -1.7669, -2.2644, -1.5574],
        [-1.2023, -1.2316, -1.2911, -1.0101, -1.4021, -1.2287],
        [-1.5825, -1.1061, -3.3945, -1.4244, -1.8499, -1.5587],
        [-6.1021, -1.5516, -1.7883, -1.9353, -2.6044, -1.3247],
        [-1.3751, -1.3560, -2.0250, -4.0068, -2.2658, -1.2704],
        [-5.5342, -2.0275, -1.8705, -2.1048, -2.1667, -1.9451],
        [-3.8122, -2.6174, -1.9679, -0.9336, -1.9087, -1.3522],
        [-1.3684, -1.2305, -1.7109, -0.8767, -1.9236, -1.8294],
        [-3.8674, -1.4293, -2.4669, -1.6723, -1.5316, -1.4065],
        [-2.0103, -1.4315, -1.2411, -1.2793, -1.4313, -2.1146],
        [-1.6457, -1.8571, -1.1783, -1.1328, -1.5215, -1.5176]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.7498, -10.2247, -15.6289, -14.7747,  -8.3540, -12.2328, -12.4257,
        -10.5597, -15.0995, -14.4989,  -7.3660, -10.9161, -15.3064, -12.2991,
        -15.6488, -12.5919,  -8.9394, -12.3741,  -9.5083,  -8.8530],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.6100e+00, -7.3043e-01, -1.3829e+00, -1.8290e-01, -1.8276e+00,
         -8.7419e-01],
        [-3.3931e+00, -3.0051e-02, -1.2099e+00, -1.6626e-01, -1.1509e+00,
         -8.1944e-01],
        [-4.5929e+00, -4.1144e+00, -9.1568e-01, -1.3802e-01, -8.1859e-01,
         -4.9200e-01],
        [-1.5918e+00, -4.8345e+00, -3.9974e+00, -6.2390e-02, -2.0573e+00,
         -8.2856e-01],
        [-1.9812e+00, -7.6170e-01, -9.5376e-01, -3.5942e-03, -2.3928e-03,
         -1.4572e+00],
        [-4.2042e+00, -1.5252e-02, -3.5696e+00, -6.7487e-01, -3.0338e-01,
         -3.3937e-01],
        [-3.2889e+00, -3.5199e-03, -4.2993e+00, -2.5984e-01, -2.6116e+00,
         -3.4299e-02],
        [-2.8860e+00, -1.3442e-01, -5.5804e-01, -1.2100e+00, -2.4934e+00,
         -8.1101e-01],
        [-3.4343e+00, -2.7896e+00, -2.5054e+00, -1.3715e+00, -2.9582e+00,
         -1.2381e+00],
        [-3.3246e+00, -2.4208e+00, -3.1464e+00, -4.5694e-01, -2.9974e+00,
         -2.1184e-01],
        [-8.1340e-01, -4.5820e-02, -4.2719e-01, -2.6845e-01, -3.8301e-01,
         -6.1782e-02],
        [-1.5153e+00, -3.0716e-01, -3.9773e+00, -6.1425e-01, -2.1367e+00,
         -1.3934e+00],
        [-4.9077e+00, -3.4811e-01, -1.0572e+00, -1.7088e+00, -4.3790e+00,
         -5.8938e-01],
        [-9.5586e-01, -4.6235e-02, -2.5277e+00, -4.2806e+00, -3.2235e+00,
         -3.4837e-01],
        [-4.8881e+00, -1.5274e+00, -2.2133e+00, -2.6524e+00, -2.5934e+00,
         -1.9098e+00],
        [-2.3318e+00, -2.1135e+00, -1.3089e+00, -1.9633e-02, -1.7912e+00,
         -1.1268e-01],
        [-1.5348e+00, -4.0305e-04, -1.1538e+00, -1.6207e-02, -2.0908e+00,
         -1.9759e+00],
        [-3.7958e+00, -9.9395e-02, -3.3454e+00, -1.1308e+00, -1.0075e+00,
         -8.6458e-01],
        [-1.1801e+00, -1.9093e-01, -1.6588e-01, -1.8494e-01, -4.9113e-01,
         -1.9441e+00],
        [-6.4451e-01, -2.2640e+00, -6.6086e-01, -7.4103e-03, -8.3139e-01,
         -1.2179e+00]], device='cuda:0')
tensor([ -8.6081,  -6.7697, -11.0716, -13.3719,  -5.1599,  -9.1066, -10.4976,
         -8.0929, -14.2971, -12.5580,  -1.9997,  -9.9440, -12.9903, -11.3823,
        -15.7845,  -7.6777,  -6.7720, -10.2434,  -4.1571,  -5.6261],
       device='cuda:0')
tensor([-3.1418, -3.4549, -4.5573, -1.4028, -3.1942, -3.1261, -1.9281, -2.4668,
        -0.8025, -1.9409, -5.3663, -0.9721, -2.3162, -0.9168,  0.1357, -4.9142,
        -2.1675, -2.1307, -5.3511, -3.2269], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.2275, -1.2555, -2.0583, -2.4939, -1.4655, -1.4065],
        [-1.1614, -1.3309, -2.1139, -1.3819, -1.4853, -2.1145],
        [-3.2874, -1.4350, -3.0628, -1.3689, -1.3420, -1.6559],
        [-2.2371, -1.7137, -2.2759, -2.5446, -1.6800, -1.6631],
        [-6.9585, -2.0884, -1.7441, -1.1560, -2.2073, -1.7715],
        [-5.5770, -3.4573, -2.3634, -1.4984, -1.8977, -1.6135],
        [-1.2631, -1.5487, -2.0711, -2.0463, -1.5469, -1.3142],
        [-2.4777, -1.8873, -3.8445, -1.2726, -1.8504, -2.0792],
        [-8.0998, -1.2589, -2.2201, -1.2751, -2.3840, -1.7914],
        [-3.0062, -1.4136, -2.0525, -2.3598, -1.4180, -1.2229],
        [-6.0427, -2.6623, -1.7710, -1.2856, -1.9934, -2.0394],
        [-3.5268, -2.2495, -2.8543, -0.9231, -2.2965, -1.3206],
        [-1.2618, -1.1463, -1.9917, -0.7986, -1.0714, -1.0887],
        [-2.0011, -1.4637, -1.8736, -1.5102, -1.6046, -1.1924],
        [-8.2114, -1.3321, -2.6118, -1.6352, -1.5214, -1.2983],
        [-2.7346, -1.5599, -1.1833, -1.7822, -2.1613, -1.4386],
        [-4.9234, -2.3930, -1.6628, -1.1205, -1.8687, -1.3890],
        [-4.0963, -3.0404, -2.0453, -1.0425, -2.1830, -1.3953],
        [-1.3257, -1.9563, -2.6898, -1.8708, -1.4169, -1.2444],
        [-1.7667, -1.5256, -2.3765, -1.3745, -1.5016, -0.9789]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.9073,  -9.5878, -12.1521, -12.1145, -15.9258, -16.4073,  -9.7903,
        -13.4118, -17.0292, -11.4730, -15.7944, -13.1708,  -7.3585,  -9.6456,
        -16.6102, -10.8598, -13.3575, -13.8027, -10.5041,  -9.5238],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-5.7058e-01, -4.5336e-01, -2.1305e+00, -1.9001e+00, -2.9870e-01,
         -7.7182e-01],
        [-8.1755e-01, -2.6550e-02, -2.0292e+00, -3.3672e-01, -9.1326e-01,
         -1.8967e+00],
        [-2.9128e+00, -1.5343e-02, -3.8200e+00, -4.8408e-01, -3.4935e-01,
         -1.3557e+00],
        [-1.9761e+00, -8.6841e-01, -2.5175e+00, -2.4896e+00, -1.6966e+00,
         -1.4289e+00],
        [-4.6323e+00, -1.5198e+00, -1.0384e+00, -3.1105e-01, -2.8431e+00,
         -1.4327e+00],
        [-3.6093e+00, -2.9969e+00, -2.4204e+00, -7.3580e-01, -1.7214e+00,
         -1.1963e+00],
        [-1.9632e-01, -1.4332e+00, -2.2354e+00, -1.1690e+00, -3.5101e-01,
         -3.5096e-01],
        [-2.6592e+00, -9.2063e-01, -4.7543e+00, -4.0510e-02, -2.1963e+00,
         -1.5537e+00],
        [-5.3874e+00, -7.6591e-01, -2.4295e+00, -2.6922e-01, -3.1765e+00,
         -2.0055e+00],
        [-2.5719e+00, -9.6400e-02, -2.2931e+00, -2.2613e+00, -1.1782e+00,
         -2.5642e-01],
        [-5.1774e+00, -2.8232e+00, -1.1311e+00, -5.9622e-01, -1.9816e+00,
         -2.3348e+00],
        [-2.1612e+00, -1.9080e+00, -2.7014e+00, -3.3118e-02, -3.0880e+00,
         -2.6978e-02],
        [-8.2927e-01, -6.1303e-01, -1.7333e+00, -3.4709e-02, -7.1120e-04,
         -1.3978e-01],
        [-2.4867e+00, -8.4209e-04, -1.4883e+00, -5.0142e-01, -1.1930e+00,
         -1.8451e-01],
        [-5.1924e+00, -3.9955e-02, -3.1909e+00, -1.0930e+00, -1.4387e+00,
         -3.8532e-01],
        [-2.4826e+00, -4.9725e-01, -4.6390e-01, -1.3664e+00, -2.8202e+00,
         -9.9693e-01],
        [-3.3870e+00, -2.4419e+00, -1.1098e+00, -3.2475e-01, -1.6881e+00,
         -1.3633e-03],
        [-3.0888e+00, -3.3032e+00, -2.1731e+00, -9.3354e-03, -2.5652e+00,
         -4.9441e-01],
        [-1.2427e+00, -2.2765e+00, -3.0461e+00, -1.3434e+00, -1.1320e-01,
         -1.5539e-01],
        [-1.8154e+00, -4.4336e-02, -3.1358e+00, -9.7865e-02, -9.8655e-01,
         -1.6737e-01]], device='cuda:0')
tensor([ -6.1250,  -6.0200,  -8.9372, -10.9771, -11.7773, -12.6801,  -5.7359,
        -12.1246, -14.0341,  -8.6574, -14.0444,  -9.9186,  -3.3508,  -5.8548,
        -11.3403,  -8.6272,  -8.9529, -11.6340,  -8.1772,  -6.2473],
       device='cuda:0')
tensor([-3.7822, -3.5678, -3.2148, -1.1375, -4.1485, -3.7272, -4.0544, -1.2872,
        -2.9951, -2.8156, -1.7500, -3.2522, -4.0078, -3.7909, -5.2699, -2.2326,
        -4.4046, -2.1687, -2.3268, -3.2765], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.7934, -1.3253, -2.5903, -1.3420, -2.3022, -4.1150],
        [-4.9476, -1.5174, -2.3219, -1.4444, -2.5123, -1.1449],
        [-3.3761, -1.2848, -1.2730, -2.7009, -2.0785, -1.2905],
        [-5.6952, -2.4273, -2.0304, -2.8095, -2.1442, -2.0922],
        [-3.9656, -2.8338, -1.8809, -1.4399, -2.0131, -2.7599],
        [-1.2432, -1.0274, -2.1124, -1.2760, -1.9349, -0.9942],
        [-3.1804, -1.4203, -2.4338, -1.3676, -2.1298, -1.4682],
        [-3.6768, -1.4434, -2.4152, -1.1548, -2.1847, -1.9561],
        [-6.4692, -1.1700, -2.3442, -2.0679, -2.0492, -2.0865],
        [-1.8870, -1.5747, -2.6624, -1.5905, -1.3167, -1.1918],
        [-2.9831, -2.7530, -2.0513, -1.3292, -2.1106, -1.6789],
        [-1.1923, -1.1592, -1.3459, -1.2331, -1.7808, -1.3753],
        [-2.8466, -1.2795, -2.3391, -1.1379, -2.1034, -2.0239],
        [-5.7314, -1.9070, -3.8519, -2.0555, -1.6547, -0.9369],
        [-1.6238, -1.4530, -1.2398, -1.7695, -1.7477, -1.3702],
        [-1.8502, -2.2552, -1.6291, -1.7306, -1.5395, -1.1868],
        [-4.9390, -2.7597, -2.6475, -1.1348, -1.7336, -1.3471],
        [-4.1297, -4.1046, -2.9790, -2.5220, -2.5014, -1.6478],
        [-1.1426, -1.3912, -1.2532, -1.0333, -1.3539, -1.0771],
        [-3.3303, -1.3483, -2.2631, -1.6741, -1.9947, -2.5616]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-15.4681, -13.8884, -12.0038, -17.1988, -14.8932,  -8.5881, -12.0000,
        -12.8310, -16.1870, -10.2232, -12.9060,  -8.0866, -11.7304, -16.1374,
         -9.2039, -10.1914, -14.5617, -17.8845,  -7.2512, -13.1720],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.0860e+00, -4.0482e-01, -3.3421e+00, -1.9742e-01, -2.7964e+00,
         -3.2511e+00],
        [-3.4456e+00, -1.7712e-03, -2.3977e+00, -7.3265e-01, -4.1944e+00,
         -1.8774e-01],
        [-3.1653e+00, -1.5950e-01, -6.5937e-01, -2.8241e+00, -2.5926e+00,
         -1.0294e-03],
        [-4.2254e+00, -2.6016e+00, -2.6586e+00, -3.4485e+00, -2.5986e+00,
         -2.5442e+00],
        [-3.1964e+00, -2.8772e+00, -1.6582e+00, -3.1742e-01, -2.1792e+00,
         -3.9444e+00],
        [-7.1582e-01, -6.7451e-02, -1.8163e+00, -5.8434e-01, -2.1250e+00,
         -3.1881e-02],
        [-4.0189e+00, -3.3023e-02, -2.8222e+00, -2.1853e-01, -2.4728e+00,
         -4.7811e-01],
        [-2.8911e+00, -1.4206e-01, -2.6128e+00, -2.5797e-04, -3.3010e+00,
         -2.4004e+00],
        [-5.1465e+00, -1.5906e-02, -1.8938e+00, -1.9895e+00, -2.5815e+00,
         -2.3594e+00],
        [-1.1073e+00, -7.6278e-01, -3.1000e+00, -1.1683e+00, -3.1267e-01,
         -4.1335e-01],
        [-1.9327e+00, -3.4067e+00, -2.2883e+00, -8.5344e-02, -2.3997e+00,
         -5.7143e-01],
        [-4.2768e-01, -4.8195e-02, -4.3788e-01, -8.2014e-01, -1.6715e+00,
         -5.2158e-01],
        [-3.5237e+00, -6.4934e-01, -2.5127e+00, -5.8335e-03, -2.1087e+00,
         -1.3114e+00],
        [-4.5312e+00, -9.3246e-01, -5.2200e+00, -2.0503e+00, -1.3221e+00,
         -3.9268e-02],
        [-1.1949e+00, -3.6151e-01, -1.3471e-03, -1.3646e+00, -1.9044e+00,
         -7.5793e-02],
        [-2.0586e+00, -2.0393e+00, -1.5302e+00, -1.4596e+00, -9.7997e-01,
         -2.0655e-01],
        [-3.5274e+00, -3.1284e+00, -3.9946e+00, -3.6663e-01, -1.3198e+00,
         -1.9880e-01],
        [-2.7423e+00, -3.5352e+00, -3.4858e+00, -2.1173e+00, -2.6129e+00,
         -1.4325e+00],
        [-6.7502e-01, -8.5658e-01, -2.6336e-01, -1.1287e-01, -3.8699e-03,
         -1.0677e-01],
        [-3.3987e+00, -2.9052e-01, -2.2934e+00, -1.0511e+00, -2.6629e+00,
         -1.8920e+00]], device='cuda:0')
tensor([-14.0779, -10.9598,  -9.4020, -18.0768, -14.1729,  -5.3408, -10.0436,
        -11.3477, -13.9865,  -6.8644, -10.6842,  -3.9270, -10.1117, -14.0953,
         -4.9025,  -8.2742, -12.5356, -15.9260,  -2.0185, -11.5886],
       device='cuda:0')
tensor([-1.3903, -2.9286, -2.6018,  0.8780, -0.7203, -3.2473, -1.9565, -1.4833,
        -2.2005, -3.3587, -2.2218, -4.1596, -1.6188, -2.0422, -4.3013, -1.9172,
        -2.0261, -1.9586, -5.2327, -1.5834], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.6867, -1.4396, -2.0129, -1.2872, -1.7457, -1.0156],
        [-3.2316, -1.2716, -5.1008, -1.5257, -1.3816, -1.4248],
        [-1.2318, -1.3225, -1.4674, -1.6484, -1.9154, -2.0358],
        [-5.6759, -2.1949, -1.9473, -1.0676, -1.9928, -1.6490],
        [-3.6530, -3.2216, -2.5724, -1.1573, -1.2657, -1.4796],
        [-1.3776, -1.2174, -1.1590, -1.4458, -1.3685, -1.4782],
        [-2.0827, -1.3170, -2.2978, -1.1164, -2.7367, -3.0743],
        [-5.1773, -1.3549, -1.7870, -1.1342, -2.3392, -1.2054],
        [-4.5413, -1.2564, -1.7113, -1.8998, -2.4157, -1.3528],
        [-4.2629, -3.9683, -2.2744, -1.9861, -2.6428, -2.5876],
        [-4.6475, -5.2157, -2.7912, -0.9641, -2.2861, -1.3252],
        [-1.5815, -1.1112, -1.5708, -0.9330, -1.2208, -1.2230],
        [-4.9994, -1.3446, -3.4522, -1.3762, -2.5180, -1.1496],
        [-4.1523, -1.5604, -2.3953, -1.2574, -1.4434, -1.4915],
        [-1.8432, -1.1255, -3.2000, -1.5376, -1.8778, -1.4597],
        [-1.8553, -1.1558, -1.0920, -1.2378, -1.8765, -2.7133],
        [-8.1045, -2.0941, -1.8057, -1.1036, -1.9020, -1.3874],
        [-3.4309, -5.3172, -3.0732, -1.3272, -3.0229, -1.6146],
        [-1.2390, -1.2068, -2.0387, -1.4935, -1.6327, -1.3640],
        [-4.0298, -1.2505, -2.4610, -1.2418, -1.4169, -1.5254]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.1876, -13.9361,  -9.6213, -14.5275, -13.3497,  -8.0466, -12.6249,
        -12.9979, -13.1772, -17.7220, -17.2299,  -7.6403, -14.8400, -12.3003,
        -11.0439,  -9.9306, -16.3973, -17.7860,  -8.9748, -11.9254],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.8843e+00, -2.8722e-01, -1.8690e+00, -5.4391e-02, -2.0007e+00,
         -5.6687e-01],
        [-2.6487e+00, -1.9128e-02, -6.2488e+00, -8.2521e-01, -5.2570e-01,
         -6.4414e-01],
        [-8.5978e-01, -2.6439e-01, -8.5619e-01, -9.8795e-01, -1.8994e+00,
         -2.0507e+00],
        [-3.9682e+00, -1.8240e+00, -2.2512e+00, -4.7030e-02, -2.1073e+00,
         -1.0739e+00],
        [-2.3378e+00, -2.9512e+00, -3.0375e+00, -2.3565e-01, -1.1491e-01,
         -9.2538e-01],
        [-9.8237e-01, -4.5549e-01, -2.3903e-02, -4.5488e-01, -6.1453e-02,
         -7.5933e-01],
        [-2.3348e+00, -4.4023e-02, -2.6769e+00, -2.2619e-02, -3.4687e+00,
         -2.2233e+00],
        [-3.9562e+00, -9.8991e-02, -1.1766e+00, -2.6139e-03, -3.5652e+00,
         -3.0761e-01],
        [-3.7542e+00, -1.7154e-02, -1.9106e+00, -1.6845e+00, -3.3950e+00,
         -4.3625e-01],
        [-3.8073e+00, -5.1576e+00, -3.3180e+00, -2.5670e+00, -4.0976e+00,
         -3.6550e+00],
        [-2.9952e+00, -4.6007e+00, -2.2444e+00, -3.3715e-02, -2.8763e+00,
         -5.1710e-02],
        [-2.1813e+00, -5.1711e-02, -8.0351e-01, -9.4004e-02, -6.0463e-01,
         -5.9113e-01],
        [-4.8623e+00, -9.0187e-03, -5.1633e+00, -3.1434e-01, -3.0852e+00,
         -3.0343e-01],
        [-3.3022e+00, -3.6840e-01, -2.7627e+00, -1.3896e-01, -8.0067e-01,
         -1.0416e+00],
        [-2.0328e+00, -7.6164e-02, -3.4039e+00, -7.4512e-01, -1.9431e+00,
         -6.5527e-01],
        [-1.9621e+00, -5.4044e-03, -2.7248e-01, -9.1483e-02, -1.9468e+00,
         -2.7151e+00],
        [-5.1253e+00, -1.5587e+00, -1.4964e+00, -2.0088e-01, -1.8104e+00,
         -6.0538e-02],
        [-2.4556e+00, -5.2901e+00, -3.4283e+00, -4.8365e-01, -4.1878e+00,
         -1.2442e+00],
        [-1.5557e-01, -7.2920e-02, -2.1062e+00, -6.7503e-01, -8.5402e-01,
         -3.6817e-01],
        [-4.3726e+00, -1.8784e-01, -2.7864e+00, -2.0427e-03, -6.7015e-01,
         -7.0446e-01]], device='cuda:0')
tensor([ -7.6624, -10.9117,  -6.9185, -11.2716,  -9.6024,  -2.7374, -10.7704,
         -9.1072, -11.1978, -22.6025, -12.8021,  -4.3262, -13.7377,  -8.4145,
         -8.8563,  -6.9933, -10.2522, -17.0896,  -4.2320,  -8.7235],
       device='cuda:0')
tensor([-2.5252, -3.0244, -2.7029, -3.2559, -3.7473, -5.3092, -1.8545, -3.8908,
        -1.9794,  4.8805, -4.4278, -3.3141, -1.1023, -3.8857, -2.1875, -2.9373,
        -6.1451, -0.6964, -4.7428, -3.2019], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-8.5982, -2.4085, -1.5505, -1.2445, -2.2148, -1.2587],
        [-3.3866, -3.5697, -2.6853, -1.4287, -1.9464, -1.5647],
        [-1.2317, -1.7545, -2.8761, -1.4503, -1.5080, -1.4467],
        [-3.5262, -2.0995, -1.9796, -1.5456, -1.3974, -0.8898],
        [-2.8874, -1.1363, -2.5309, -1.1974, -1.3455, -1.3195],
        [-4.6052, -1.3333, -5.3565, -1.2506, -1.8754, -2.3524],
        [-2.2678, -1.1224, -1.7305, -1.1438, -1.8545, -1.7681],
        [-4.0211, -2.2793, -1.7694, -1.1516, -2.1351, -1.4510],
        [-4.1618, -3.8021, -2.0772, -1.2782, -2.9106, -2.0343],
        [-1.6022, -1.1959, -1.6640, -1.0497, -1.4553, -1.2080],
        [-2.9441, -1.4229, -2.1122, -1.3337, -1.3610, -1.8698],
        [-3.5700, -1.4008, -3.6135, -1.1853, -1.4658, -1.1603],
        [-5.3062, -1.2082, -1.6966, -1.2411, -1.4913, -2.0294],
        [-1.6343, -1.1940, -1.6487, -1.1822, -2.1899, -1.7289],
        [-4.0467, -2.7295, -1.9787, -1.0969, -2.3852, -1.4571],
        [-2.7226, -3.6137, -2.0503, -1.5929, -1.9219, -1.9395],
        [-1.2153, -1.2677, -1.4081, -0.9982, -1.5288, -1.1404],
        [-1.8303, -1.6437, -3.3109, -1.8963, -1.8639, -1.6016],
        [-6.3809, -1.2184, -5.0189, -1.2520, -2.0786, -1.7847],
        [-9.8867, -1.2804, -1.3664, -2.9824, -1.6018, -1.1708]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-17.2753, -14.5814, -10.2673, -11.4381, -10.4172, -16.7735,  -9.8872,
        -12.8075, -16.2641,  -8.1753, -11.0438, -12.3958, -12.9729,  -9.5780,
        -13.6941, -13.8409,  -7.5585, -12.1468, -17.7335, -18.2885],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-5.0452e+00, -2.1914e+00, -5.3673e-01, -5.9048e-01, -2.8218e+00,
         -5.6411e-01],
        [-2.0525e+00, -3.7596e+00, -3.3682e+00, -3.2921e-01, -1.8858e+00,
         -1.0238e+00],
        [-7.6543e-01, -1.8991e+00, -3.3841e+00, -5.9781e-01, -5.3886e-01,
         -8.6644e-01],
        [-3.8156e+00, -1.5251e+00, -1.9222e+00, -6.1566e-01, -5.1958e-01,
         -1.0170e-01],
        [-2.7693e+00, -9.4478e-03, -2.8714e+00, -1.6529e-01, -4.4780e-01,
         -2.1853e-01],
        [-4.0612e+00, -3.9984e-02, -5.6209e+00, -2.0047e-01, -2.1870e+00,
         -2.0301e+00],
        [-2.3863e+00, -8.7571e-04, -1.6380e+00, -1.5586e-02, -1.8903e+00,
         -1.5325e+00],
        [-2.7465e+00, -2.0033e+00, -9.8425e-01, -1.6539e-01, -2.6989e+00,
         -1.6421e-01],
        [-3.6081e+00, -3.9301e+00, -2.0057e+00, -2.4464e-01, -4.2258e+00,
         -2.3065e+00],
        [-2.1380e+00, -2.3866e-01, -1.3406e+00, -4.4430e-02, -4.2850e-01,
         -1.1286e-01],
        [-3.3753e+00, -3.1284e-02, -2.3868e+00, -1.1657e-01, -3.9602e-01,
         -1.1476e+00],
        [-2.9510e+00, -1.3814e-01, -4.9052e+00, -1.4962e-02, -1.2379e+00,
         -3.4588e-01],
        [-4.2021e+00, -3.5636e-01, -8.7989e-01, -1.1818e-01, -1.1150e+00,
         -2.1958e+00],
        [-1.6343e+00, -6.3003e-03, -1.5338e+00, -2.4547e-01, -2.5360e+00,
         -1.4110e+00],
        [-2.9409e+00, -2.6667e+00, -2.2766e+00, -4.2569e-02, -3.4829e+00,
         -3.0399e-01],
        [-1.9131e+00, -3.7531e+00, -1.9787e+00, -5.9309e-01, -2.0642e+00,
         -2.1413e+00],
        [-8.0657e-01, -5.0376e-01, -7.5842e-01, -1.9251e-03, -7.2580e-01,
         -7.0038e-03],
        [-2.3692e+00, -3.9183e-01, -4.6020e+00, -1.5260e+00, -1.9618e+00,
         -5.9860e-01],
        [-4.6017e+00, -3.1934e-01, -5.9628e+00, -2.4899e-01, -3.0512e+00,
         -1.9549e+00],
        [-6.0700e+00, -3.1407e-02, -1.0578e+00, -3.2315e+00, -1.6353e+00,
         -1.3752e-01]], device='cuda:0')
tensor([-11.7497, -12.4191,  -8.0518,  -8.4999,  -6.4817, -14.1397,  -7.4635,
         -8.7626, -16.3210,  -4.3030,  -7.4535,  -9.5930,  -8.8673,  -7.3669,
        -11.7137, -12.4434,  -2.8035, -11.4494, -16.1389, -12.1636],
       device='cuda:0')
tensor([-5.5256, -2.1623, -2.2156, -2.9382, -3.9355, -2.6337, -2.4237, -4.0449,
         0.0569, -3.8723, -3.5902, -2.8028, -4.1056, -2.2111, -1.9804, -1.3975,
        -4.7551, -0.6975, -1.5946, -6.1249], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.2843, -1.5728, -2.5172, -2.1218, -1.3975, -1.4033],
        [-5.4749, -1.4581, -3.0283, -1.5899, -1.6842, -1.2382],
        [-5.6462, -1.4706, -1.2691, -1.3954, -1.7296, -1.2197],
        [-1.5763, -1.3232, -1.1884, -1.0663, -1.1462, -1.3663],
        [-2.5354, -2.2135, -1.4722, -3.3199, -2.0134, -1.2914],
        [-1.0575, -1.3833, -3.1112, -3.3705, -1.1794, -1.0864],
        [-3.7068, -1.4307, -2.6394, -1.7460, -3.0829, -2.4357],
        [-3.0772, -0.9511, -1.6161, -1.4046, -2.4052, -1.6027],
        [-7.0588, -3.0998, -2.0785, -1.3760, -2.3362, -2.4116],
        [-3.8293, -2.8571, -2.2533, -1.7579, -2.3863, -1.3552],
        [-1.4691, -1.4474, -2.3733, -0.9304, -1.8052, -1.1871],
        [-2.0468, -1.3307, -2.7749, -1.2774, -1.5603, -1.7162],
        [-4.0365, -1.3630, -2.1719, -1.1780, -1.3264, -1.1584],
        [-2.5162, -1.2306, -2.1615, -1.5886, -2.0274, -2.3752],
        [-1.0563, -1.2181, -1.1761, -1.3008, -1.6869, -1.4597],
        [-4.8421, -2.2794, -2.0804, -1.1289, -2.1752, -1.4645],
        [-3.0761, -3.3054, -2.0615, -1.5531, -1.7451, -1.2616],
        [-1.1553, -1.6189, -1.6249, -1.0641, -1.2117, -1.0525],
        [-3.5125, -1.4772, -2.5773, -1.7320, -1.6367, -1.2292],
        [-3.4759, -1.3015, -2.0503, -1.2875, -1.7465, -1.2305]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.2969, -14.4735, -12.7307,  -7.6667, -12.8458, -11.1883, -15.0416,
        -11.0570, -18.3608, -14.4392,  -9.2126, -10.7062, -11.2343, -11.8994,
         -7.8980, -13.9706, -13.0029,  -7.7274, -12.1648, -11.0922],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.4458e+00, -1.6429e+00, -2.7403e+00, -1.7272e+00, -6.3230e-01,
         -9.2212e-01],
        [-4.9429e+00, -2.9704e-01, -4.5068e+00, -9.2877e-01, -1.7107e+00,
         -2.8771e-01],
        [-2.5086e+00, -1.0902e-01, -2.8573e-01, -3.8888e-01, -6.3993e-01,
         -1.2010e-01],
        [-4.8209e-01, -1.0831e+00, -5.8775e-01, -2.1026e-01, -4.0939e-03,
         -1.2746e+00],
        [-2.0000e+00, -3.0947e+00, -9.4421e-01, -2.6343e+00, -2.3253e+00,
         -3.3250e-01],
        [-2.5091e-01, -9.5395e-01, -3.4133e+00, -3.4176e+00, -6.3345e-02,
         -7.9791e-02],
        [-3.8080e+00, -3.0804e-01, -3.2469e+00, -1.7738e+00, -3.5784e+00,
         -2.3261e+00],
        [-3.5127e+00, -1.3188e-01, -1.2700e+00, -5.4871e-01, -2.7371e+00,
         -1.5806e+00],
        [-5.0481e+00, -2.4554e+00, -2.3621e+00, -4.0359e-01, -3.0649e+00,
         -2.3931e+00],
        [-3.1555e+00, -3.1220e+00, -2.7036e+00, -5.9230e-01, -3.4414e+00,
         -4.2736e-02],
        [-1.7855e+00, -1.2115e+00, -2.3648e+00, -1.0276e-01, -1.7979e+00,
         -1.8067e-01],
        [-2.4411e+00, -3.9291e-04, -3.4546e+00, -6.7750e-02, -1.0705e+00,
         -8.6013e-01],
        [-3.1361e+00,  0.0000e+00, -2.1690e+00, -6.9141e-06, -5.6379e-01,
         -1.9971e-02],
        [-2.1862e+00, -2.6544e-03, -1.7329e+00, -1.0352e+00, -2.5994e+00,
         -2.8350e+00],
        [-5.9183e-01, -6.2505e-02, -6.0500e-01, -5.2150e-01, -1.4654e+00,
         -9.3248e-01],
        [-3.4427e+00, -1.9760e+00, -2.7280e+00, -3.1538e-01, -2.7684e+00,
         -4.7267e-01],
        [-2.4032e+00, -3.4535e+00, -2.1116e+00, -4.9864e-01, -1.4311e+00,
         -1.6692e-01],
        [-9.2016e-03, -1.6370e+00, -1.1558e+00, -2.0356e-01, -1.7108e-02,
         -2.3275e-02],
        [-3.5165e+00, -3.7057e-02, -2.9147e+00, -1.1791e+00, -1.4995e+00,
         -3.8166e-01],
        [-2.8287e+00, -4.4715e-01, -2.0036e+00, -3.7596e-01, -2.1431e+00,
         -3.1842e-01]], device='cuda:0')
tensor([ -9.1106, -12.6739,  -4.0523,  -3.6420, -11.3309,  -8.1789, -15.0414,
         -9.7810, -15.7271, -13.0576,  -7.4432,  -7.8944,  -5.8889, -10.3913,
         -4.1787, -11.7032, -10.0649,  -3.0459,  -9.5286,  -8.1169],
       device='cuda:0')
tensor([-1.1863e+00, -1.7996e+00, -8.6784e+00, -4.0247e+00, -1.5149e+00,
        -3.0094e+00, -2.1172e-04, -1.2760e+00, -2.6337e+00, -1.3816e+00,
        -1.7694e+00, -2.8119e+00, -5.3454e+00, -1.5081e+00, -3.7193e+00,
        -2.2674e+00, -2.9380e+00, -4.6815e+00, -2.6362e+00, -2.9753e+00],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([[-3.6026, -2.0542, -1.3280, -2.8997, -1.3315, -1.3979],
        [-1.4821, -1.0472, -1.2989, -1.3345, -1.5242, -1.5427],
        [-1.1876, -1.9046, -1.0220, -1.4343, -1.5289, -3.2347],
        [-4.2593, -3.5944, -1.6486, -1.0319, -2.0512, -1.5284],
        [-2.4998, -4.1330, -1.9149, -1.3923, -2.0251, -1.4720],
        [-1.2593, -1.5558, -2.0934, -2.2837, -1.3937, -1.1787],
        [-2.6950, -1.3369, -2.0356, -1.3486, -1.4604, -1.6951],
        [-4.4803, -1.4322, -3.8424, -1.1224, -1.8223, -1.2608],
        [-4.8768, -1.5778, -1.1270, -1.1339, -1.5753, -1.3089],
        [-1.4955, -1.6475, -1.8342, -1.1324, -1.3435, -1.1506],
        [-3.7962, -2.7901, -1.9206, -0.9031, -1.9291, -1.6907],
        [-1.7514, -1.8547, -1.8369, -1.0749, -1.5258, -1.3401],
        [-5.5483, -1.3537, -2.8445, -1.1546, -1.2049, -1.0940],
        [-1.1681, -1.4952, -2.6258, -1.6941, -1.1215, -0.9900],
        [-6.4812, -2.0132, -1.8867, -1.7811, -1.3432, -2.1966],
        [-1.6686, -1.8340, -1.6308, -2.1937, -1.1878, -1.1661],
        [-3.3820, -2.3372, -2.7338, -4.1650, -2.3200, -2.3567],
        [-3.3436, -0.9916, -2.3142, -1.2758, -3.0966, -1.5848],
        [-8.7379, -3.0024, -1.8836, -1.4433, -3.0692, -1.7782],
        [-6.4486, -3.9387, -2.8226, -2.5116, -2.4530, -1.6502]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.6139,  -8.2296, -10.3122, -14.1138, -13.4371,  -9.7646, -10.5717,
        -13.9604, -11.5997,  -8.6038, -13.0299,  -9.3837, -13.2000,  -9.0948,
        -15.7020,  -9.6810, -17.2948, -12.6066, -19.9146, -19.8247],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.8882e+00, -1.2702e+00, -6.7423e-02, -3.0304e+00, -1.2235e-01,
         -1.4172e-01],
        [-1.2572e+00, -6.2889e-02, -4.3798e-01, -5.1162e-01, -1.0837e+00,
         -1.2634e+00],
        [-1.0390e+00, -1.1866e+00, -3.5068e-01, -9.2024e-01, -9.8398e-01,
         -2.6940e+00],
        [-3.1759e+00, -4.5554e+00, -1.2693e+00, -4.4209e-02, -2.2967e+00,
         -8.6471e-01],
        [-1.3777e+00, -3.9414e+00, -1.6236e+00, -7.0907e-01, -1.8942e+00,
         -1.1153e+00],
        [-4.1707e-01, -1.5352e+00, -2.1982e+00, -1.7443e+00, -7.7975e-04,
         -1.3387e-03],
        [-2.9176e+00, -4.8008e-01, -1.9178e+00, -1.7817e-01, -7.3566e-01,
         -9.4838e-01],
        [-3.7364e+00, -7.7104e-01, -5.9955e+00, -6.5323e-03, -2.5596e+00,
         -2.9819e-01],
        [-4.1189e+00, -1.7559e+00, -9.0402e-02, -9.5725e-05, -1.4780e+00,
         -5.3530e-01],
        [-3.3445e-01, -3.9162e-01, -2.1333e+00, -1.4879e-01, -1.8516e-02,
         -5.2531e-03],
        [-2.3633e+00, -2.9970e+00, -1.7142e+00, -2.7061e-05, -1.9447e+00,
         -9.4057e-01],
        [-4.6113e-01, -1.1986e+00, -1.4610e+00, -9.1470e-02, -3.3468e-01,
         -2.9639e-01],
        [-6.0047e+00, -7.8585e-01, -3.3614e+00, -5.3078e-02, -3.0965e-01,
         -6.6583e-02],
        [-3.2970e-01, -1.6008e-01, -3.8728e+00, -8.0718e-01, -4.4617e-01,
         -3.2280e-01],
        [-4.9211e+00, -2.5185e+00, -1.9472e+00, -1.7603e+00, -3.7795e-01,
         -2.1189e+00],
        [-2.5460e+00, -1.7329e+00, -1.3935e+00, -1.6542e+00, -8.4048e-01,
         -7.4174e-02],
        [-3.8636e+00, -1.9054e+00, -3.1423e+00, -4.2802e+00, -2.3410e+00,
         -2.3598e+00],
        [-4.4187e+00, -6.4004e-01, -2.5562e+00, -3.2003e-01, -3.8740e+00,
         -1.6748e+00],
        [-5.7843e+00, -2.6638e+00, -2.0232e+00, -4.7080e-01, -4.9687e+00,
         -1.9725e+00],
        [-4.3845e+00, -4.2674e+00, -2.4325e+00, -1.6847e+00, -3.6618e+00,
         -9.3960e-01]], device='cuda:0')
tensor([ -7.5202,  -4.6169,  -7.1745, -12.2061, -10.6612,  -5.8969,  -7.1777,
        -13.3673,  -7.9786,  -3.0320,  -9.9598,  -3.8433, -10.5813,  -5.9387,
        -13.6439,  -8.2414, -17.8923, -13.4838, -17.8833, -17.3704],
       device='cuda:0')
tensor([-5.0936, -3.6127, -3.1376, -1.9077, -2.7759, -3.8678, -3.3940, -0.5931,
        -3.6211, -5.5718, -3.0702, -5.5404, -2.6187, -3.1561, -2.0581, -1.4397,
         0.5975,  0.8772, -2.0312, -2.4543], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.5739, -1.3016, -1.5759, -1.6423, -1.2421, -1.2516],
        [-2.6121, -1.6117, -1.5818, -1.1170, -1.0993, -2.5336],
        [-7.4154, -1.3062, -2.4033, -1.3813, -1.5697, -1.4000],
        [-2.8490, -1.8677, -1.8383, -2.4613, -1.1711, -1.1087],
        [-4.5092, -1.0550, -3.4702, -1.6211, -1.6589, -1.3789],
        [-2.8069, -0.9336, -1.2302, -0.9951, -2.1237, -1.0822],
        [-3.5313, -1.5307, -1.8946, -1.3487, -1.4607, -1.2616],
        [-2.8550, -2.3875, -2.4051, -1.1689, -2.1695, -1.8175],
        [-1.2256, -1.4608, -1.1515, -1.0222, -1.4930, -1.3593],
        [-3.6392, -2.0855, -3.1467, -2.9742, -2.6837, -5.4758],
        [-2.4078, -1.5559, -1.6991, -1.4995, -1.9113, -0.8204],
        [-0.9348, -1.2841, -2.2313, -1.5796, -1.2105, -1.4868],
        [-7.5590, -2.1056, -3.2217, -1.6008, -2.0291, -1.1977],
        [-4.3634, -2.1566, -1.9685, -0.8733, -1.9274, -1.2861],
        [-1.4755, -1.2584, -2.0506, -0.9438, -1.0832, -1.0573],
        [-3.0513, -1.5498, -2.4175, -1.4645, -1.4883, -1.1082],
        [-4.3850, -1.1001, -2.2263, -1.2469, -1.8912, -1.3967],
        [-3.1365, -1.1731, -1.2815, -1.8158, -1.8911, -1.2787],
        [-3.2097, -2.4525, -1.6428, -1.2048, -2.4050, -1.2701],
        [-3.1435, -2.9467, -1.7149, -1.4874, -2.1094, -1.5577]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.5874, -10.5554, -15.4759, -11.2961, -13.6934,  -9.1717, -11.0277,
        -12.8034,  -7.7123, -20.0051,  -9.8940,  -8.7272, -17.7140, -12.5754,
         -7.8687, -11.0795, -12.2462, -10.5767, -12.1848, -12.9597],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.6319e+00, -1.1040e-01, -1.0175e+00, -1.0474e+00, -1.8447e-01,
         -6.2940e-01],
        [-2.8797e+00, -3.4747e-01, -8.5337e-01, -1.8787e-04, -5.2769e-01,
         -2.4359e+00],
        [-5.1702e+00, -8.9395e-01, -2.8134e+00, -6.7515e-01, -6.9322e-01,
         -3.7942e-01],
        [-5.2677e+00, -2.4991e+00, -1.7415e+00, -2.1706e+00, -3.8801e-01,
         -4.6520e-02],
        [-4.3285e+00, -9.9282e-02, -3.6903e+00, -1.3695e+00, -1.4795e+00,
         -9.8234e-01],
        [-3.2779e+00, -1.9893e-01, -1.8239e-04, -3.4850e-02, -2.4976e+00,
         -3.2262e-01],
        [-2.5735e+00, -4.9450e-01, -1.9845e+00, -4.5803e-01, -7.1087e-01,
         -7.0355e-01],
        [-1.9630e+00, -2.3747e+00, -3.4241e+00, -1.6893e-03, -2.5955e+00,
         -1.6977e+00],
        [-5.7956e-01, -5.4492e-01, -8.5831e-06, -2.6336e-02, -1.7374e-01,
         -4.5938e-01],
        [-4.2455e+00, -2.7628e+00, -3.5514e+00, -3.4169e+00, -3.2668e+00,
         -4.1867e+00],
        [-2.5572e+00, -3.1557e-02, -7.4356e-01, -5.8919e-01, -2.3507e+00,
         -2.1942e-02],
        [-1.1886e-01, -4.1781e-01, -2.5894e+00, -9.8338e-01, -1.0016e-02,
         -6.2781e-01],
        [-4.7936e+00, -1.7621e+00, -4.0014e+00, -1.7042e+00, -1.9205e+00,
         -6.2366e-01],
        [-3.9640e+00, -1.5210e+00, -1.6239e+00, -6.9144e-03, -2.2563e+00,
         -5.9644e-02],
        [-1.8424e+00, -7.6885e-01, -1.9744e+00, -1.5504e-01, -4.1853e-02,
         -3.2633e-01],
        [-2.9281e+00, -2.9366e-01, -3.3283e+00, -4.2231e-01, -1.1507e+00,
         -2.4852e-02],
        [-3.9853e+00, -1.2361e-01, -2.4714e+00, -2.8256e-01, -2.4597e+00,
         -5.0218e-01],
        [-3.0829e+00, -2.3363e-01, -1.2626e-01, -1.3687e+00, -2.2822e+00,
         -5.0483e-01],
        [-2.2842e+00, -2.3566e+00, -1.5056e+00, -3.8479e-01, -2.9621e+00,
         -6.2261e-01],
        [-1.6676e+00, -3.4831e+00, -9.0957e-01, -2.9870e-01, -2.5201e+00,
         -4.4737e-01]], device='cuda:0')
tensor([ -6.6211,  -7.0444, -10.6253, -12.1134, -11.9494,  -6.3321,  -6.9249,
        -12.0567,  -1.7839, -21.4301,  -6.2942,  -4.7473, -14.8053,  -9.4316,
         -5.1088,  -8.1480,  -9.8247,  -7.5986, -10.1159,  -9.3264],
       device='cuda:0')
tensor([-3.9664, -3.5110, -4.8506,  0.8174, -1.7440, -2.8396, -4.1028, -0.7467,
        -5.9284,  1.4249, -3.5998, -3.9799, -2.9087, -3.1437, -2.7599, -2.9316,
        -2.4214, -2.9781, -2.0690, -3.6333], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.2211, -1.1339, -1.8778, -1.0938, -1.3622, -1.1274],
        [-2.7161, -1.1137, -3.0905, -1.2385, -1.3077, -1.1403],
        [-4.5481, -1.4264, -1.9744, -1.1541, -1.3100, -1.5140],
        [-2.1098, -1.4524, -1.3426, -1.7063, -1.9852, -1.8711],
        [-1.2640, -1.2171, -1.1656, -1.1694, -1.7922, -2.5368],
        [-5.2082, -2.4775, -2.1880, -1.2166, -1.7196, -1.2807],
        [-1.8375, -3.1341, -2.5394, -2.3829, -1.7393, -1.3735],
        [-1.0937, -1.4588, -1.7558, -1.0499, -1.4062, -1.0564],
        [-2.7738, -1.3304, -2.5753, -1.6067, -1.6968, -1.2941],
        [-2.5263, -1.2944, -2.6050, -1.1799, -1.5905, -1.3261],
        [-1.8590, -1.0900, -1.2855, -1.7161, -1.6996, -1.2752],
        [-3.8024, -1.9537, -1.6458, -1.2476, -1.7224, -1.3218],
        [-5.2865, -2.5003, -2.4186, -0.8962, -2.1538, -1.8617],
        [-1.6327, -1.1904, -1.8994, -1.0961, -1.5169, -1.3241],
        [-3.1567, -1.2870, -3.1323, -1.2719, -1.4042, -2.0522],
        [-9.3741, -1.6250, -1.6563, -1.2566, -1.4415, -1.0884],
        [-5.2833, -1.3590, -1.3696, -2.4517, -1.5198, -1.4652],
        [-4.4596, -2.7951, -2.0855, -1.6514, -2.2070, -1.2728],
        [-2.9646, -2.2831, -2.2899, -1.9039, -2.4515, -1.4322],
        [-1.6331, -1.6006, -1.8211, -0.9835, -1.3247, -1.2883]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -7.8162, -10.6066, -11.9270, -10.4673,  -9.1451, -14.0906, -13.0068,
         -7.8207, -11.2772, -10.5223,  -8.9254, -11.6936, -15.1172,  -8.6596,
        -12.3043, -16.4420, -13.4485, -14.4713, -13.3252,  -8.6513],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-8.7463e-01, -4.1103e-02, -1.6777e+00, -1.6596e-01, -1.3580e-01,
         -3.0091e-02],
        [-2.5709e+00, -4.4055e-02, -3.5435e+00, -4.4584e-04, -3.6732e-01,
         -6.4412e-02],
        [-3.5927e+00, -2.1092e-01, -1.7964e+00, -7.8144e-03, -6.4706e-01,
         -1.1736e+00],
        [-2.3622e+00, -4.2585e-01, -1.9431e-01, -9.9720e-01, -1.9258e+00,
         -1.7689e+00],
        [-9.2149e-01, -1.7250e-01, -2.3538e-01, -2.0159e-01, -1.7343e+00,
         -2.7691e+00],
        [-3.7219e+00, -2.6541e+00, -3.0656e+00, -5.6120e-01, -1.0752e+00,
         -2.9628e-01],
        [-5.7492e-01, -3.2306e+00, -3.0424e+00, -1.5855e+00, -1.4022e+00,
         -6.5979e-01],
        [-4.5083e-01, -1.1775e+00, -1.5062e+00, -9.1608e-02, -5.8294e-01,
         -5.0031e-02],
        [-3.2184e+00, -2.7020e-02, -3.1521e+00, -7.5470e-01, -1.6456e+00,
         -3.3882e-01],
        [-1.9842e+00, -2.3636e-01, -2.7903e+00, -1.8200e-03, -1.7296e+00,
         -2.1143e-01],
        [-1.5469e+00, -2.0070e-02, -6.4081e-01, -1.1881e+00, -1.7572e+00,
         -3.6018e-01],
        [-2.4033e+00, -1.1380e+00, -6.5915e-01, -4.5890e-01, -1.2858e+00,
         -1.3698e-01],
        [-3.4261e+00, -2.1264e+00, -2.7694e+00, -5.7697e-05, -2.5107e+00,
         -1.8685e+00],
        [-2.1416e+00, -2.6311e-01, -1.6785e+00, -1.0429e-01, -5.1200e-01,
         -3.0652e-01],
        [-3.3033e+00, -1.3554e-04, -4.2227e+00, -4.3593e-02, -9.7274e-01,
         -1.9124e+00],
        [-5.3611e+00, -4.4232e-01, -8.9585e-01, -2.2229e-01, -8.3999e-01,
         -8.7317e-02],
        [-4.4763e+00, -2.4469e-01, -3.1839e-01, -2.3040e+00, -1.2632e+00,
         -9.3357e-01],
        [-3.5308e+00, -2.7533e+00, -2.2100e+00, -1.3065e+00, -2.4091e+00,
         -4.6576e-01],
        [-2.2057e+00, -2.1295e+00, -3.5250e+00, -6.6522e-01, -3.3557e+00,
         -7.7448e-02],
        [-2.2780e+00, -1.5681e+00, -1.4974e+00, -4.8100e-03, -1.7415e-01,
         -2.2112e-01]], device='cuda:0')
tensor([ -2.9253,  -6.5906,  -7.4285,  -7.6742,  -6.0344, -11.3742, -10.4955,
         -3.8591,  -9.1367,  -6.9537,  -5.5133,  -6.0821, -12.7012,  -5.0061,
        -10.4548,  -7.8489,  -9.5401, -12.6754, -11.9586,  -5.7436],
       device='cuda:0')
tensor([-4.8910, -4.0160, -4.4985, -2.7931, -3.1107, -2.7164, -2.5114, -3.9617,
        -2.1405, -3.5686, -3.4121, -5.6115, -2.4160, -3.6535, -1.8495, -8.5931,
        -3.9084, -1.7959, -1.3665, -2.9077], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.7412, -1.2037, -1.0655, -1.4501, -1.6385, -1.5619],
        [-6.8622, -2.0003, -2.0515, -1.1698, -2.3605, -1.2094],
        [-3.5046, -3.1871, -2.0011, -2.3796, -1.9824, -1.4918],
        [-1.5796, -1.1827, -2.3948, -1.1901, -1.4016, -1.1858],
        [-2.7483, -1.4086, -3.8608, -1.3187, -1.6106, -1.1486],
        [-3.1445, -1.2417, -1.9537, -1.2658, -1.8609, -1.5799],
        [-2.6456, -1.4298, -1.7618, -1.2080, -1.4249, -1.7795],
        [-3.3429, -1.2363, -1.6566, -1.0818, -1.9411, -1.0921],
        [-3.5442, -2.4260, -1.8521, -1.1150, -2.8862, -1.9479],
        [-2.1391, -3.6530, -2.5148, -1.0127, -2.1969, -1.5205],
        [-1.2218, -1.7637, -1.1748, -1.8440, -2.5894, -1.6989],
        [-1.0407, -1.9453, -3.8755, -1.9859, -1.1701, -3.4725],
        [-3.0889, -2.1262, -2.5899, -1.1436, -1.4418, -1.8107],
        [-1.0793, -1.9547, -2.4075, -1.7987, -1.4602, -0.9372],
        [-3.1995, -2.7499, -3.0127, -1.7824, -1.2558, -1.5048],
        [-2.9900, -1.2848, -2.3413, -1.7150, -3.9955, -1.2688],
        [-4.9761, -2.8220, -2.2021, -1.4169, -2.1068, -3.8646],
        [-2.4204, -3.0311, -2.2720, -6.0144, -2.2746, -2.7569],
        [-1.5784, -1.6004, -1.2437, -0.8918, -1.1885, -1.3051],
        [-3.1072, -1.7617, -2.1922, -1.3772, -2.7377, -1.2879]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.6609, -15.6538, -14.5465,  -8.9346, -12.0957, -11.0466, -10.2496,
        -10.3508, -13.7714, -13.0371, -10.2927, -13.4899, -12.2012,  -9.6375,
        -13.5050, -13.5953, -17.3885, -18.7695,  -7.8079, -12.4639],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.6020e+00, -1.0786e-02, -3.7337e-02, -6.7512e-01, -1.6051e+00,
         -1.2666e+00],
        [-4.6735e+00, -1.3417e+00, -2.6412e+00, -2.7013e-01, -3.2473e+00,
         -1.8867e-01],
        [-2.5847e+00, -3.7148e+00, -2.0401e+00, -1.1758e+00, -2.0343e+00,
         -6.5842e-01],
        [-2.1944e+00, -4.2732e-03, -2.6933e+00, -2.4502e-01, -1.3664e-01,
         -1.4820e-01],
        [-3.2025e+00, -6.3238e-02, -5.0725e+00, -1.0004e-01, -1.2612e+00,
         -6.1608e-01],
        [-2.6249e+00, -4.5920e-01, -1.9104e+00, -4.4342e-01, -2.4708e+00,
         -1.6095e+00],
        [-2.6120e+00, -1.1059e-01, -1.0631e+00, -9.8923e-02, -8.9374e-01,
         -1.1998e+00],
        [-3.7687e+00, -8.7500e-05, -1.5238e+00, -4.7431e-03, -2.0627e+00,
         -2.0511e-02],
        [-2.4642e+00, -2.3951e+00, -1.6878e+00, -4.3991e-02, -4.8997e+00,
         -1.9384e+00],
        [-8.3228e-01, -3.4955e+00, -2.2619e+00, -1.1908e-02, -2.5311e+00,
         -8.6064e-01],
        [-2.7107e-02, -9.1380e-01, -2.1458e-05, -1.2826e+00, -3.1657e+00,
         -1.6086e+00],
        [-1.5965e-01, -1.3491e+00, -4.4535e+00, -1.5918e+00, -4.3273e-03,
         -3.1211e+00],
        [-2.5285e+00, -2.8716e+00, -3.0756e+00, -1.0674e-01, -1.3265e+00,
         -1.2620e+00],
        [-2.3743e-02, -2.6492e+00, -3.7368e+00, -1.2717e+00, -9.6321e-05,
         -1.3864e-02],
        [-3.9545e+00, -2.4825e+00, -3.2479e+00, -1.7877e+00, -2.5178e-01,
         -7.2730e-01],
        [-3.6623e+00, -1.2100e+00, -2.6525e+00, -1.0217e+00, -5.0270e+00,
         -1.0708e+00],
        [-3.4942e+00, -1.9222e+00, -2.9878e+00, -2.9947e-01, -2.5126e+00,
         -4.0154e+00],
        [-1.0494e+00, -3.5654e+00, -2.8353e+00, -3.8414e+00, -3.1786e+00,
         -4.2974e+00],
        [-1.9252e+00, -1.7193e+00, -8.2267e-02, -1.1492e-01, -2.1807e-01,
         -6.0839e-01],
        [-3.5370e+00, -6.0352e-01, -2.1553e+00, -3.2305e-01, -4.1293e+00,
         -6.0725e-01]], device='cuda:0')
tensor([ -6.1969, -12.3625, -12.2082,  -5.4218, -10.3155,  -9.5183,  -5.9782,
         -7.3805, -13.4292,  -9.9933,  -6.9978, -10.6795, -11.1709,  -7.6954,
        -12.4517, -14.6442, -15.2318, -18.7674,  -4.6681, -11.3554],
       device='cuda:0')
tensor([-3.4640e+00, -3.2913e+00, -2.3384e+00, -3.5127e+00, -1.7802e+00,
        -1.5283e+00, -4.2714e+00, -2.9704e+00, -3.4216e-01, -3.0437e+00,
        -3.2950e+00, -2.8105e+00, -1.0302e+00, -1.9421e+00, -1.0534e+00,
         1.0489e+00, -2.1568e+00, -2.1439e-03, -3.1398e+00, -1.1084e+00],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([[-3.4357, -1.7156, -2.1429, -1.2050, -1.7507, -2.4194],
        [-4.1271, -1.1029, -2.8796, -1.5686, -2.0197, -1.9447],
        [-5.2911, -1.2982, -1.4240, -1.2615, -2.4763, -2.1548],
        [-5.2323, -2.3065, -2.1166, -1.1755, -2.6335, -1.4841],
        [-3.1929, -4.5869, -2.1641, -1.6256, -1.7626, -1.8594],
        [-1.1687, -1.4575, -1.6658, -1.1518, -1.3368, -1.2559],
        [-3.8234, -1.4579, -2.7100, -1.5155, -1.4700, -0.8867],
        [-5.6372, -1.2429, -2.8032, -1.1499, -1.3302, -1.2536],
        [-2.7804, -1.3709, -1.5016, -1.1085, -1.7084, -2.2056],
        [-1.7373, -1.2411, -1.6130, -1.1297, -2.2783, -2.0601],
        [-4.5237, -2.6409, -2.3403, -1.1597, -2.0173, -1.4553],
        [-3.8506, -2.7010, -2.1006, -2.2028, -1.9591, -1.3280],
        [-2.1390, -1.6047, -1.4647, -1.3560, -1.3183, -1.2904],
        [-1.9463, -1.6239, -3.9273, -1.4841, -2.5443, -1.2804],
        [-5.6780, -1.2616, -2.4775, -1.2754, -2.2368, -1.5281],
        [-2.5840, -1.2252, -1.2400, -2.7881, -2.0202, -1.0731],
        [-5.8885, -2.5904, -2.0183, -1.2204, -2.5329, -1.4954],
        [-3.9233, -2.8496, -1.5958, -1.3671, -2.5423, -1.5903],
        [-1.5822, -0.9354, -1.1069, -0.9531, -1.5732, -0.9413],
        [-1.8582, -1.3059, -2.6419, -1.4765, -2.1761, -2.9873]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.6693, -13.6426, -13.9058, -14.9485, -15.1915,  -8.0364, -11.8637,
        -13.4170, -10.6753, -10.0595, -14.1371, -14.1421,  -9.1731, -12.8063,
        -14.4574, -10.9305, -15.7460, -13.8685,  -7.0922, -12.4458],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.7258e+00, -7.0272e-01, -2.2917e+00, -1.0419e-04, -2.1714e+00,
         -3.3234e+00],
        [-4.0790e+00, -6.7067e-01, -2.8154e+00, -9.3472e-01, -2.2573e+00,
         -2.2760e+00],
        [-4.8511e+00, -1.3116e-01, -1.2084e+00, -1.4371e-01, -3.1255e+00,
         -1.8093e+00],
        [-3.7624e+00, -2.0475e+00, -2.8722e+00, -3.4034e-01, -4.0823e+00,
         -4.4783e-01],
        [-2.4781e+00, -4.8890e+00, -2.2881e+00, -5.8523e-01, -1.4749e+00,
         -1.9757e+00],
        [-5.8271e-01, -1.2219e+00, -1.2833e+00, -1.9624e-01, -1.0998e-01,
         -4.1734e-01],
        [-3.9496e+00, -3.6884e-02, -3.5943e+00, -5.6503e-01, -9.0998e-01,
         -4.7263e-03],
        [-4.1557e+00, -1.2434e-01, -3.5268e+00, -1.6820e-04, -4.6824e-01,
         -3.9551e-02],
        [-2.9089e+00, -1.2931e+00, -6.8308e-01, -1.7911e-02, -1.7007e+00,
         -2.2100e+00],
        [-2.1968e+00, -1.4392e-01, -1.6159e+00, -1.3027e-02, -2.7888e+00,
         -1.2724e+00],
        [-3.3710e+00, -2.7022e+00, -3.4957e+00, -2.5326e-01, -2.2959e+00,
         -1.6116e-01],
        [-3.5587e+00, -2.6154e+00, -1.9597e+00, -1.4522e+00, -2.2384e+00,
         -2.2670e-01],
        [-2.9562e+00, -1.6951e+00, -8.8820e-01, -5.1516e-01, -5.2390e-02,
         -3.2237e-01],
        [-2.0525e+00, -3.9593e-01, -5.9264e+00, -4.5247e-01, -3.5236e+00,
         -4.4486e-01],
        [-4.0424e+00, -4.6335e-02, -2.6294e+00, -2.9815e-01, -3.4035e+00,
         -1.2102e+00],
        [-2.5831e+00, -7.5574e-02, -4.6557e-01, -3.0227e+00, -2.6048e+00,
         -1.1097e-01],
        [-4.5715e+00, -2.9379e+00, -2.6097e+00, -4.7805e-01, -3.5862e+00,
         -9.7602e-01],
        [-3.0781e+00, -2.9918e+00, -4.4899e-01, -2.3627e-01, -3.7859e+00,
         -1.2242e+00],
        [-1.8295e+00, -4.8802e-02, -1.5050e-01, -1.0248e-01, -1.5507e+00,
         -1.3800e-03],
        [-2.2478e+00, -2.5812e-01, -3.0961e+00, -6.5410e-01, -2.6244e+00,
         -1.8848e+00]], device='cuda:0')
tensor([-11.2151, -13.0330, -11.2691, -13.5526, -13.6911,  -3.8114,  -9.0605,
         -8.3148,  -8.8138,  -8.0309, -12.2792, -12.0511,  -6.4294, -12.7957,
        -11.6299,  -8.8627, -15.1594, -11.7653,  -3.6834, -10.7654],
       device='cuda:0')
tensor([-1.4542, -0.6096, -2.6367, -1.3959, -1.5004, -4.2250, -2.8031, -5.1023,
        -1.8615, -2.0286, -1.8579, -2.0911, -2.7437, -0.0105, -2.8275, -2.0678,
        -0.5867, -2.1031, -3.4088, -1.6804], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.1523, -1.3502, -1.2958, -0.9940, -1.4022, -1.1450],
        [-3.7679, -1.5841, -3.5058, -1.4380, -1.6366, -3.7508],
        [-4.7760, -1.2777, -1.9301, -1.2733, -1.5740, -1.2725],
        [-2.0678, -1.2982, -1.4004, -2.1683, -1.5758, -1.1449],
        [-4.8020, -2.0867, -1.8449, -1.6249, -2.4827, -2.5590],
        [-2.5969, -3.0435, -2.7039, -0.9737, -1.6864, -1.3163],
        [-1.3135, -1.4782, -1.8213, -0.8848, -1.1125, -1.1226],
        [-2.1855, -1.6919, -3.4379, -1.5772, -2.1779, -1.0422],
        [-2.3986, -1.1311, -3.3096, -2.5379, -2.4639, -1.5828],
        [-3.1646, -1.5621, -1.3149, -2.3739, -1.8851, -1.0420],
        [-5.5260, -2.4556, -2.0238, -2.6747, -2.7590, -2.0907],
        [-3.2273, -3.7361, -3.5615, -1.6812, -1.8369, -1.3362],
        [-1.2176, -1.7303, -2.9883, -1.0402, -1.2115, -1.0599],
        [-3.0562, -1.4646, -2.3355, -1.5285, -1.5563, -1.1497],
        [-3.7348, -1.3490, -2.4848, -1.1802, -1.4701, -1.1753],
        [-3.3837, -1.2736, -1.2094, -1.1335, -3.2188, -1.9654],
        [-1.9084, -1.6606, -2.1067, -1.1414, -1.9732, -2.9233],
        [-5.1899, -2.8903, -2.2202, -1.0215, -2.0763, -1.8909],
        [-5.5293, -4.2947, -2.9912, -1.3439, -3.0287, -1.7142],
        [-1.2514, -1.3949, -1.7495, -2.0458, -1.4401, -1.4339]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -7.3394, -15.6834, -12.1036,  -9.6553, -15.4002, -12.3208,  -7.7329,
        -12.1126, -13.4238, -11.3426, -17.5298, -15.3793,  -9.2478, -11.0907,
        -11.3943, -12.1844, -11.7136, -15.2892, -18.9022,  -9.3156],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.7738e-01, -8.1309e-01, -3.1116e-01, -5.3446e-03, -3.0102e-01,
         -5.9835e-02],
        [-4.1670e+00, -1.6353e-01, -4.6144e+00, -3.7904e-01, -1.3423e+00,
         -2.2779e+00],
        [-3.7073e+00, -8.0568e-01, -1.7548e+00, -3.9859e-01, -1.7595e+00,
         -9.7799e-03],
        [-1.6234e+00, -9.7827e-03, -9.2271e-01, -1.9949e+00, -1.6208e+00,
         -3.8466e-02],
        [-3.4471e+00, -1.5354e+00, -1.7435e+00, -1.6379e+00, -3.5941e+00,
         -3.6678e+00],
        [-1.1413e+00, -3.0464e+00, -2.4827e+00, -4.0331e-02, -9.9181e-01,
         -4.0294e-02],
        [-1.3187e+00, -1.4878e+00, -1.4239e+00, -8.5269e-03, -2.6350e-02,
         -5.1598e-02],
        [-2.5250e+00, -4.7429e-01, -5.0768e+00, -6.8086e-01, -2.9306e+00,
         -1.1575e-01],
        [-2.2493e+00, -1.5305e-01, -4.4216e+00, -2.4601e+00, -3.7496e+00,
         -1.4681e+00],
        [-2.9440e+00, -4.4930e-01, -8.4276e-01, -2.5747e+00, -2.3680e+00,
         -8.3785e-02],
        [-3.9815e+00, -2.7526e+00, -2.5769e+00, -3.3296e+00, -4.1355e+00,
         -2.5469e+00],
        [-1.8954e+00, -3.6965e+00, -3.1903e+00, -9.2525e-01, -1.4363e+00,
         -9.0193e-02],
        [-4.7501e-02, -2.0883e+00, -3.3451e+00, -3.1283e-01, -1.9853e-01,
         -2.7453e-01],
        [-2.9270e+00, -7.7236e-04, -2.4170e+00, -7.1524e-01, -1.1528e+00,
         -3.9696e-01],
        [-3.1179e+00, -6.0148e-01, -3.2268e+00, -8.7694e-02, -1.4145e+00,
         -2.4174e-02],
        [-3.3850e+00, -8.5215e-01, -4.1839e-02, -5.0852e-03, -4.0564e+00,
         -1.8733e+00],
        [-2.5258e+00, -2.8388e-01, -2.9230e+00, -1.1942e-01, -2.3680e+00,
         -2.7751e+00],
        [-3.9616e+00, -3.1957e+00, -2.5883e+00, -4.4516e-02, -2.4338e+00,
         -1.9922e+00],
        [-3.9777e+00, -3.7461e+00, -3.2835e+00, -6.3779e-01, -3.4990e+00,
         -1.3566e+00],
        [-2.6422e-01, -9.4679e-01, -1.5084e+00, -1.4774e+00, -2.3286e-03,
         -6.8718e-01]], device='cuda:0')
tensor([ -1.9678, -12.9442,  -8.4356,  -6.2100, -15.6258,  -7.7428,  -4.3169,
        -11.8033, -14.5017,  -9.2625, -19.3230, -11.2340,  -6.2668,  -7.6098,
         -8.4725, -10.2138, -10.9952, -14.2161, -16.5008,  -4.8863],
       device='cuda:0')
tensor([-5.3716, -2.7392, -3.6679, -3.4453,  0.2257, -4.5780, -3.4160, -0.3092,
         1.0779, -2.0800,  1.7932, -4.1453, -2.9810, -3.4809, -2.9218, -1.9706,
        -0.7185, -1.0731, -2.4013, -4.4293], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.9011, -2.2971, -1.1718, -1.6643, -1.6191, -1.9101],
        [-6.0863, -2.9172, -1.7946, -1.7448, -1.9954, -1.5306],
        [-3.0539, -3.3415, -2.0484, -3.3361, -2.1395, -1.2586],
        [-1.2595, -1.4851, -1.6959, -1.3569, -1.4489, -1.1686],
        [-3.8432, -1.5312, -4.0431, -1.2583, -2.4677, -1.1115],
        [-3.1716, -1.2254, -1.5696, -1.1159, -1.6444, -1.4822],
        [-2.1473, -1.7337, -2.2253, -2.0993, -1.4059, -0.9714],
        [-4.1370, -3.1295, -1.8526, -1.3526, -1.8997, -1.6598],
        [-3.6504, -4.5943, -1.7288, -1.7599, -2.0341, -1.5102],
        [-1.3290, -1.0934, -2.4532, -1.0655, -1.2509, -1.1186],
        [-2.9548, -1.4133, -1.9168, -1.4537, -1.3878, -1.0738],
        [-3.4672, -1.3699, -1.5379, -1.2198, -1.3187, -1.1313],
        [-1.0555, -1.4289, -1.6154, -2.3408, -1.3154, -1.0884],
        [-3.6393, -2.2643, -2.2747, -2.6234, -1.5650, -3.7110],
        [-4.9239, -1.6550, -1.8688, -0.9478, -3.0321, -2.3194],
        [-4.9978, -2.1697, -1.9614, -2.2365, -1.9576, -1.1995],
        [-2.9074, -4.5994, -1.9288, -1.3535, -2.7077, -2.7958],
        [-1.5360, -1.1934, -1.2643, -1.0109, -1.6573, -1.2571],
        [-1.8384, -1.2565, -1.6223, -1.1995, -1.3482, -0.9181],
        [-5.1399, -3.2502, -1.8507, -2.3596, -1.5844, -1.1423]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.5634, -16.0688, -15.1780,  -8.4150, -14.2551, -10.2092, -10.5829,
        -14.0312, -15.2777,  -8.3106, -10.2003, -10.0448,  -8.8443, -16.0778,
        -14.7470, -14.5225, -16.2927,  -7.9192,  -8.1830, -15.3270],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.5658e+00, -1.3756e+00, -1.6931e-01, -1.1925e+00, -1.5052e+00,
         -2.3340e+00],
        [-3.8559e+00, -3.2744e+00, -1.9429e+00, -1.4400e+00, -2.1812e+00,
         -8.0561e-01],
        [-1.6155e+00, -4.0015e+00, -2.5049e+00, -2.3865e+00, -2.2790e+00,
         -6.7098e-02],
        [-1.0742e-01, -1.3018e+00, -1.3333e+00, -2.5172e-01, -6.7091e-02,
         -2.6405e-03],
        [-4.1092e+00, -3.0678e-01, -4.9042e+00, -9.3094e-03, -3.4423e+00,
         -3.5310e-01],
        [-2.5692e+00, -6.2459e-01, -6.8162e-01, -5.9773e-02, -1.7754e+00,
         -1.1329e+00],
        [-2.1823e+00, -5.3795e-01, -3.1259e+00, -1.8253e+00, -8.0967e-01,
         -7.6791e-02],
        [-3.2286e+00, -3.4541e+00, -1.7158e+00, -7.6316e-01, -1.7061e+00,
         -9.9405e-01],
        [-2.5142e+00, -4.9927e+00, -1.1759e+00, -7.4454e-01, -2.2397e+00,
         -7.9599e-01],
        [-1.2935e+00, -8.8471e-02, -2.5429e+00, -1.6881e-01, -3.5462e-01,
         -9.6632e-02],
        [-3.2131e+00, -1.3471e-04, -1.8608e+00, -3.9151e-01, -3.4664e-01,
         -6.4320e-02],
        [-2.1273e+00, -1.2444e-02, -1.0313e+00, -9.4220e-02, -3.0668e-01,
         -2.1339e-01],
        [-1.0529e+00, -9.6536e-01, -1.5426e+00, -2.3819e+00, -4.9399e-01,
         -8.3853e-02],
        [-3.3568e+00, -2.2203e+00, -2.2220e+00, -2.3885e+00, -1.3846e+00,
         -3.6871e+00],
        [-6.1631e+00, -9.7453e-01, -1.4511e+00, -6.6898e-02, -3.4142e+00,
         -2.6934e+00],
        [-3.6919e+00, -1.6110e+00, -2.1627e+00, -2.5459e+00, -2.1301e+00,
         -4.3777e-01],
        [-1.9580e+00, -5.2481e+00, -1.9587e+00, -2.2881e-02, -4.0259e+00,
         -4.0299e+00],
        [-2.1751e+00, -2.0621e-01, -1.2865e-01, -2.9289e-01, -1.2076e+00,
         -1.9895e-03],
        [-1.9701e+00, -2.2881e-01, -1.1143e+00, -8.6895e-02, -5.8718e-01,
         -1.2943e-01],
        [-4.0380e+00, -3.4091e+00, -1.6623e+00, -2.5528e+00, -1.5285e+00,
         -2.0797e-01]], device='cuda:0')
tensor([-10.1424, -13.5001, -12.8544,  -3.0640, -13.1249,  -6.8435,  -8.5580,
        -11.8618, -12.4630,  -4.5449,  -5.8765,  -3.7854,  -6.5206, -15.2593,
        -14.7632, -12.5793, -17.2435,  -4.0124,  -4.1167, -13.3987],
       device='cuda:0')
tensor([-2.4210, -2.5687, -2.3236, -5.3511, -1.1302, -3.3657, -2.0249, -2.1694,
        -2.8147, -3.7656, -4.3238, -6.2594, -2.3237, -0.8184,  0.0161, -1.9432,
         0.9509, -3.9067, -4.0663, -1.9283], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-5.0683, -1.3431, -2.8920, -1.1566, -1.9247, -2.2448],
        [-2.2064, -1.3625, -1.2655, -1.7352, -1.6228, -1.6090],
        [-4.5385, -2.7161, -1.9284, -1.1892, -2.5597, -1.7266],
        [-3.3173, -3.7590, -2.2865, -1.5950, -2.0346, -1.8549],
        [-1.1405, -1.2288, -1.4541, -1.5260, -1.4394, -1.2745],
        [-1.2327, -1.5543, -2.8756, -1.6589, -1.4578, -1.1612],
        [-1.6742, -1.9165, -2.1854, -1.2200, -1.1843, -1.1056],
        [-5.9928, -1.1646, -2.9296, -1.1063, -1.3825, -2.5389],
        [-4.1972, -1.3103, -1.8004, -1.2176, -1.5697, -1.5630],
        [-2.6958, -1.1405, -2.3368, -1.1119, -1.7175, -2.2250],
        [-4.4602, -2.1990, -1.8069, -1.1122, -2.0996, -1.4787],
        [-5.6202, -3.2704, -2.9737, -1.5241, -2.2973, -2.9981],
        [-1.5445, -1.2689, -1.2762, -1.1193, -1.5325, -1.2481],
        [-1.1002, -1.2532, -1.6609, -1.3869, -1.4886, -1.7351],
        [-6.2804, -1.4321, -1.5755, -1.2968, -1.8214, -1.6320],
        [-2.3132, -1.2354, -1.0648, -1.8432, -2.9540, -1.3830],
        [-5.8369, -2.0613, -2.6967, -2.2329, -2.5826, -3.5289],
        [-3.2670, -2.6384, -1.7717, -0.9201, -2.0784, -1.6750],
        [-1.3740, -1.0924, -1.4127, -1.0365, -1.2661, -1.0744],
        [-3.4479, -1.5059, -2.7916, -1.3307, -1.8268, -0.9927]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-14.6296,  -9.8014, -14.6584, -14.8474,  -8.0632,  -9.9404,  -9.2860,
        -15.1148, -11.6582, -11.2275, -13.1565, -18.6837,  -7.9895,  -8.6249,
        -14.0382, -10.7937, -18.9392, -12.3508,  -7.2561, -11.8956],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.8897e+00, -1.9254e-01, -3.7983e+00, -1.5413e-03, -2.7203e+00,
         -2.6666e+00],
        [-2.1255e+00, -1.4320e-01, -5.6044e-03, -1.1582e+00, -1.6038e+00,
         -1.0248e+00],
        [-3.3810e+00, -2.7826e+00, -2.2441e+00, -2.9554e-01, -3.5902e+00,
         -1.5575e+00],
        [-2.1297e+00, -4.6380e+00, -2.6443e+00, -4.2120e-01, -2.2507e+00,
         -1.9485e+00],
        [-2.8001e-01, -3.1211e-01, -8.0067e-01, -7.2952e-01, -2.4936e-01,
         -7.6442e-02],
        [-1.0932e+00, -2.5600e-01, -3.7860e+00, -9.8034e-01, -5.0137e-01,
         -8.7082e-01],
        [-1.4765e+00, -1.0975e+00, -2.6276e+00, -3.7508e-01, -1.5049e-01,
         -2.8613e-01],
        [-5.5698e+00, -6.2811e-01, -2.8129e+00, -3.0432e-02, -7.5660e-01,
         -3.0335e+00],
        [-4.1577e+00, -2.3022e-01, -1.0284e+00, -4.3757e-02, -1.2698e+00,
         -6.6893e-01],
        [-2.9900e+00, -1.3144e-02, -2.6080e+00, -1.6913e-02, -1.5356e+00,
         -2.4897e+00],
        [-3.5411e+00, -1.7274e+00, -1.6191e+00, -4.3454e-03, -2.5662e+00,
         -5.5467e-01],
        [-4.5374e+00, -2.9723e+00, -3.4908e+00, -8.3777e-01, -2.7302e+00,
         -3.5827e+00],
        [-2.0261e+00, -2.2974e-01, -2.2789e-01, -2.5182e-02, -2.7182e-01,
         -1.8312e-02],
        [-7.0365e-01, -1.2988e-01, -1.1592e+00, -3.5848e-01, -9.1235e-01,
         -1.0437e+00],
        [-4.8131e+00, -4.8779e-01, -4.8253e-01, -3.3184e-01, -2.4614e+00,
         -1.3881e+00],
        [-1.8504e+00, -2.0681e-01, -1.4847e-01, -1.6007e+00, -4.5608e+00,
         -4.1183e-01],
        [-5.2584e+00, -1.7223e+00, -3.8896e+00, -2.7907e+00, -3.3592e+00,
         -4.4297e+00],
        [-2.0500e+00, -2.3250e+00, -1.0744e+00, -2.5427e-04, -2.4120e+00,
         -1.0934e+00],
        [-1.6877e+00, -1.0874e-01, -4.4309e-01, -2.4994e-01, -5.5891e-01,
         -1.4372e-02],
        [-4.1170e+00, -1.5578e-01, -3.7833e+00, -9.2259e-02, -1.7526e+00,
         -4.0915e-02]], device='cuda:0')
tensor([-13.2690,  -6.0611, -13.8510, -14.0324,  -2.4481,  -7.4877,  -6.0133,
        -12.8314,  -7.3989,  -9.6534, -10.0128, -18.1511,  -2.7990,  -4.3072,
         -9.9647,  -8.7790, -21.4498,  -8.9551,  -3.0628,  -9.9418],
       device='cuda:0')
tensor([-1.3606, -3.7403, -0.8074, -0.8150, -5.6151, -2.4527, -3.2727, -2.2834,
        -4.2593, -1.5740, -3.1437, -0.5326, -5.1905, -4.3178, -4.0735, -2.0146,
         2.5106, -3.3957, -4.1933, -1.9538], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.7899, -2.4421, -1.7237, -1.0935, -1.6432, -1.3994],
        [-4.4815, -3.2920, -1.5285, -1.6065, -2.6455, -1.6842],
        [-1.6131, -1.2345, -1.3349, -1.1311, -1.4667, -1.1776],
        [-2.2404, -1.4642, -2.2670, -1.6555, -2.1599, -1.1948],
        [-0.8736, -4.5211, -1.3306, -4.6824, -2.0078, -1.2975],
        [-7.5970, -5.0059, -2.4787, -3.3192, -2.2238, -2.6294],
        [-2.2404, -2.6662, -2.5203, -1.1876, -2.3300, -1.6834],
        [-1.2553, -2.7555, -2.2493, -3.9125, -1.5817, -1.3317],
        [-2.0029, -2.4062, -1.7623, -1.2418, -1.7681, -0.9351],
        [-3.9218, -1.6174, -1.4362, -1.5683, -1.9206, -2.6825],
        [-2.1049, -1.9987, -1.7578, -1.6924, -1.1632, -1.3782],
        [-1.9025, -1.7479, -1.7491, -1.9641, -1.4301, -1.2631],
        [-2.8940, -1.4254, -2.1387, -1.7367, -1.5246, -1.5075],
        [-2.9832, -1.0063, -1.9041, -1.1571, -4.0895, -2.4450],
        [-2.2307, -1.2411, -2.4467, -1.4213, -1.5377, -1.1554],
        [-4.8861, -2.6324, -1.8581, -1.2775, -2.3624, -1.8808],
        [-2.7972, -2.5311, -2.2156, -2.4181, -1.7716, -1.6708],
        [-1.1847, -2.4394, -2.8562, -0.8256, -1.1342, -1.0723],
        [-5.2171, -1.4972, -2.2485, -1.3340, -2.1584, -0.7337],
        [-2.4987, -1.4039, -1.9913, -1.8184, -1.6341, -1.3543]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.0918, -15.2382,  -7.9579, -10.9816, -14.7130, -23.2539, -12.6279,
        -13.0860, -10.1165, -13.1469, -10.0952, -10.0568, -11.2269, -13.5853,
        -10.0328, -14.8973, -13.4043,  -9.5124, -13.1889, -10.7006],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.5844e+00, -2.2951e+00, -1.3885e+00, -2.5415e-01, -9.2743e-01,
         -1.1971e-02],
        [-3.7344e+00, -3.3146e+00, -7.1489e-01, -7.4613e-01, -3.7271e+00,
         -1.4438e+00],
        [-2.1776e+00, -3.1480e-01, -5.2534e-01, -1.0141e-01, -3.8422e-01,
         -1.0158e-01],
        [-2.8034e+00, -4.1971e-02, -2.7202e+00, -9.2996e-01, -2.6586e+00,
         -2.6406e-01],
        [-3.8630e-01, -4.5338e+00, -4.0941e-01, -5.1115e+00, -2.9766e+00,
         -5.3365e-01],
        [-4.9840e+00, -5.0872e+00, -3.2642e+00, -3.5389e+00, -2.3698e+00,
         -2.7615e+00],
        [-1.1750e+00, -2.8655e+00, -3.8051e+00, -1.1158e+00, -2.6614e+00,
         -1.6920e+00],
        [-4.1829e-03, -4.0479e+00, -2.4074e+00, -3.7613e+00, -7.5302e-01,
         -1.0641e-01],
        [-1.9321e+00, -4.1594e+00, -1.1487e+00, -1.4858e-01, -1.9789e+00,
         -1.9343e-03],
        [-2.2251e+00, -2.4549e-01, -8.0437e-01, -9.9949e-01, -1.3051e+00,
         -2.9038e+00],
        [-1.4218e+00, -2.3719e+00, -1.6462e+00, -1.3413e+00, -1.3735e-03,
         -6.5291e-01],
        [-2.3631e+00, -1.9964e+00, -1.6118e+00, -1.4008e+00, -1.0238e+00,
         -3.2981e-01],
        [-2.8221e+00, -4.8435e-01, -2.2429e+00, -1.6288e+00, -1.0339e+00,
         -1.2475e+00],
        [-3.5959e+00, -5.5376e-01, -1.9174e+00, -7.2996e-01, -4.7066e+00,
         -3.0371e+00],
        [-2.0746e+00, -1.9017e-01, -3.0884e+00, -5.7446e-01, -1.2675e+00,
         -3.7147e-01],
        [-3.9419e+00, -2.4793e+00, -1.5804e+00, -1.5265e-01, -3.5892e+00,
         -2.2583e+00],
        [-1.8622e+00, -2.4512e+00, -2.2716e+00, -1.4777e+00, -1.5829e+00,
         -1.4751e+00],
        [-9.4219e-01, -3.1835e+00, -2.9776e+00, -1.1114e-03, -1.0888e-01,
         -8.1285e-02],
        [-5.4635e+00, -2.1017e-02, -2.4612e+00, -4.1461e-02, -2.9630e+00,
         -6.2092e-02],
        [-2.2839e+00, -1.8122e-01, -2.0526e+00, -1.2972e+00, -1.8166e+00,
         -9.2483e-02]], device='cuda:0')
tensor([ -8.4615, -13.6809,  -3.6050,  -9.4182, -13.9512, -22.0056, -13.3148,
        -11.0803,  -9.3697,  -8.4833,  -7.4354,  -8.7257,  -9.4596, -14.5407,
         -7.5666, -14.0017, -11.1206,  -7.2945, -11.0124,  -7.7240],
       device='cuda:0')
tensor([-4.6302, -1.5572, -4.3529, -1.5634, -0.7618, -1.2483,  0.6869, -2.0057,
        -0.7468, -4.6637, -2.6598, -1.3311, -1.7673,  0.9554, -2.4662, -0.8956,
        -2.2837, -2.2178, -2.1765, -2.9766], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.0668, -1.2196, -3.6921, -1.4234, -1.6793, -1.0007],
        [-4.6528, -1.3610, -2.7711, -1.3949, -2.6952, -1.3244],
        [-1.7104, -0.9601, -1.8756, -2.4881, -2.9995, -1.3680],
        [-7.8626, -2.1655, -1.8123, -1.7776, -2.3778, -1.6533],
        [-6.9051, -2.0653, -3.7250, -0.9858, -1.8153, -1.5447],
        [-1.2217, -1.2090, -2.1446, -0.8998, -1.1973, -1.0877],
        [-3.6384, -1.3788, -1.8734, -1.3359, -1.7622, -1.0040],
        [-3.3010, -1.5349, -5.6802, -1.2421, -1.5444, -1.5496],
        [-6.8085, -1.2111, -4.1535, -1.1172, -1.5688, -1.8829],
        [-3.4913, -1.2028, -1.2393, -1.1737, -2.5914, -2.6764],
        [-5.0689, -2.6038, -2.2043, -1.1370, -2.1914, -1.2777],
        [-2.3757, -3.0068, -2.9125, -1.4939, -2.1325, -1.3760],
        [-1.6535, -1.2743, -1.7214, -1.2311, -2.2275, -1.8915],
        [-2.5681, -1.4904, -3.4691, -1.2495, -1.3599, -1.0514],
        [-3.1355, -1.3132, -2.6667, -1.1897, -1.3722, -1.1582],
        [-2.4092, -1.1837, -2.3445, -1.2611, -1.5886, -1.5202],
        [-1.8787, -1.2784, -1.7436, -1.2058, -2.1425, -2.6085],
        [-5.9802, -2.2788, -1.7924, -1.1541, -2.1140, -1.3849],
        [-4.3207, -3.2207, -2.7618, -1.5989, -1.6608, -1.4019],
        [-1.7045, -1.2193, -1.8766, -1.1456, -1.4909, -1.2651]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.0820, -14.1994, -11.4016, -17.6491, -17.0412,  -7.7601, -10.9926,
        -14.8522, -16.7420, -12.3750, -14.4830, -13.2974,  -9.9993, -11.1883,
        -10.8355, -10.3073, -10.8576, -14.7042, -14.9648,  -8.7020],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.0907e+00, -4.2758e-01, -4.4288e+00, -3.5417e-01, -1.6729e+00,
         -1.3372e-01],
        [-3.9783e+00, -3.5549e-02, -4.0140e+00, -6.1380e-01, -4.1173e+00,
         -8.3805e-01],
        [-1.7416e+00, -2.1262e-02, -1.8988e+00, -2.2107e+00, -3.4615e+00,
         -3.3973e-01],
        [-6.1489e+00, -1.6702e+00, -1.4388e+00, -2.0784e+00, -3.3186e+00,
         -1.1259e+00],
        [-4.7175e+00, -1.5287e+00, -3.1988e+00, -1.6935e-02, -1.4498e+00,
         -8.3968e-01],
        [-8.7940e-01, -3.5970e-02, -2.0622e+00, -8.1440e-02, -2.2762e-01,
         -5.2820e-02],
        [-4.0741e+00, -1.9300e-02, -1.7233e+00, -1.4601e-01, -1.6362e+00,
         -6.9175e-02],
        [-2.5591e+00, -1.5729e-01, -6.8447e+00, -1.4546e-01, -1.3199e+00,
         -1.4661e+00],
        [-5.6149e+00, -2.4490e-01, -4.1596e+00, -1.5002e-03, -1.3182e+00,
         -1.7667e+00],
        [-3.5685e+00, -1.1584e-01, -4.9015e-01, -3.3045e-04, -3.0353e+00,
         -2.4542e+00],
        [-3.7654e+00, -2.5823e+00, -3.1568e+00, -1.5403e-03, -2.9244e+00,
         -7.0556e-02],
        [-1.2788e+00, -2.9870e+00, -3.5056e+00, -5.4839e-01, -2.7406e+00,
         -4.1713e-01],
        [-2.2108e+00, -5.2649e-01, -1.4941e+00, -2.5771e-01, -2.7243e+00,
         -1.7825e+00],
        [-3.0145e+00, -1.3172e-01, -5.0349e+00, -9.0873e-04, -3.4734e-01,
         -1.2916e-01],
        [-2.6207e+00, -2.8039e-03, -3.3103e+00, -1.9439e-01, -9.2618e-01,
         -4.5775e-03],
        [-2.2660e+00, -5.3636e-01, -1.8448e+00, -1.8527e-01, -1.4230e+00,
         -9.9814e-01],
        [-2.1190e+00, -8.7095e-01, -1.9153e+00, -2.4724e-02, -2.5016e+00,
         -2.1949e+00],
        [-4.1675e+00, -1.9543e+00, -8.9633e-01, -4.0332e-02, -2.5933e+00,
         -3.2444e-02],
        [-4.1154e+00, -2.9661e+00, -3.1289e+00, -8.3125e-01, -1.5526e+00,
         -6.3987e-01],
        [-2.2893e+00, -3.9767e-02, -1.8143e+00, -9.5386e-02, -2.2589e-01,
         -1.5467e-02]], device='cuda:0')
tensor([-10.1079, -13.5970,  -9.6736, -15.7807, -11.7513,  -3.3395,  -7.6681,
        -12.4925, -13.1058,  -9.6644, -12.5010, -11.4775,  -8.9959,  -8.6586,
         -7.0589,  -7.2536,  -9.6265,  -9.6842, -13.2342,  -4.4801],
       device='cuda:0')
tensor([-1.9741, -0.6024, -1.7279, -1.8684, -5.2899, -4.4206, -3.3246, -2.3597,
        -3.6362, -2.7107, -1.9820, -1.8199, -1.0034, -2.5298, -3.7766, -3.0537,
        -1.2311, -5.0200, -1.7307, -4.2219], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.8031, -1.5822, -2.5493, -2.8479, -1.7081, -1.2030],
        [-7.1530, -0.9425, -1.2700, -1.2130, -2.3919, -1.8214],
        [-2.4473, -2.2067, -1.7426, -1.3043, -1.4739, -0.9323],
        [-3.7303, -2.2310, -1.9797, -1.0184, -2.5049, -1.7271],
        [-1.1696, -1.3329, -2.5154, -1.1259, -1.4902, -1.2583],
        [-4.4851, -1.2322, -1.6955, -1.2505, -1.5305, -0.9834],
        [-4.7617, -1.8951, -1.9074, -1.7671, -1.6697, -0.9913],
        [-1.7900, -2.2681, -1.2061, -4.0009, -1.6536, -2.3283],
        [-4.8880, -2.8757, -1.9593, -1.8523, -2.5556, -1.0596],
        [-2.6462, -3.3764, -2.3264, -2.1915, -2.6403, -1.3501],
        [-1.2180, -1.4920, -2.1594, -1.1835, -1.3780, -1.1769],
        [-2.9347, -1.2698, -2.5264, -1.2757, -1.8864, -1.7174],
        [-3.4723, -1.4341, -1.9099, -1.2511, -1.4731, -1.1124],
        [-4.3558, -1.2768, -1.1908, -1.7932, -1.9356, -1.5414],
        [-1.0108, -1.2563, -1.8259, -1.1382, -1.4394, -1.2333],
        [-4.7355, -2.2893, -3.1438, -1.0400, -2.1664, -1.6710],
        [-2.6041, -1.6451, -2.8082, -1.0202, -1.8789, -1.3583],
        [-3.8282, -1.4888, -1.6774, -1.0802, -1.1934, -0.9638],
        [-2.3934, -1.4095, -2.3815, -1.5975, -2.3854, -1.0868],
        [-3.4915, -1.1994, -2.6474, -1.1113, -2.4574, -1.3671]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.6936, -14.7918, -10.1071, -13.1914,  -8.8922, -11.1772, -12.9922,
        -13.2468, -15.1905, -14.5309,  -8.6078, -11.6105, -10.6531, -12.0936,
         -7.9038, -15.0459, -11.3147, -10.2317, -11.2540, -12.2742],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.6970e+00, -8.3431e-01, -3.0301e+00, -3.2315e+00, -1.3166e+00,
         -4.2716e-01],
        [-6.4917e+00, -1.6668e-01, -6.3765e-02, -2.5459e-01, -2.9091e+00,
         -2.3654e+00],
        [-1.3489e+00, -1.4510e+00, -1.8199e+00, -1.1005e-01, -3.5244e-01,
         -4.3675e-02],
        [-2.3867e+00, -2.0595e+00, -2.1737e+00, -1.5595e-02, -3.7048e+00,
         -1.3025e+00],
        [-3.2234e-01, -8.1619e-01, -2.5535e+00, -2.3136e-01, -8.9661e-02,
         -1.7301e-02],
        [-4.5445e+00, -6.2912e-01, -1.3525e+00, -2.8970e-02, -1.0955e+00,
         -2.6147e-03],
        [-3.9032e+00, -1.1281e+00, -1.6300e+00, -1.3319e+00, -1.9300e+00,
         -4.0718e-03],
        [-1.5351e+00, -1.5903e+00, -1.0990e-01, -4.0688e+00, -1.5548e+00,
         -3.1453e+00],
        [-3.1784e+00, -3.1419e+00, -2.5094e+00, -1.6025e+00, -3.4702e+00,
         -3.2506e-02],
        [-1.5084e+00, -4.1997e+00, -3.4607e+00, -1.1474e+00, -3.4867e+00,
         -1.4520e-02],
        [-2.6359e-01, -1.3939e+00, -2.2581e+00, -1.9598e-01, -3.0846e-01,
         -2.0568e-01],
        [-3.3108e+00, -3.2916e-02, -2.8602e+00, -6.7902e-03, -2.1129e+00,
         -9.6216e-01],
        [-2.8096e+00, -4.5763e-02, -1.5077e+00, -2.2194e-01, -1.2092e+00,
         -2.6751e-02],
        [-3.9482e+00, -6.9383e-03, -1.3271e-02, -1.3787e+00, -2.3191e+00,
         -1.0669e+00],
        [-1.1576e-02, -2.6294e-02, -1.8544e+00, -1.9127e-02, -3.9050e-01,
         -2.8870e-01],
        [-4.2753e+00, -2.2559e+00, -4.8805e+00, -2.1230e-02, -2.6953e+00,
         -9.5923e-01],
        [-1.7276e+00, -9.0596e-01, -2.9504e+00, -9.0089e-03, -1.7899e+00,
         -2.5887e-01],
        [-4.8782e+00, -1.0782e+00, -1.2304e+00, -5.6410e-04, -9.9123e-02,
         -3.8459e-02],
        [-2.6086e+00, -2.8718e-02, -3.1907e+00, -7.2428e-01, -3.1960e+00,
         -2.8376e-01],
        [-2.9762e+00, -5.7072e-01, -3.1484e+00, -6.1891e-03, -3.6254e+00,
         -4.6025e-01]], device='cuda:0')
tensor([-10.5366, -12.2512,  -5.1260, -11.6428,  -4.0304,  -7.6532,  -9.9273,
        -12.0043, -13.9348, -13.8174,  -4.6257,  -9.2858,  -5.8210,  -8.7332,
         -2.5906, -15.0875,  -7.6417,  -7.3249, -10.0320, -10.7872],
       device='cuda:0')
tensor([-1.1570, -2.5406, -4.9810, -1.5486, -4.8618, -3.5240, -3.0649, -1.2426,
        -1.2556, -0.7135, -3.9821, -2.3247, -4.8321, -3.3603, -5.3132,  0.0415,
        -3.6730, -2.9068, -1.2220, -1.4870], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.6033, -1.2277, -1.2466, -1.0234, -1.6557, -1.4820],
        [-2.7774, -1.3242, -2.5990, -1.4019, -2.5887, -0.8871],
        [-2.1119, -1.3170, -2.6663, -1.5030, -1.5864, -1.1854],
        [-2.1871, -1.4487, -1.4785, -3.2252, -2.2295, -1.1893],
        [-3.8113, -2.2826, -2.0302, -2.5716, -1.8973, -1.9948],
        [-2.7398, -2.3842, -2.3230, -1.1940, -1.9426, -1.7637],
        [-1.4277, -1.2466, -1.7611, -0.9097, -1.3284, -1.0801],
        [-1.9436, -1.3785, -2.2442, -1.7566, -2.2275, -3.6216],
        [-2.3168, -1.4200, -2.1699, -1.1308, -2.1949, -1.3225],
        [-2.6895, -1.3882, -1.1997, -1.9481, -1.8059, -1.3302],
        [-5.6669, -3.1622, -1.8950, -1.6010, -3.4806, -1.9973],
        [-3.6865, -2.8109, -1.8782, -1.2295, -2.3397, -1.9046],
        [-1.6152, -1.2126, -1.2894, -2.2511, -1.4693, -1.3209],
        [-2.0837, -1.4599, -2.1472, -1.9312, -1.2515, -1.1334],
        [-3.7275, -1.4230, -1.3956, -1.2569, -1.2868, -1.5610],
        [-3.0829, -3.9739, -1.3698, -1.8187, -1.3676, -1.0672],
        [-2.3378, -1.9995, -1.7880, -2.6943, -1.3094, -1.8445],
        [-4.9484, -1.6788, -3.2327, -2.6660, -1.8130, -1.6525],
        [-3.0867, -1.0089, -2.0781, -1.2986, -4.2606, -1.1487],
        [-5.8740, -2.1062, -2.7766, -1.4958, -2.1643, -1.5454]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.2389, -11.5783, -10.3699, -11.7583, -14.5877, -12.3472,  -7.7536,
        -13.1720, -10.5549, -10.3615, -17.8030, -13.8494,  -9.1584, -10.0069,
        -10.6506, -12.6801, -11.9735, -15.9915, -12.8816, -15.9623],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.1003e+00, -1.0120e-01, -1.2707e-01, -4.7731e-03, -1.1268e+00,
         -8.1001e-01],
        [-3.2644e+00, -6.7735e-03, -3.2664e+00, -3.6657e-01, -3.3157e+00,
         -1.8939e-03],
        [-1.7385e+00, -1.3962e-01, -3.3569e+00, -8.5387e-01, -1.7071e+00,
         -2.4865e-01],
        [-2.0688e+00, -1.9412e-01, -1.2994e+00, -3.6191e+00, -3.0929e+00,
         -3.6925e-01],
        [-2.3454e+00, -2.2111e+00, -2.7348e+00, -3.0530e+00, -1.7157e+00,
         -2.2521e+00],
        [-1.7671e+00, -2.2295e+00, -2.7102e+00, -5.5151e-02, -1.9916e+00,
         -1.7636e+00],
        [-1.5940e-01, -8.0887e-01, -1.3068e+00, -2.4865e-02, -7.3319e-01,
         -2.7825e-02],
        [-2.5736e+00, -5.1591e-03, -2.1895e+00, -1.2204e+00, -2.8789e+00,
         -1.7851e+00],
        [-1.9523e+00, -1.0565e+00, -2.3523e+00, -4.6164e-03, -3.2435e+00,
         -7.3774e-01],
        [-2.2420e+00, -1.1961e-02, -4.6703e-01, -1.7726e+00, -2.1748e+00,
         -6.1458e-01],
        [-4.0607e+00, -3.7748e+00, -2.1724e+00, -1.4194e+00, -5.4327e+00,
         -2.1575e+00],
        [-2.8159e+00, -2.9471e+00, -1.5308e+00, -1.3164e-01, -3.4341e+00,
         -2.0522e+00],
        [-2.3378e+00, -1.3398e-01, -2.4938e-01, -1.7900e+00, -4.5126e-01,
         -3.5535e-01],
        [-2.0478e+00, -1.1636e-02, -2.4857e+00, -1.6299e+00, -1.5521e-04,
         -4.3218e-01],
        [-2.8172e+00, -7.1877e-02, -2.4986e-01, -1.1107e-01, -1.2034e-01,
         -6.2232e-01],
        [-2.0199e+00, -5.8740e+00, -1.0658e+00, -1.2127e+00, -6.5239e-01,
         -2.1622e-01],
        [-3.4840e+00, -2.8688e+00, -1.6737e+00, -2.3476e+00, -7.0934e-01,
         -1.0883e+00],
        [-5.6118e+00, -9.4187e-01, -4.5207e+00, -3.0546e+00, -1.4413e+00,
         -1.1025e+00],
        [-3.9191e+00, -2.8129e-01, -2.1087e+00, -3.1972e-01, -4.8347e+00,
         -6.6327e-01],
        [-4.2350e+00, -1.0784e+00, -4.0561e+00, -6.0705e-01, -2.7495e+00,
         -1.4776e+00]], device='cuda:0')
tensor([ -4.2701, -10.2217,  -8.0446, -10.6436, -14.3121, -10.5173,  -3.0610,
        -10.6526,  -9.3468,  -7.2830, -19.0177, -12.9118,  -5.3177,  -6.6074,
         -3.9927, -11.0409, -12.1718, -16.6727, -12.1267, -14.2036],
       device='cuda:0')
tensor([-3.9687, -1.3566, -2.3253, -1.1147, -0.2756, -1.8300, -4.6926, -2.5194,
        -1.2081, -3.0785,  1.2146, -0.9376, -3.8407, -3.3996, -6.6579, -1.6392,
         0.1983,  0.6812, -0.7549, -1.7587], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.5682, -1.2048, -1.5713, -2.3123, -1.7629, -1.3538],
        [-6.0401, -3.1231, -1.8533, -1.2259, -2.8559, -2.6828],
        [-5.6848, -2.0478, -2.8158, -0.9082, -1.8998, -1.3093],
        [-1.2601, -1.1969, -2.0356, -0.9793, -1.4223, -1.0494],
        [-5.3468, -1.3989, -1.6262, -1.5578, -1.2959, -1.0511],
        [-5.2670, -1.4278, -1.4774, -1.4983, -1.4429, -1.2333],
        [-4.8396, -1.9022, -2.7144, -1.9225, -1.1719, -1.0162],
        [-1.9679, -1.6431, -1.8725, -1.9983, -1.4459, -1.5737],
        [-4.0831, -1.5255, -1.9122, -3.4255, -1.8406, -1.4053],
        [-3.6059, -0.9528, -2.5361, -2.5463, -2.3946, -1.4355],
        [-8.5165, -2.6571, -2.5615, -1.3754, -2.1021, -1.4156],
        [-3.1312, -2.8548, -2.7687, -4.3224, -2.1824, -1.7007],
        [-1.2101, -1.9520, -2.0851, -1.3214, -1.6152, -1.1903],
        [-3.6141, -1.3662, -4.3833, -1.2677, -2.4552, -3.5670],
        [-3.0963, -1.2367, -1.5897, -1.2929, -1.3551, -1.3401],
        [-2.0696, -1.4188, -1.4109, -3.6358, -1.7181, -1.0039],
        [-4.5186, -1.9225, -2.1427, -2.5716, -2.4577, -1.9690],
        [-7.1596, -3.5203, -4.8788, -1.1999, -2.0749, -1.4010],
        [-1.3961, -1.5330, -2.0640, -4.0921, -1.2448, -1.1021],
        [-2.7029, -1.4806, -2.4664, -1.4899, -1.3693, -0.8587]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.7733, -17.7811, -14.6657,  -7.9436, -12.2767, -12.3468, -13.5669,
        -10.5013, -14.1923, -13.4711, -18.6282, -16.9602,  -9.3741, -16.6534,
         -9.9108, -11.2572, -15.5821, -20.2345, -11.4322, -10.3677],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.7337e+00, -1.1074e-01, -1.5108e+00, -2.1975e+00, -2.0667e+00,
         -4.1875e-01],
        [-5.0553e+00, -4.2829e+00, -1.8147e+00, -5.8628e-01, -4.4172e+00,
         -3.9734e+00],
        [-4.1184e+00, -1.5383e+00, -2.5724e+00, -2.9183e-02, -1.8586e+00,
         -1.8153e-02],
        [-8.8631e-01, -3.7895e-02, -1.8354e+00, -2.5617e-01, -1.1121e+00,
         -1.8895e-02],
        [-4.7832e+00, -7.1393e-02, -9.0921e-01, -7.6118e-01, -1.0890e-01,
         -3.1840e-02],
        [-3.3744e+00, -3.4261e-03, -8.5515e-01, -6.8542e-01, -2.4283e-02,
         -1.4076e-01],
        [-3.5893e+00, -2.2808e+00, -3.0284e+00, -1.4631e+00, -8.3900e-02,
         -1.1885e-01],
        [-2.7496e+00, -1.6887e+00, -1.9178e+00, -1.3961e+00, -9.6784e-01,
         -7.4988e-01],
        [-3.9825e+00, -6.9255e-01, -1.9298e+00, -3.8045e+00, -1.6635e+00,
         -9.5018e-01],
        [-4.1614e+00, -1.2343e-01, -3.0408e+00, -1.8404e+00, -2.7273e+00,
         -1.3501e+00],
        [-5.5136e+00, -1.8702e+00, -3.5877e+00, -2.2381e-01, -2.5591e+00,
         -1.2133e+00],
        [-2.2138e+00, -3.2131e+00, -4.0420e+00, -2.8327e+00, -2.7893e+00,
         -1.5570e+00],
        [-6.4023e-01, -2.3161e+00, -2.0671e+00, -3.4493e-01, -1.0648e+00,
         -1.0991e-03],
        [-4.2302e+00, -1.9670e-02, -5.4844e+00, -1.9468e-02, -3.1809e+00,
         -2.7125e+00],
        [-2.9021e+00, -1.0456e-01, -6.0959e-01, -3.7942e-01, -1.0971e+00,
         -7.9482e-01],
        [-1.9553e+00, -2.2520e-02, -8.9303e-01, -4.0734e+00, -1.8931e+00,
         -9.5583e-03],
        [-3.6332e+00, -1.1684e+00, -2.9641e+00, -3.1600e+00, -3.4108e+00,
         -1.9995e+00],
        [-4.7908e+00, -3.2931e+00, -3.9168e+00, -2.9220e-01, -2.2845e+00,
         -3.8452e-01],
        [-5.3997e-02, -1.6654e+00, -1.8974e+00, -3.2102e+00, -1.8723e-01,
         -2.9310e-01],
        [-2.8091e+00, -4.7407e-01, -2.7858e+00, -6.6322e-01, -2.0621e-01,
         -5.4210e-03]], device='cuda:0')
tensor([-10.0382, -20.1299, -10.1350,  -4.1468,  -6.6658,  -5.0834, -10.5643,
         -9.4700, -13.0231, -13.2434, -14.9677, -16.6478,  -6.4342, -15.6472,
         -5.8876,  -8.8469, -16.3360, -14.9619,  -7.3073,  -6.9438],
       device='cuda:0')
tensor([-2.7351,  2.3487, -4.5307, -3.7968, -5.6109, -7.2633, -3.0026, -1.0313,
        -1.1692, -0.2276, -3.6606, -0.3124, -2.9399, -1.0062, -4.0232, -2.4102,
         0.7539, -5.2726, -4.1249, -3.4239], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.9619, -1.5619, -1.6551, -1.7058, -1.3940, -1.3400],
        [-0.9542, -1.9055, -1.3832, -1.7047, -1.6543, -1.3216],
        [-5.3207, -3.1888, -1.6125, -2.8604, -1.9077, -0.9545],
        [-2.5234, -2.7368, -1.5572, -1.1914, -1.6686, -1.2518],
        [-1.2738, -1.6128, -1.5563, -1.9071, -1.3833, -1.0545],
        [-5.1571, -1.3070, -3.6079, -1.1928, -1.2743, -1.4818],
        [-2.6181, -1.6118, -1.2791, -1.3927, -1.7590, -1.1990],
        [-4.3734, -1.0140, -1.8010, -1.3457, -1.4432, -1.1235],
        [-2.7125, -1.7879, -2.3436, -1.2767, -2.0629, -1.1203],
        [-1.6632, -1.3239, -2.1776, -1.9655, -1.1160, -1.3656],
        [-1.4997, -2.6630, -2.7000, -1.5606, -1.2489, -1.4791],
        [-4.4613, -1.3778, -2.5167, -1.3542, -1.3767, -0.9738],
        [-2.1447, -1.0766, -1.9278, -1.0663, -1.4145, -1.8224],
        [-2.2835, -0.9139, -1.2978, -1.1550, -1.3096, -1.3832],
        [-1.5908, -1.4175, -1.4596, -1.1859, -1.5683, -1.9341],
        [-4.7458, -3.9154, -1.8951, -1.1029, -1.8219, -1.5258],
        [-4.0996, -3.0866, -2.3612, -1.0300, -3.1265, -2.2588],
        [-1.8063, -1.1932, -1.3681, -0.9956, -1.4967, -1.1261],
        [-6.9525, -1.2748, -2.3898, -1.3014, -3.6133, -2.4480],
        [-2.3771, -1.3801, -1.7181, -1.6610, -1.9677, -1.2018]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.6187,  -8.9235, -15.8446, -10.9292,  -8.7877, -14.0210,  -9.8596,
        -11.1009, -11.3038,  -9.6119, -11.1513, -12.0606,  -9.4523,  -8.3429,
         -9.1561, -15.0070, -15.9627,  -7.9860, -17.9797, -10.3058],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.7668e+00, -5.6991e-01, -1.1530e+00, -1.2554e+00, -3.9161e-01,
         -2.3168e-03],
        [-1.4571e-02, -1.1421e+00, -7.1118e-01, -1.0601e+00, -1.4986e+00,
         -7.8968e-01],
        [-3.2548e+00, -3.4993e+00, -1.5527e+00, -3.0470e+00, -1.7635e+00,
         -5.4975e-02],
        [-1.4178e+00, -2.7655e+00, -7.3713e-01, -1.1764e-01, -9.7948e-01,
         -1.2207e-04],
        [-1.0245e+00, -1.3879e+00, -1.1242e+00, -1.4563e+00, -4.9797e-01,
         -2.2122e-02],
        [-4.3982e+00, -3.7563e-02, -4.3273e+00, -3.8776e-02, -3.0329e-02,
         -8.8233e-01],
        [-2.1693e+00, -7.4219e-01, -1.3682e-01, -5.3228e-01, -1.8552e+00,
         -1.2709e-01],
        [-3.0910e+00, -1.3602e-04, -1.8620e+00, -5.2576e-01, -1.0173e+00,
         -4.1988e-02],
        [-1.3837e+00, -1.0114e+00, -2.7443e+00, -5.7870e-01, -1.8384e+00,
         -1.1809e-01],
        [-4.2695e-01, -2.4677e-01, -2.2436e+00, -1.4455e+00, -1.3616e-02,
         -7.0854e-01],
        [-1.3913e+00, -3.5285e+00, -2.7888e+00, -9.1760e-01, -4.1497e-01,
         -1.0122e+00],
        [-3.9669e+00, -3.3408e-02, -3.1664e+00, -3.2998e-01, -7.9493e-01,
         -2.5153e-05],
        [-1.8389e+00, -1.2159e-02, -1.6809e+00, -2.8125e-02, -1.2277e+00,
         -2.1542e+00],
        [-2.6355e+00, -1.4100e-01, -3.9610e-01, -1.1465e-01, -5.3813e-01,
         -8.5851e-01],
        [-1.7134e+00, -1.6244e-01, -8.9374e-01, -3.5033e-01, -1.1818e+00,
         -1.4157e+00],
        [-3.4205e+00, -4.6155e+00, -2.0693e+00, -1.2234e-01, -1.5537e+00,
         -6.3369e-01],
        [-2.8655e+00, -2.7812e+00, -2.3815e+00, -9.2684e-02, -3.7318e+00,
         -2.4530e+00],
        [-2.3740e+00, -6.1616e-02, -7.3055e-01, -1.7628e-02, -6.1312e-01,
         -1.0056e-01],
        [-6.1097e+00, -4.6783e-02, -3.0232e+00, -1.9540e-01, -4.4923e+00,
         -2.1034e+00],
        [-1.8916e+00, -4.8183e-02, -1.0046e+00, -1.2951e+00, -2.7338e+00,
         -1.9213e-01]], device='cuda:0')
tensor([ -5.1390,  -5.2162, -13.1722,  -6.0177,  -5.5130,  -9.7145,  -5.5629,
         -6.5382,  -7.6747,  -5.0850, -10.0534,  -8.2917,  -6.9420,  -4.6839,
         -5.7174, -12.4150, -14.3058,  -3.8975, -15.9708,  -7.1655],
       device='cuda:0')
tensor([-4.4796, -3.7073, -2.6724, -4.9116, -3.2747, -4.3065, -4.2967, -4.5627,
        -3.6291, -4.5269, -1.0979, -3.7689, -2.5103, -3.6590, -3.4387, -2.5920,
        -1.6569, -4.0885, -2.0089, -3.1403], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-5.1227, -3.4286, -2.1502, -1.1275, -2.8721, -1.9695],
        [-2.3259, -4.1817, -3.0217, -3.0073, -1.8999, -1.3079],
        [-1.5132, -2.3161, -2.2000, -1.7135, -1.2561, -1.2886],
        [-3.0434, -1.3892, -1.7845, -1.2842, -1.4251, -0.9683],
        [-3.9639, -1.1812, -2.9366, -1.0913, -1.2575, -1.3048],
        [-4.0043, -1.3667, -1.1425, -1.1664, -1.9836, -1.2903],
        [-7.0139, -1.7100, -2.9954, -1.1989, -2.4367, -1.1726],
        [-5.8719, -3.7107, -2.3844, -2.1399, -2.7428, -1.5461],
        [-1.3843, -1.2376, -1.6873, -1.3585, -1.5029, -1.3415],
        [-2.1015, -1.6284, -2.8805, -1.2970, -1.4083, -2.2186],
        [-6.7743, -1.3081, -2.8563, -1.2631, -1.6906, -2.0158],
        [-2.2020, -1.7309, -1.7648, -3.1812, -1.9475, -1.5526],
        [-4.2161, -2.5391, -2.5694, -1.7833, -1.9252, -1.6511],
        [-3.6452, -4.8943, -1.9501, -1.6014, -2.3233, -1.3264],
        [-1.1803, -1.1668, -2.5008, -1.2120, -1.2211, -1.1681],
        [-4.0957, -1.3777, -2.0150, -1.3628, -1.2983, -1.4216],
        [-1.2625, -1.3749, -1.9766, -1.2393, -1.3369, -1.1787],
        [-2.1413, -1.1321, -4.3058, -1.0933, -1.2508, -1.3497],
        [-2.2731, -1.1680, -1.2926, -1.0711, -1.8405, -3.8307],
        [-4.5505, -2.0861, -1.6701, -1.1627, -1.7818, -1.3870]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-16.6706, -15.7443, -10.2874,  -9.8947, -11.7354, -10.9538, -16.5275,
        -18.3959,  -8.5121, -11.5343, -15.9082, -12.3792, -14.6841, -15.7406,
         -8.4491, -11.5711,  -8.3690, -11.2729, -11.4759, -12.6380],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.3996e+00, -6.0144e+00, -3.1809e+00, -2.8691e-01, -4.4684e+00,
         -1.8106e+00],
        [-1.3342e+00, -3.8363e+00, -3.3031e+00, -2.4652e+00, -1.9254e+00,
         -5.5582e-01],
        [-1.5280e+00, -3.1040e+00, -1.9417e+00, -1.2581e+00, -1.5395e-01,
         -2.1197e-01],
        [-3.0123e+00, -6.5297e-03, -1.4280e+00, -5.7542e-04, -7.4131e-01,
         -2.8633e-02],
        [-3.4134e+00, -3.2992e-01, -3.8494e+00, -4.0962e-02, -6.8153e-01,
         -4.5812e-01],
        [-4.3832e+00, -1.4059e+00, -1.7999e-02, -7.3878e-03, -2.1533e+00,
         -5.0193e-01],
        [-4.9070e+00, -6.0460e-01, -4.3879e+00, -3.1269e-01, -3.1271e+00,
         -4.5195e-02],
        [-4.6440e+00, -4.6759e+00, -2.8517e+00, -1.0048e+00, -4.6690e+00,
         -1.2061e-01],
        [-1.5486e+00, -3.2073e-01, -1.3216e+00, -4.4253e-01, -3.3878e-01,
         -2.5810e-01],
        [-2.5561e+00, -4.8285e-01, -3.8304e+00, -1.9388e-02, -5.2568e-01,
         -1.4710e+00],
        [-4.9007e+00, -5.0296e-02, -3.5948e+00, -3.1140e-01, -2.0757e+00,
         -2.2496e+00],
        [-1.7092e+00, -8.3961e-01, -1.7867e+00, -3.6536e+00, -2.6176e+00,
         -1.3631e+00],
        [-2.7548e+00, -2.7736e+00, -4.4608e+00, -1.8441e+00, -1.8399e+00,
         -1.0773e+00],
        [-2.7885e+00, -5.3046e+00, -1.7898e+00, -4.9762e-01, -3.0587e+00,
         -1.5580e-01],
        [-1.3368e-01, -5.1029e-03, -2.7052e+00, -3.8104e-01, -4.6180e-02,
         -3.0143e-01],
        [-3.8117e+00, -8.9035e-01, -1.8889e+00, -2.9289e-01, -2.4214e-01,
         -9.7499e-01],
        [-8.1003e-01, -5.1685e-02, -1.9548e+00, -3.7672e-02, -2.9157e-01,
         -4.4555e-01],
        [-2.5603e+00, -7.2181e-01, -5.1495e+00, -2.9337e-04, -4.3069e-01,
         -4.7650e-01],
        [-2.7565e+00, -2.1069e-01, -8.6211e-01, -3.0148e-02, -1.7202e+00,
         -3.1484e+00],
        [-3.3176e+00, -1.4200e+00, -8.1426e-01, -1.6344e-04, -1.1599e+00,
         -5.3271e-02]], device='cuda:0')
tensor([-19.1608, -13.4201,  -8.1977,  -5.2173,  -8.7733,  -8.4698, -13.3846,
        -17.9660,  -4.2303,  -8.8853, -13.1825, -11.9698, -14.7505, -13.5950,
         -3.5726,  -8.1009,  -3.5913,  -9.3391,  -8.7280,  -6.7652],
       device='cuda:0')
tensor([ 2.4902, -2.3242, -2.0897, -4.6774, -2.9621, -2.4840, -3.1429, -0.4299,
        -4.2817, -2.6489, -2.7258, -0.4094,  0.0664, -2.1455, -4.8765, -3.4702,
        -4.7777, -1.9338, -2.7478, -5.8729], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.4725, -1.7784, -1.8516, -1.3766, -1.8743, -1.6298],
        [-1.2926, -1.2959, -1.3658, -1.3944, -1.7640, -1.3272],
        [-4.4387, -1.2751, -2.1708, -1.3577, -2.3179, -2.2078],
        [-4.6863, -1.3873, -3.3403, -1.3267, -2.7743, -1.0334],
        [-1.7856, -1.3189, -1.4646, -2.5179, -2.3883, -1.3973],
        [-4.4991, -2.2498, -2.4632, -3.0215, -1.9451, -1.9826],
        [-5.1780, -2.8649, -5.4448, -1.1423, -1.8166, -1.3637],
        [-1.2335, -1.1479, -1.8195, -2.6299, -1.5279, -1.0859],
        [-4.6959, -1.5025, -2.3567, -2.0247, -1.4696, -0.9898],
        [-4.8559, -1.3593, -1.3083, -1.3642, -1.5925, -1.2951],
        [-2.8986, -1.3020, -1.2790, -1.1066, -1.2692, -1.2506],
        [-1.4791, -1.9897, -2.1573, -3.5737, -1.8240, -1.2376],
        [-1.0029, -1.5021, -2.2130, -4.1785, -1.2078, -1.0649],
        [-2.1626, -1.6742, -4.2801, -3.9662, -2.6940, -1.4239],
        [-2.3887, -1.1814, -1.6757, -1.2366, -3.6171, -1.9513],
        [-5.9651, -2.6107, -2.3778, -1.8834, -1.6608, -1.3368],
        [-3.6592, -3.0744, -2.3200, -2.1472, -2.3539, -2.1921],
        [-1.2199, -1.4441, -1.9132, -0.9781, -1.3407, -1.2072],
        [-2.2117, -1.2883, -2.3717, -1.3632, -1.4456, -1.3719],
        [-3.6040, -1.3958, -1.8507, -1.2729, -1.9535, -1.2287]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.9832,  -8.4401, -13.7681, -14.5484, -10.8726, -16.1612, -17.8104,
         -9.4447, -13.0391, -11.7753,  -9.1059, -12.2614, -11.1692, -16.2011,
        -12.0507, -15.8346, -15.7468,  -8.1032, -10.0524, -11.3056],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.1334e+00, -9.0950e-01, -1.4198e+00, -5.4670e-01, -1.8554e+00,
         -1.2927e+00],
        [-3.3195e-01, -5.0086e-01, -5.0763e-01, -3.4359e-01, -1.2445e+00,
         -2.9959e-01],
        [-4.8589e+00, -7.2287e-02, -2.3375e+00, -2.9201e-01, -2.7342e+00,
         -1.4035e+00],
        [-3.6347e+00, -3.4645e-01, -4.3405e+00, -4.3318e-01, -4.5502e+00,
         -1.3822e-01],
        [-1.4373e+00, -3.6794e-02, -1.2224e+00, -2.5101e+00, -3.4624e+00,
         -4.8784e-01],
        [-3.6248e+00, -2.1725e+00, -4.0009e+00, -3.8483e+00, -1.8616e+00,
         -2.1179e+00],
        [-3.2803e+00, -2.2618e+00, -4.5212e+00, -1.8025e-01, -1.3962e+00,
         -1.7163e-01],
        [-8.1163e-02, -1.1053e-01, -1.4076e+00, -2.0773e+00, -1.3629e+00,
         -3.6144e-01],
        [-4.1190e+00, -1.4328e-02, -2.8856e+00, -2.0195e+00, -1.1035e+00,
         -9.0969e-02],
        [-2.9466e+00, -6.4731e-05, -3.8409e-01, -3.3510e-01, -1.6865e-01,
         -5.6429e-02],
        [-1.7464e+00, -7.9729e-01, -7.2015e-01, -1.6204e-01, -5.9605e-07,
         -1.2195e+00],
        [-3.6142e-01, -2.7954e+00, -2.6885e+00, -2.9693e+00, -1.8918e+00,
         -3.4901e-01],
        [-2.7618e-02, -1.1953e+00, -2.3010e+00, -4.0172e+00, -1.5955e-01,
         -2.7662e-02],
        [-2.1958e+00, -9.2789e-01, -5.5403e+00, -4.7635e+00, -3.0543e+00,
         -8.4279e-01],
        [-2.9181e+00, -1.1087e+00, -1.3429e+00, -2.3106e-01, -4.3377e+00,
         -2.2593e+00],
        [-3.9369e+00, -2.0284e+00, -3.2559e+00, -1.6014e+00, -9.7654e-01,
         -1.1239e+00],
        [-2.6381e+00, -3.5141e+00, -2.6853e+00, -9.8357e-01, -3.5020e+00,
         -2.8002e+00],
        [-4.8851e-01, -1.1348e+00, -1.6958e+00, -4.4341e-03, -2.4638e-02,
         -2.1328e-02],
        [-2.7295e+00, -4.9623e-02, -2.5320e+00, -1.8907e-01, -4.5579e-01,
         -4.3876e-01],
        [-3.1545e+00, -1.1728e-01, -1.4579e+00, -3.0681e-01, -2.6839e+00,
         -3.3254e-01]], device='cuda:0')
tensor([ -7.1576,  -3.2282, -11.6985, -13.4432,  -9.1569, -17.6261, -11.8114,
         -5.4009, -10.2329,  -3.8909,  -4.6453, -11.0555,  -7.7283, -17.3247,
        -12.1977, -12.9230, -16.1233,  -3.3694,  -6.3947,  -8.0529],
       device='cuda:0')
tensor([-3.8255, -5.2119, -2.0696, -1.1052, -1.7157,  1.4648, -5.9990, -4.0438,
        -2.8062, -7.8844, -4.4606, -1.2059, -3.4409,  1.1236,  0.1470, -2.9116,
         0.3765, -4.7338, -3.6578, -3.2526], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.1433, -1.1135, -1.2751, -1.9425, -1.8071, -1.2728],
        [-4.3253, -2.2690, -1.7620, -1.2839, -1.9549, -1.3977],
        [-4.1390, -3.0768, -1.6643, -1.6095, -2.0296, -1.4894],
        [-1.3486, -1.1844, -1.2280, -1.0320, -1.5970, -1.2105],
        [-5.0442, -1.5037, -2.7433, -1.4307, -1.9590, -1.3082],
        [-3.7947, -1.4400, -2.6295, -2.0625, -2.2452, -1.1260],
        [-4.8320, -1.4190, -1.5721, -3.8555, -1.5617, -1.2863],
        [-4.0813, -1.6980, -1.9758, -2.6577, -2.1792, -2.1681],
        [-6.1159, -4.8981, -4.2884, -1.1049, -1.8314, -1.2883],
        [-1.3386, -1.0351, -1.2819, -0.7492, -1.4660, -1.2312],
        [-2.2062, -1.3810, -1.6922, -1.4789, -1.2821, -1.0092],
        [-7.0676, -1.3844, -1.4889, -1.3106, -1.3651, -1.4648],
        [-1.1693, -1.9214, -2.0917, -3.2230, -1.2473, -1.1424],
        [-3.2658, -1.9292, -1.7332, -4.0310, -2.0906, -2.6646],
        [-4.7711, -1.4114, -1.2171, -0.9299, -3.9742, -1.5865],
        [-5.0955, -1.9122, -3.4936, -2.0533, -1.9994, -1.9776],
        [-2.9082, -3.5827, -2.3024, -1.7743, -2.0694, -1.5308],
        [-2.0757, -1.1782, -1.8875, -0.9346, -1.5540, -1.2141],
        [-3.7701, -1.2991, -2.5099, -1.2452, -1.3199, -0.9961],
        [-2.7335, -1.7534, -1.9965, -1.4058, -1.8631, -1.0090]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.5542, -12.9928, -14.0085,  -7.6005, -13.9891, -13.2979, -14.5265,
        -14.7602, -19.5269,  -7.1020,  -9.0497, -14.0815, -10.7951, -15.7144,
        -13.8903, -16.5316, -14.1678,  -8.8442, -11.1402, -10.7613],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.6935e+00, -6.5376e-02, -6.1169e-01, -1.5961e+00, -2.0564e+00,
         -1.4297e-02],
        [-3.1458e+00, -2.0866e+00, -1.4887e+00, -6.4573e-01, -2.0299e+00,
         -4.0999e-01],
        [-3.1721e+00, -3.3414e+00, -8.4901e-01, -5.1541e-01, -2.3053e+00,
         -7.3199e-01],
        [-1.4042e+00, -1.4581e-01, -9.3239e-02, -4.3948e-03, -1.0342e+00,
         -3.2616e-04],
        [-5.5058e+00, -2.1992e-01, -3.4521e+00, -4.6489e-01, -2.0999e+00,
         -4.5976e-01],
        [-3.0935e+00, -1.2420e+00, -3.1799e+00, -2.0850e+00, -3.3876e+00,
         -3.3428e-02],
        [-3.8947e+00, -3.5236e-02, -1.3538e+00, -4.2100e+00, -1.5848e+00,
         -5.2210e-01],
        [-3.1407e+00, -2.2164e-01, -2.5087e+00, -3.3547e+00, -2.5187e+00,
         -2.6687e+00],
        [-4.2262e+00, -4.6377e+00, -3.5539e+00, -1.5634e-01, -1.4310e+00,
         -1.3721e-04],
        [-1.0896e+00, -5.1810e-02, -6.2073e-02, -4.3355e-02, -1.3821e+00,
         -4.0875e-01],
        [-2.5967e+00, -4.1535e-01, -1.0149e+00, -5.4605e-01, -7.6425e-02,
         -9.1712e-02],
        [-4.5052e+00, -3.9557e-02, -6.9586e-01, -1.6210e-01, -2.8544e-01,
         -5.8978e-01],
        [-6.5343e-01, -2.1257e+00, -2.4972e+00, -2.9269e+00, -2.7957e-01,
         -2.7735e-01],
        [-3.1491e+00, -1.4966e+00, -1.2867e+00, -3.6334e+00, -2.3968e+00,
         -2.7236e+00],
        [-6.6414e+00, -5.8682e-01, -1.9303e-01, -9.5412e-03, -4.2306e+00,
         -1.5577e+00],
        [-3.4840e+00, -1.0884e+00, -5.1140e+00, -2.1350e+00, -2.1375e+00,
         -1.7011e+00],
        [-2.2834e+00, -4.2528e+00, -3.4076e+00, -5.2615e-01, -1.8960e+00,
         -5.5411e-01],
        [-3.3727e+00, -1.1680e-02, -1.6034e+00, -1.1297e-01, -9.3827e-01,
         -2.5346e-02],
        [-4.0581e+00, -5.6637e-01, -2.8641e+00, -1.7973e-01, -3.5291e-01,
         -6.8575e-02],
        [-2.7068e+00, -5.4156e-01, -1.6574e+00, -4.0477e-01, -2.4147e+00,
         -2.2005e-01]], device='cuda:0')
tensor([ -8.0374,  -9.8067, -10.9152,  -2.6822, -12.2023, -13.0214, -11.6006,
        -14.4132, -14.0052,  -3.0376,  -4.7411,  -6.2780,  -8.7601, -14.6862,
        -13.2192, -15.6598, -12.9201,  -6.0645,  -8.0897,  -7.9452],
       device='cuda:0')
tensor([-3.5168, -3.1861, -3.0933, -4.9184, -1.7869, -0.2766, -2.9258, -0.3470,
        -5.5217, -4.0644, -4.3085, -7.8035, -2.0349, -1.0282, -0.6711, -0.8718,
        -1.2477, -2.7797, -3.0505, -2.8160], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[ -1.3317,  -1.2468,  -2.0068,  -1.4624,  -1.0418,  -1.1684],
        [ -2.4767,  -1.5052,  -2.0473,  -1.5847,  -1.6562,  -1.5472],
        [-16.6960,  -1.6863,  -1.9424,  -1.3120,  -1.4372,  -2.5607],
        [ -1.7665,  -1.7910,  -1.5436,  -2.9565,  -1.3302,  -1.0618],
        [ -3.9049,  -1.5057,  -4.3909,  -5.8606,  -2.0459,  -1.6839],
        [ -2.4052,  -1.0824,  -1.2715,  -0.9804,  -1.8654,  -1.7999],
        [ -4.1095,  -1.6408,  -2.3980,  -1.4400,  -2.8298,  -5.8329],
        [ -3.4731,  -3.2756,  -2.8171,  -2.0055,  -1.8494,  -1.5207],
        [ -1.3635,  -1.1201,  -1.1672,  -1.8926,  -1.4315,  -1.2009],
        [ -4.4265,  -1.6296,  -3.0442,  -1.2337,  -1.2181,  -0.9347],
        [ -2.1874,  -1.4734,  -1.9047,  -1.2340,  -2.2411,  -0.9956],
        [-14.9222,  -1.4004,  -3.1266,  -1.1672,  -1.6236,  -1.5369],
        [ -3.4980,  -1.3044,  -1.3087,  -1.2219,  -1.9760,  -1.7198],
        [ -4.6398,  -1.7942,  -2.4650,  -1.1404,  -1.9760,  -1.4896],
        [ -3.0011,  -5.7771,  -2.4180,  -2.1630,  -2.2751,  -1.4798],
        [ -1.2206,  -1.3106,  -1.5956,  -1.8458,  -1.3811,  -1.2527],
        [ -4.6758,  -1.3413,  -3.6555,  -1.5618,  -1.3203,  -1.2845],
        [ -3.6678,  -1.4096,  -1.4639,  -1.2519,  -1.3724,  -1.5029],
        [ -2.1593,  -2.4322,  -3.1859,  -1.8173,  -1.5327,  -3.8956],
        [ -1.4324,  -1.5902,  -2.9520,  -2.2476,  -1.3661,  -1.9644]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.2578, -10.8173, -25.6345, -10.4494, -19.3921,  -9.4048, -18.2509,
        -14.9414,  -8.1758, -12.4867, -10.0361, -23.7769, -11.0289, -13.5049,
        -17.1141,  -8.6063, -13.8391, -10.6684, -15.0230, -11.5528],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.2767e+00, -8.5216e-01, -1.8386e+00, -8.9063e-01, -3.6204e-04,
         -1.8656e-01],
        [-2.7650e+00, -4.6627e-01, -1.9727e+00, -8.1825e-01, -1.4269e+00,
         -5.4394e-01],
        [-6.5246e+00, -2.1088e+00, -2.3130e+00, -5.0081e-01, -5.9122e-03,
         -2.6007e+00],
        [-2.1229e+00, -2.0436e+00, -1.2742e+00, -2.9717e+00, -1.0794e+00,
         -2.0056e-01],
        [-4.2285e+00, -6.5597e-01, -4.7434e+00, -5.1060e+00, -1.9908e+00,
         -1.3402e+00],
        [-3.1334e+00, -8.4190e-01, -3.1794e-01, -3.7408e-03, -1.7692e+00,
         -1.9702e+00],
        [-2.4097e+00, -1.8326e-01, -3.8033e+00, -4.3224e-01, -4.4386e+00,
         -4.9076e+00],
        [-2.9867e+00, -3.8723e+00, -4.1713e+00, -7.6082e-01, -1.1776e+00,
         -5.3372e-01],
        [-1.5111e+00, -2.2933e-02, -1.8255e-03, -1.7590e+00, -5.6875e-01,
         -7.8889e-02],
        [-4.6947e+00, -5.7684e-01, -3.2227e+00, -6.2477e-02, -2.8636e-02,
         -1.1741e-01],
        [-2.1943e+00, -9.4444e-02, -1.7662e+00, -8.7325e-02, -2.8457e+00,
         -3.1185e-04],
        [-7.5055e+00, -5.1683e-01, -3.0146e+00, -4.0430e-02, -1.4217e+00,
         -1.1531e+00],
        [-3.2081e+00, -7.2106e-02, -8.7470e-01, -1.4466e-01, -2.5479e+00,
         -1.1860e+00],
        [-3.1953e+00, -7.7104e-01, -3.7906e+00, -1.8445e-01, -2.1083e+00,
         -9.0787e-01],
        [-2.2691e+00, -6.1400e+00, -2.7522e+00, -1.1776e+00, -3.1549e+00,
         -6.9609e-01],
        [-8.3923e-05, -6.6123e-01, -1.1587e+00, -1.2078e+00, -8.2150e-02,
         -2.6398e-02],
        [-4.8795e+00, -8.8948e-02, -4.9547e+00, -7.3396e-01, -4.7045e-02,
         -2.6699e-01],
        [-2.1533e+00, -5.1025e-02, -4.6041e-01, -1.3159e-01, -4.1109e-03,
         -8.2290e-01],
        [-1.4907e+00, -3.2784e+00, -3.6925e+00, -1.1650e+00, -1.2064e+00,
         -3.9656e+00],
        [-1.4453e+00, -1.2593e+00, -3.8679e+00, -1.5092e+00, -6.7622e-01,
         -1.7047e+00]], device='cuda:0')
tensor([ -5.0450,  -7.9931, -14.0537,  -9.6924, -18.0651,  -8.0364, -16.1747,
        -13.5025,  -3.9425,  -8.7028,  -6.9883, -13.6522,  -8.0334, -10.9576,
        -16.1900,  -3.1363, -10.9711,  -3.6234, -14.7987, -10.4626],
       device='cuda:0')
tensor([ -3.2128,  -2.8242, -11.5808,  -0.7571,  -1.3270,  -1.3684,  -2.0762,
         -1.4389,  -4.2333,  -3.7839,  -3.0477, -10.1248,  -2.9955,  -2.5473,
         -0.9241,  -5.4700,  -2.8680,  -7.0450,  -0.2244,  -1.0902],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([[-1.8741, -1.3120, -1.3167, -1.4098, -1.5894, -1.2105],
        [-2.2082, -1.6616, -1.9761, -1.2927, -1.6608, -1.9548],
        [-3.4909, -1.2146, -3.0791, -1.1575, -1.4608, -1.7887],
        [-9.9049, -1.3171, -1.0877, -1.8560, -1.6192, -1.1949],
        [-3.8743, -4.0360, -1.5650, -1.1650, -2.1080, -1.3472],
        [-3.5673, -4.7638, -2.0300, -1.3199, -2.2526, -1.6508],
        [-1.2134, -1.2767, -1.4697, -1.3691, -1.5682, -1.6268],
        [-1.4222, -1.6729, -2.2892, -1.3938, -1.2622, -0.9572],
        [-4.0410, -1.4242, -2.3119, -1.1224, -1.3410, -1.1939],
        [-6.9281, -1.2439, -2.0792, -1.2320, -1.8963, -2.3411],
        [-2.6035, -1.3212, -1.3262, -1.2083, -1.8154, -1.4531],
        [-3.6734, -2.9240, -1.6699, -1.1274, -1.6651, -1.5988],
        [-5.0091, -2.1673, -2.7639, -1.0450, -2.0158, -1.6746],
        [-1.3193, -1.3728, -1.4865, -1.4868, -1.4581, -1.3835],
        [-2.8809, -1.2603, -2.1960, -1.3361, -1.5873, -1.1066],
        [-5.4387, -1.4671, -2.3009, -1.1054, -2.0179, -2.2379],
        [-2.4349, -1.2506, -2.4060, -1.7884, -2.0590, -1.1405],
        [-4.6885, -2.3279, -1.7792, -1.4852, -2.0772, -1.6206],
        [-2.6431, -3.0036, -2.5553, -1.3684, -2.1739, -1.6687],
        [-1.0909, -1.2182, -1.3415, -1.1113, -1.8275, -1.2135]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.7126, -10.7541, -12.1916, -16.9797, -14.0955, -15.5843,  -8.5239,
         -8.9974, -11.4344, -15.7207,  -9.7277, -12.6586, -14.6757,  -8.5071,
        -10.3670, -14.5680, -11.0794, -13.9786, -13.4131,  -7.8030],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.7590e+00, -7.1216e-01, -3.9391e-01, -4.0135e-01, -7.2592e-01,
         -9.0133e-02],
        [-2.7716e+00, -3.7428e-01, -2.0155e+00, -1.8099e-01, -1.4360e+00,
         -1.0794e+00],
        [-2.6149e+00, -4.6379e-01, -3.9978e+00, -3.2201e-02, -1.2646e+00,
         -2.1325e+00],
        [-5.8020e+00, -3.4752e-03, -2.6923e-01, -1.5834e+00, -1.6078e+00,
         -7.0645e-02],
        [-3.0351e+00, -4.4166e+00, -6.7914e-01, -3.5529e-02, -2.4590e+00,
         -2.9160e-01],
        [-2.4581e+00, -5.4018e+00, -1.9656e+00, -2.2710e-01, -3.0156e+00,
         -1.3374e+00],
        [-1.0876e+00, -5.7832e-01, -8.9914e-01, -5.4092e-01, -7.4241e-01,
         -1.2131e+00],
        [-1.2926e+00, -5.6919e-01, -2.8645e+00, -2.5818e-01, -2.0883e-03,
         -5.0060e-02],
        [-3.4737e+00, -1.7476e-01, -2.8038e+00, -3.1083e-03, -9.5720e-01,
         -4.4017e-01],
        [-5.1982e+00, -5.3109e-01, -1.5840e+00, -4.2249e-02, -2.2444e+00,
         -2.6411e+00],
        [-2.8351e+00, -8.6214e-02, -8.7338e-01, -6.1607e-02, -1.8811e+00,
         -7.8503e-01],
        [-2.5336e+00, -3.4252e+00, -1.0644e+00, -2.3427e-01, -9.8038e-01,
         -8.7804e-01],
        [-3.7207e+00, -1.5205e+00, -2.9996e+00, -1.6202e-01, -2.2204e+00,
         -1.3280e+00],
        [-1.1449e+00, -8.9766e-01, -9.2487e-01, -6.7714e-01, -1.9398e-01,
         -4.4670e-01],
        [-3.0694e+00, -6.5752e-02, -2.3509e+00, -1.6694e-01, -1.2822e+00,
         -3.2056e-01],
        [-4.2525e+00, -1.3099e+00, -2.7546e+00, -1.7691e-02, -2.9574e+00,
         -2.6684e+00],
        [-2.3697e+00, -9.2257e-02, -3.3430e+00, -1.2418e+00, -2.6771e+00,
         -3.9819e-02],
        [-3.5755e+00, -2.1433e+00, -1.2168e+00, -1.2738e+00, -2.4329e+00,
         -8.9503e-01],
        [-1.5805e+00, -3.0783e+00, -3.0250e+00, -2.8893e-01, -2.8977e+00,
         -1.3879e+00],
        [-1.4824e-01, -3.9952e-01, -4.4548e-01, -1.6457e-03, -1.5559e+00,
         -1.4015e-01]], device='cuda:0')
tensor([ -5.0824,  -7.8577, -10.5057,  -9.3365, -10.9169, -14.4056,  -5.0616,
         -5.0366,  -7.8527, -12.2411,  -6.5224,  -9.1160, -11.9512,  -4.2853,
         -7.2558, -13.9604,  -9.7637, -11.5373, -12.2585,  -2.6910],
       device='cuda:0')
tensor([-3.6302, -2.8964, -1.6859, -7.6432, -3.1786, -1.1788, -3.4623, -3.9609,
        -3.5817, -3.4797, -3.2053, -3.5426, -2.7246, -4.2218, -3.1112, -0.6075,
        -1.3157, -2.4413, -1.1545, -5.1120], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.0047, -2.8204, -2.9869, -1.8302, -2.1028, -1.6851],
        [-1.1494, -1.3514, -2.4333, -1.1129, -1.3598, -1.3232],
        [-2.3957, -1.4982, -4.2061, -1.2565, -1.3182, -0.9157],
        [-5.0062, -1.3054, -2.2416, -1.1125, -1.6603, -1.3882],
        [-4.2716, -1.5670, -1.4604, -1.6650, -1.6559, -2.0224],
        [-1.2935, -1.1930, -2.2167, -1.1394, -1.7536, -1.9554],
        [-5.5027, -3.3299, -2.2400, -1.1520, -2.1650, -1.3503],
        [-2.2815, -2.7428, -1.6272, -1.3114, -2.1759, -1.3616],
        [-1.1931, -1.6370, -2.3155, -1.5378, -1.4452, -1.1367],
        [-3.9659, -1.6164, -2.5016, -1.6156, -1.6688, -0.9691],
        [-5.4863, -1.1547, -1.8305, -1.1648, -1.6380, -1.5305],
        [-3.6471, -1.3090, -1.2641, -1.3004, -2.0065, -1.3883],
        [-5.5831, -2.2115, -2.3622, -1.2509, -2.8610, -1.5305],
        [-2.6621, -3.0982, -2.4069, -2.3300, -2.3288, -1.6159],
        [-3.2339, -1.1136, -1.6984, -0.8968, -1.6286, -1.4087],
        [-2.3442, -1.2694, -2.3906, -1.2390, -1.5470, -1.4061],
        [-1.9288, -1.4138, -1.7120, -1.2245, -1.9843, -1.1460],
        [-1.9275, -1.2905, -1.5304, -1.5991, -1.7742, -1.4286],
        [-1.0642, -1.1518, -1.4253, -1.1885, -1.4818, -1.2949],
        [-4.4143, -2.9019, -2.0009, -1.1157, -2.4147, -1.6791]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-14.4301,  -8.7299, -11.5905, -12.7141, -12.6423,  -9.5516, -15.7398,
        -11.5005,  -9.2653, -12.3374, -12.8047, -10.9154, -15.7992, -14.4419,
         -9.9800, -10.1963,  -9.4094,  -9.5503,  -7.6065, -14.5265],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.9654e+00, -2.9950e+00, -4.1027e+00, -6.9635e-01, -2.5094e+00,
         -1.4060e+00],
        [-1.7823e-01, -8.5822e-01, -2.8063e+00, -5.2326e-02, -3.3859e-02,
         -3.2575e-01],
        [-2.8497e+00, -2.3446e-01, -5.3969e+00, -1.6534e-04, -1.1864e-01,
         -4.7779e-02],
        [-4.1486e+00, -7.4184e-04, -2.6078e+00, -5.9590e-02, -1.9377e+00,
         -9.6812e-01],
        [-3.5779e+00, -3.4722e-01, -4.4783e-01, -1.0881e+00, -1.6958e+00,
         -1.9537e+00],
        [-1.0969e+00, -2.0010e-01, -2.5944e+00, -1.6293e-01, -1.6870e+00,
         -2.0432e+00],
        [-3.9375e+00, -4.9146e+00, -3.5640e+00, -7.0672e-02, -2.6287e+00,
         -7.1314e-02],
        [-1.2529e+00, -2.4136e+00, -8.2927e-01, -4.4197e-01, -2.6361e+00,
         -6.6373e-01],
        [-6.9977e-01, -1.6037e+00, -2.6084e+00, -7.9386e-01, -3.8935e-01,
         -7.6758e-02],
        [-3.9342e+00, -1.9210e-01, -3.4359e+00, -8.2832e-01, -1.4813e+00,
         -9.7542e-02],
        [-4.5608e+00, -2.9498e-01, -1.5405e+00, -4.9663e-04, -1.7692e+00,
         -1.1983e+00],
        [-3.5617e+00, -2.2686e-01, -1.4726e-03, -2.6334e-01, -2.4678e+00,
         -5.9816e-01],
        [-4.0779e+00, -1.9314e+00, -3.3500e+00, -5.5227e-01, -3.9496e+00,
         -1.1177e+00],
        [-1.2879e+00, -3.6097e+00, -2.9274e+00, -1.1015e+00, -3.3600e+00,
         -8.5833e-01],
        [-4.4104e+00, -1.8660e-02, -1.1440e+00, -2.0264e-02, -1.3649e+00,
         -7.0149e-01],
        [-2.8535e+00, -1.1806e-02, -2.7693e+00, -7.3979e-03, -9.4997e-01,
         -7.7877e-01],
        [-1.7554e+00, -1.7643e-02, -9.6340e-01, -1.4499e-01, -2.8266e+00,
         -4.4134e-01],
        [-1.7830e+00, -1.0462e-01, -5.6181e-01, -9.9970e-01, -1.9949e+00,
         -7.0810e-01],
        [-5.1444e-01, -9.2139e-03, -9.9163e-01, -2.3646e-01, -8.5497e-01,
         -3.9405e-01],
        [-3.8747e+00, -3.3596e+00, -2.1885e+00, -7.2756e-03, -3.5582e+00,
         -8.6423e-01]], device='cuda:0')
tensor([-13.6748,  -4.2547,  -8.6477,  -9.7226,  -9.1106,  -7.7845, -15.1868,
         -8.2375,  -6.1718,  -9.9694,  -9.3644,  -7.1194, -14.9788, -13.1449,
         -7.6597,  -7.3707,  -6.1494,  -6.1521,  -3.0008, -13.8525],
       device='cuda:0')
tensor([-0.7553, -4.4752, -2.9427, -2.9916, -3.5317, -1.7671, -0.5530, -3.2629,
        -3.0935, -2.3680, -3.4403, -3.7960, -0.8204, -1.2970, -2.3202, -2.8255,
        -3.2600, -3.3982, -4.6058, -0.6740], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.5828, -3.4198, -2.8034, -1.7448, -2.9315, -2.4464],
        [-1.6111, -1.2436, -1.2470, -1.2621, -1.4269, -1.0922],
        [-1.8644, -1.3554, -3.5302, -1.1337, -1.1430, -1.0588],
        [-3.2840, -1.3219, -1.7404, -1.2883, -1.6834, -1.4972],
        [-0.9708, -1.8832, -1.6167, -2.3303, -1.5437, -1.2852],
        [-6.2088, -2.5015, -1.7604, -3.9398, -1.7997, -1.1518],
        [-4.2399, -4.1623, -2.3954, -1.0684, -1.8267, -1.4515],
        [-1.1354, -2.0151, -2.1398, -1.3721, -1.3751, -1.1938],
        [-3.4670, -1.4381, -2.6851, -1.2365, -1.8741, -1.1487],
        [-6.8742, -1.4987, -1.5029, -1.2786, -1.5325, -2.7595],
        [-0.9272, -1.7544, -1.6899, -1.4011, -1.1357, -1.0323],
        [-2.8024, -1.6870, -2.0824, -1.5206, -1.9282, -4.4376],
        [-2.8695, -0.9406, -1.3894, -1.1068, -1.8490, -1.5990],
        [-5.6010, -3.5090, -2.7306, -1.6412, -1.7543, -0.9854],
        [-3.1898, -2.9749, -1.9349, -2.7426, -1.9112, -1.8819],
        [-2.0458, -1.1216, -1.3190, -1.8233, -1.4623, -1.1797],
        [-5.4003, -1.1966, -2.7835, -1.3384, -1.8055, -2.1650],
        [-3.8185, -1.4238, -1.9590, -1.9187, -1.7826, -0.9986],
        [-1.5646, -1.2167, -1.7521, -2.6709, -1.5864, -1.2061],
        [-4.4173, -2.1613, -1.5293, -3.3227, -2.3694, -1.5121]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-17.9287,  -7.8829, -10.0855, -10.8153,  -9.6299, -17.3621, -15.1442,
         -9.2313, -11.8496, -15.4465,  -7.9406, -14.4581,  -9.7543, -16.2216,
        -14.6353,  -8.9516, -14.6893, -11.9012,  -9.9967, -15.3121],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.8323e+00, -2.8971e+00, -2.6555e+00, -9.5920e-01, -3.5019e+00,
         -2.7578e+00],
        [-2.1330e+00, -5.6380e-01, -7.6636e-03, -1.0355e+00, -6.8261e-01,
         -7.7693e-02],
        [-1.7320e+00, -2.8490e-01, -4.1825e+00, -2.1350e-02, -2.0167e-01,
         -2.0241e-01],
        [-2.7278e+00, -1.5479e-01, -1.2234e+00, -4.1865e-01, -1.6875e+00,
         -1.1851e+00],
        [-5.7578e-01, -1.2280e+00, -1.5541e+00, -2.2251e+00, -1.3691e+00,
         -5.0213e-01],
        [-4.4280e+00, -2.5632e+00, -1.8790e+00, -4.2140e+00, -1.4797e+00,
         -1.1703e-02],
        [-3.0497e+00, -3.9606e+00, -2.6764e+00, -2.7995e-03, -1.5931e+00,
         -9.0033e-01],
        [-2.4012e-02, -2.2474e+00, -2.1147e+00, -5.5072e-01, -3.5551e-01,
         -8.1346e-02],
        [-3.5359e+00, -1.1704e+00, -3.2525e+00, -3.8855e-02, -2.1067e+00,
         -2.3257e-01],
        [-3.5144e+00, -8.9413e-01, -1.3196e+00, -4.0598e-02, -9.7647e-01,
         -3.0109e+00],
        [-1.2490e-01, -1.8506e+00, -1.6173e+00, -5.9387e-01, -7.9472e-03,
         -1.6181e-01],
        [-3.4312e+00, -9.3542e-01, -2.0862e+00, -1.2911e+00, -1.6399e+00,
         -3.6079e+00],
        [-3.4455e+00, -1.6282e-01, -7.1556e-01, -1.9448e-01, -1.6336e+00,
         -1.5618e+00],
        [-4.0350e+00, -2.8883e+00, -3.5847e+00, -1.0653e+00, -1.3928e+00,
         -3.1660e-01],
        [-2.2613e+00, -3.1273e+00, -1.8410e+00, -1.7111e+00, -1.7325e+00,
         -1.9116e+00],
        [-3.0307e+00, -5.1951e-02, -2.9235e-01, -1.9023e+00, -8.3809e-01,
         -6.1115e-02],
        [-5.3456e+00, -2.7083e-02, -2.9355e+00, -3.5811e-01, -1.6660e+00,
         -1.4434e+00],
        [-3.2646e+00, -5.7738e-02, -1.5951e+00, -1.7978e+00, -1.9683e+00,
         -1.4512e-01],
        [-1.1281e+00, -7.0596e-04, -1.6676e+00, -2.3133e+00, -1.6938e+00,
         -2.0721e-01],
        [-3.0875e+00, -1.8609e+00, -9.2615e-01, -3.7165e+00, -2.8301e+00,
         -1.1577e+00]], device='cuda:0')
tensor([-15.6038,  -4.5002,  -6.6248,  -7.3973,  -7.4542, -14.5756, -12.1829,
         -5.3736, -10.3370,  -9.7561,  -4.3564, -12.9917,  -7.7137, -13.2827,
        -12.5848,  -6.1765, -11.7757,  -8.8287,  -7.0108, -13.5788],
       device='cuda:0')
tensor([-2.3250, -3.3827, -3.4607, -3.4180, -2.1757, -2.7865, -2.9613, -3.8577,
        -1.5125, -5.6903, -3.5841, -1.4664, -2.0406, -2.9389, -2.0505, -2.7750,
        -2.9136, -3.0725, -2.9860, -1.7333], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.3804, -1.5208, -1.9761, -2.8259, -1.4025, -0.8659],
        [-5.6780, -2.8616, -4.3024, -4.9796, -1.3924, -1.4399],
        [-1.4313, -1.8981, -1.8343, -2.0119, -1.5697, -1.0795],
        [-6.7939, -2.0746, -4.2186, -1.6477, -2.2177, -1.5052],
        [-3.1070, -1.0882, -2.4678, -1.1050, -3.4296, -2.8286],
        [-5.7622, -3.5685, -1.5703, -1.2456, -2.1836, -1.6606],
        [-3.6194, -2.2576, -2.1831, -3.0782, -2.0553, -2.1341],
        [-1.2807, -2.5397, -1.6304, -1.0164, -1.3904, -1.1852],
        [-1.9806, -1.5374, -3.2521, -1.3179, -1.4891, -1.0148],
        [-3.0817, -1.2846, -2.3949, -1.4029, -1.9252, -1.6815],
        [-2.2667, -1.1741, -2.4513, -1.2189, -1.9346, -2.5771],
        [-8.3029, -1.3968, -1.3004, -1.2163, -1.9122, -1.2260],
        [-3.7603, -2.1817, -1.7827, -1.2102, -1.9058, -1.3619],
        [-4.2933, -3.3703, -3.2034, -1.0460, -2.1492, -1.3565],
        [-1.6691, -1.2133, -1.1372, -1.0678, -2.2511, -1.8111],
        [-4.1498, -1.3306, -2.3782, -1.2007, -1.7450, -2.2354],
        [-7.1007, -1.5674, -2.2177, -1.2456, -1.6546, -1.0113],
        [-2.6677, -1.2528, -2.4220, -1.4417, -2.4315, -1.6577],
        [-1.7011, -1.1306, -1.4361, -1.1631, -1.9206, -1.2129],
        [-5.2734, -3.3341, -2.2734, -1.2352, -2.6999, -1.4487]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.9717, -20.6539,  -9.8247, -18.4578, -14.0261, -15.9907, -15.3277,
         -9.0428, -10.5921, -11.7708, -11.6226, -15.3545, -12.2027, -15.4187,
         -9.1496, -13.0395, -14.7972, -11.8734,  -8.5644, -16.2646],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-9.1105e-01, -3.9630e-01, -2.1509e+00, -2.4464e+00, -8.7674e-01,
         -2.2825e-02],
        [-4.0870e+00, -3.8329e+00, -5.3447e+00, -5.6583e+00, -8.2193e-01,
         -3.4879e-01],
        [-1.6637e+00, -1.9906e+00, -1.7277e+00, -1.4106e+00, -2.3287e+00,
         -3.3664e-02],
        [-5.5056e+00, -1.5811e+00, -4.7722e+00, -1.4927e+00, -2.4789e+00,
         -1.2283e+00],
        [-4.0201e+00, -6.9265e-01, -2.4520e+00, -1.9918e-01, -4.2369e+00,
         -3.4854e+00],
        [-4.3776e+00, -3.8257e+00, -6.7737e-01, -2.5429e-01, -2.7942e+00,
         -1.9082e+00],
        [-2.9054e+00, -2.0461e+00, -2.5045e+00, -1.9064e+00, -2.3626e+00,
         -2.7602e+00],
        [-1.1742e+00, -3.2706e+00, -1.1522e+00, -1.1131e-02, -3.5519e-01,
         -6.4765e-02],
        [-2.4190e+00, -1.9850e-01, -4.3283e+00, -7.0652e-02, -4.5864e-01,
         -9.5313e-02],
        [-3.3195e+00, -4.6760e-03, -3.1835e+00, -9.3851e-01, -2.1430e+00,
         -1.3370e+00],
        [-1.9690e+00, -1.5945e-01, -2.0901e+00, -3.5310e-02, -2.4653e+00,
         -3.4757e+00],
        [-6.2383e+00, -1.0161e+00, -8.8329e-01, -1.2071e-01, -2.1456e+00,
         -4.3370e-01],
        [-2.7436e+00, -1.4971e+00, -1.3936e+00, -2.6294e-03, -1.7084e+00,
         -1.0911e-01],
        [-3.8910e+00, -3.1585e+00, -4.1344e+00, -1.1316e-01, -2.7225e+00,
         -4.6457e-01],
        [-2.2894e+00, -3.7408e-04, -1.9257e-01, -5.7166e-02, -2.8254e+00,
         -1.5616e+00],
        [-4.5008e+00, -6.6732e-02, -2.8798e+00, -3.2001e-02, -1.4208e+00,
         -1.8013e+00],
        [-4.5077e+00, -1.7006e-02, -2.1763e+00, -2.5858e-02, -1.6279e+00,
         -6.7949e-06],
        [-2.3054e+00, -1.1748e-01, -2.0622e+00, -6.9588e-01, -3.5786e+00,
         -1.3892e+00],
        [-1.7127e+00, -3.2518e-02, -1.2431e+00, -2.5613e-02, -1.9784e+00,
         -4.8754e-01],
        [-3.7687e+00, -4.7936e+00, -3.6194e+00, -5.7393e-01, -4.3292e+00,
         -3.1123e-01]], device='cuda:0')
tensor([ -6.8042, -20.0937,  -9.1550, -17.0588, -15.0862, -13.8374, -14.4853,
         -6.0280,  -7.5704, -10.9261, -10.1948, -10.8378,  -7.4543, -14.4842,
         -6.9265, -10.7014,  -8.3547, -10.1488,  -5.4799, -17.3961],
       device='cuda:0')
tensor([-3.1674, -0.5602, -0.6698, -1.3990,  1.0601, -2.1534, -0.8424, -3.0148,
        -3.0217, -0.8447, -1.4278, -4.5167, -4.7484, -0.9345, -2.2230, -2.3382,
        -6.4425, -1.7246, -3.0846,  1.1315], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.0306, -1.3123, -1.4808, -2.2358, -1.4217, -1.3746],
        [-5.5856, -2.6288, -1.9782, -1.2844, -2.2325, -1.4243],
        [-2.4437, -3.5065, -2.1406, -1.0916, -2.0622, -1.7859],
        [-1.2113, -2.4347, -2.5672, -2.1378, -1.4491, -2.0122],
        [-3.6517, -1.4440, -1.8243, -1.6050, -1.7488, -0.9580],
        [-4.6213, -1.2555, -2.1616, -1.2145, -1.4390, -1.3200],
        [-1.8303, -1.1736, -1.2625, -1.6485, -2.4042, -1.5643],
        [-3.3750, -1.8370, -2.6842, -1.0724, -1.9148, -1.2490],
        [-3.9410, -2.4550, -1.9311, -1.3899, -1.9680, -1.4791],
        [-1.3949, -1.1953, -1.1763, -1.0023, -1.4507, -1.1577],
        [-3.2297, -1.3308, -3.0561, -1.3537, -2.0931, -2.3304],
        [-1.5361, -1.5771, -2.2137, -1.8069, -2.3143, -0.9126],
        [-5.0753, -1.2265, -1.9700, -4.3162, -1.4282, -1.5607],
        [-4.3278, -1.7844, -2.1083, -1.5854, -3.3284, -2.1855],
        [-5.7850, -1.7145, -1.8893, -0.9950, -1.9540, -1.4316],
        [-1.2282, -1.3530, -1.6970, -1.0663, -1.7857, -2.0107],
        [-2.0903, -1.2277, -2.0986, -1.3782, -1.8340, -1.4952],
        [-3.8657, -1.9402, -1.5929, -1.5630, -1.4841, -1.4109],
        [-1.6197, -1.4123, -1.3308, -2.0544, -1.2849, -1.7964],
        [-1.6148, -1.5350, -1.6661, -2.5005, -1.5006, -2.1946]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.8558, -15.1338, -13.0306, -11.8124, -11.2318, -12.0118,  -9.8833,
        -12.1324, -13.1641,  -7.3773, -13.3938, -10.3606, -15.5768, -15.3198,
        -13.7694,  -9.1410, -10.1240, -11.8568,  -9.4984, -11.0116],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.6527e+00, -4.3602e-02, -1.2382e+00, -2.1257e+00, -1.0361e+00,
         -7.9746e-01],
        [-3.7352e+00, -3.0612e+00, -2.3226e+00, -7.4087e-01, -2.9422e+00,
         -3.7076e-02],
        [-1.1743e+00, -3.5955e+00, -2.1842e+00, -7.5375e-02, -2.4650e+00,
         -1.7596e+00],
        [-4.6592e-03, -3.4477e+00, -3.0902e+00, -1.6829e+00, -3.7826e-01,
         -2.1116e+00],
        [-3.4996e+00, -6.3190e-02, -1.2896e+00, -9.2779e-01, -2.0721e+00,
         -2.8205e-01],
        [-3.5573e+00, -1.4680e-01, -2.4593e+00, -1.6721e-01, -1.0916e+00,
         -5.4281e-01],
        [-2.1437e+00, -3.2812e-02, -4.1175e-02, -8.7033e-01, -2.5663e+00,
         -1.0083e+00],
        [-2.4651e+00, -1.1054e+00, -3.4756e+00, -1.6507e-01, -1.8063e+00,
         -1.6072e-01],
        [-3.1135e+00, -2.6352e+00, -1.9090e+00, -1.4500e-01, -1.9086e+00,
         -2.9257e-01],
        [-1.5578e+00, -2.1202e-02, -1.0065e-02, -6.8628e-03, -2.3465e-01,
         -5.4377e-02],
        [-3.7035e+00, -1.1609e+00, -3.6724e+00, -4.6539e-01, -2.4309e+00,
         -2.0741e+00],
        [-1.5962e+00, -7.2074e-02, -2.4035e+00, -1.4170e+00, -3.7385e+00,
         -4.5179e-02],
        [-3.8948e+00, -6.1750e-05, -2.2921e+00, -4.7521e+00, -1.0006e+00,
         -7.5366e-01],
        [-3.5370e+00, -3.8007e-01, -3.0158e+00, -1.7645e+00, -5.1113e+00,
         -2.7300e+00],
        [-3.4853e+00, -9.3020e-01, -1.2157e+00, -1.6070e-01, -1.8957e+00,
         -2.4106e-01],
        [-8.0434e-01, -1.1990e-01, -1.1163e+00, -4.8415e-01, -1.8962e+00,
         -2.1435e+00],
        [-2.1326e+00, -3.0756e-02, -2.4780e+00, -3.4425e-01, -1.6947e+00,
         -1.0264e+00],
        [-2.3119e+00, -1.1095e+00, -1.2575e+00, -8.3257e-01, -1.0381e-01,
         -3.4225e-01],
        [-4.2685e-01, -1.0904e+00, -8.7494e-01, -1.7442e+00, -1.9091e-02,
         -1.8612e+00],
        [-2.0053e+00, -1.3497e+00, -1.4149e+00, -2.3540e+00, -1.2870e+00,
         -1.4996e+00]], device='cuda:0')
tensor([ -6.8937, -12.8392, -11.2539, -10.7154,  -8.1344,  -7.9651,  -6.6626,
         -9.1782, -10.0039,  -1.8849, -13.5071,  -9.2724, -12.6933, -16.5387,
         -7.9287,  -6.5643,  -7.7067,  -5.9575,  -6.0167,  -9.9104],
       device='cuda:0')
tensor([-2.9621, -2.2946, -1.7767, -1.0970, -3.0974, -4.0467, -3.2208, -2.9543,
        -3.1602, -5.4923,  0.1133, -1.0882, -2.8835,  1.2190, -5.8406, -2.5767,
        -2.4172, -5.8994, -3.4817, -1.1012], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.7867, -1.1292, -1.7772, -3.0144, -2.0248, -1.3745],
        [-3.5022, -2.3502, -1.5964, -2.1581, -2.0085, -1.4498],
        [-1.7156, -3.0245, -2.6952, -1.0809, -1.8775, -2.1730],
        [-1.3803, -1.3160, -2.1138, -1.3794, -1.5082, -1.5172],
        [-3.2373, -1.4294, -1.9764, -1.3677, -1.2548, -1.0288],
        [-9.1772, -1.7240, -2.2319, -1.4245, -1.3236, -1.7016],
        [-1.2367, -1.8356, -2.4483, -2.2358, -1.2791, -0.9498],
        [-2.5969, -1.4097, -2.5675, -2.5015, -2.1516, -1.6916],
        [-1.9094, -1.0339, -1.9004, -1.1517, -1.7689, -2.9912],
        [-6.2939, -2.2581, -1.5649, -1.2875, -2.2317, -1.3603],
        [-3.1224, -2.1861, -1.8180, -1.5066, -2.0594, -1.8632],
        [-1.1723, -1.1321, -1.6122, -0.8551, -1.5498, -1.3051],
        [-3.6525, -1.4269, -3.0594, -1.2771, -1.4605, -1.1147],
        [-6.3432, -1.4462, -2.4537, -1.3026, -1.7447, -1.4347],
        [-2.5361, -1.3702, -2.2935, -1.1839, -1.8523, -1.8980],
        [-4.6711, -1.2654, -1.2570, -1.1918, -4.1378, -2.0755],
        [-4.7886, -2.8470, -1.8215, -1.1322, -1.9993, -1.3579],
        [-4.5573, -3.6671, -2.0178, -1.1184, -2.4882, -1.8143],
        [-1.1689, -1.2198, -1.1718, -1.3490, -1.9729, -1.3233],
        [-1.5460, -1.4892, -1.6077, -1.4585, -1.2879, -1.0968]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.1069, -13.0653, -12.5667,  -9.2149, -10.2944, -17.5828,  -9.9853,
        -12.9187, -10.7555, -14.9965, -12.5557,  -7.6265, -11.9912, -14.7251,
        -11.1339, -14.5986, -13.9465, -15.6631,  -8.2058,  -8.4861],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.4769e+00, -3.9744e-04, -1.9019e+00, -3.2513e+00, -2.4738e+00,
         -5.5175e-01],
        [-2.1458e+00, -2.2902e+00, -4.1670e-01, -2.3880e+00, -2.0963e+00,
         -7.6864e-01],
        [-1.9171e-01, -3.1717e+00, -3.3643e+00, -4.0107e-03, -1.7020e+00,
         -2.7161e+00],
        [-1.6226e+00, -7.5321e-01, -2.2995e+00, -4.1548e-01, -6.7848e-01,
         -1.0621e+00],
        [-3.2581e+00, -7.5247e-01, -1.9242e+00, -2.7681e-01, -2.3227e-01,
         -3.1602e-01],
        [-5.2376e+00, -2.4877e+00, -2.7311e+00, -1.0047e+00, -2.9874e-01,
         -9.4912e-01],
        [-1.0643e+00, -2.3635e+00, -2.9165e+00, -1.8997e+00, -6.2529e-02,
         -1.4829e-02],
        [-3.4239e+00, -3.6660e-01, -3.1398e+00, -2.7627e+00, -1.9275e+00,
         -1.0679e+00],
        [-2.1251e+00, -7.2057e-01, -1.7944e+00, -1.6262e-01, -1.6852e+00,
         -4.0909e+00],
        [-4.6130e+00, -1.4359e+00, -6.9540e-01, -7.4931e-02, -2.9603e+00,
         -9.8944e-01],
        [-2.1555e+00, -1.8943e+00, -1.4284e+00, -3.6390e-01, -2.3842e+00,
         -2.0205e+00],
        [-8.1658e-01, -6.8438e-03, -9.8104e-01, -4.7014e-03, -1.2646e+00,
         -5.3252e-01],
        [-3.8966e+00, -1.7731e-01, -3.9479e+00, -5.2573e-02, -6.8880e-01,
         -3.8895e-01],
        [-4.6692e+00, -2.2224e-01, -2.6806e+00, -4.3183e-01, -2.0292e+00,
         -1.0788e+00],
        [-2.5890e+00, -6.4869e-01, -2.0910e+00, -5.3503e-02, -2.1552e+00,
         -1.5413e+00],
        [-4.2555e+00, -3.2387e-01, -5.6758e-01, -8.9911e-03, -5.7119e+00,
         -1.5372e+00],
        [-3.5769e+00, -3.0288e+00, -1.8436e+00, -1.5616e-03, -2.1830e+00,
         -5.2276e-01],
        [-4.2726e+00, -3.8808e+00, -1.7374e+00, -8.4051e-02, -3.7161e+00,
         -1.8561e+00],
        [-9.3962e-01, -1.1863e-01, -1.9372e-01, -3.3412e-01, -1.9596e+00,
         -6.1088e-02],
        [-1.6170e+00, -1.6196e-01, -9.3532e-01, -4.8224e-01, -1.0974e-01,
         -5.7834e-01]], device='cuda:0')
tensor([-11.6561, -10.1056, -11.1498,  -6.8313,  -6.7599, -12.7089,  -8.3214,
        -12.6884, -10.5787, -10.7689, -10.2468,  -3.6063,  -9.1521, -11.1120,
         -9.0787, -12.4050, -11.1566, -15.5471,  -3.6068,  -3.8846],
       device='cuda:0')
tensor([-1.4508, -2.9597, -1.4169, -2.3836, -3.5345, -4.8738, -1.6639, -0.2303,
        -0.1768, -4.2276, -2.3090, -4.0202, -2.8391, -3.6131, -2.0552, -2.1936,
        -2.7898, -0.1161, -4.5990, -4.6015], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.8719, -4.0307, -2.9452, -0.9556, -2.1182, -1.5034],
        [-1.2115, -1.1753, -1.9242, -1.0356, -1.1718, -1.1056],
        [-3.4622, -1.3971, -2.5812, -1.4964, -1.3451, -0.9323],
        [-3.6504, -1.3928, -2.2914, -1.1600, -1.4823, -1.5320],
        [-3.0939, -1.1440, -3.1796, -1.2613, -2.4775, -2.5060],
        [-3.5312, -1.1367, -1.6037, -1.2039, -2.7574, -1.7377],
        [-4.5633, -3.0609, -1.8805, -1.1863, -2.0304, -1.3967],
        [-3.5439, -2.7885, -2.3310, -1.4516, -3.2320, -1.7350],
        [-1.2208, -1.5758, -1.6635, -1.4959, -1.5215, -1.4530],
        [-3.0082, -1.4244, -1.8577, -1.3918, -1.6597, -1.0004],
        [-4.6977, -1.2119, -2.1216, -1.3161, -1.3782, -1.3238],
        [-3.1241, -1.2359, -1.1877, -1.8422, -2.2319, -3.1450],
        [-5.8449, -2.7495, -1.5551, -1.1339, -2.4157, -1.2685],
        [-3.7754, -2.5817, -2.2706, -1.2091, -1.8404, -1.4947],
        [-1.2809, -1.1953, -1.2548, -1.5527, -1.5239, -2.4519],
        [-3.5084, -1.4563, -3.1355, -1.4380, -2.5590, -1.2112],
        [-4.5553, -1.3302, -2.0605, -1.2687, -2.0954, -1.2044],
        [-5.7006, -1.4677, -1.5304, -2.5607, -1.5406, -0.9815],
        [-4.8786, -2.3993, -2.1514, -1.5420, -1.9201, -1.9728],
        [-4.0091, -3.1645, -2.1716, -1.0088, -2.9575, -2.0284]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.4250,  -7.6239, -11.2143, -11.5089, -13.6624, -11.9706, -14.1181,
        -15.0821,  -8.9305, -10.3423, -12.0493, -12.7667, -14.9676, -13.1720,
         -9.2595, -13.3085, -12.5145, -13.7817, -14.8643, -15.3399],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.0458e-01, -4.0932e+00, -2.6189e+00, -1.3978e-03, -2.4490e+00,
         -7.0984e-01],
        [-1.1185e+00, -4.0154e-01, -1.6965e+00, -1.5658e-01, -4.7957e-02,
         -4.4224e-02],
        [-3.6526e+00, -2.1662e-02, -3.4078e+00, -5.1356e-01, -3.0556e-01,
         -1.7900e-01],
        [-2.9095e+00, -3.4636e-02, -2.6253e+00, -1.2558e-02, -1.1574e+00,
         -1.4903e+00],
        [-3.2773e+00, -8.1347e-01, -3.2575e+00, -2.1901e-01, -3.0178e+00,
         -3.3012e+00],
        [-3.7029e+00, -1.6123e-01, -1.4424e+00, -2.3299e-02, -3.3861e+00,
         -1.3439e+00],
        [-3.1610e+00, -4.1191e+00, -1.4630e+00, -2.2502e-01, -2.3353e+00,
         -1.0123e-01],
        [-3.1135e+00, -2.6498e+00, -2.4395e+00, -5.2693e-01, -4.8576e+00,
         -1.5890e+00],
        [-9.1081e-01, -1.5501e+00, -1.4023e+00, -6.5131e-01, -5.7053e-01,
         -6.9642e-01],
        [-3.1813e+00, -1.2607e-02, -1.5738e+00, -2.1761e-01, -1.4603e+00,
         -2.7693e-01],
        [-3.8013e+00, -1.7054e-01, -2.3032e+00, -3.5393e-01, -3.6756e-01,
         -3.3458e-01],
        [-3.5255e+00, -4.7259e-02, -5.5342e-03, -1.3897e+00, -2.6429e+00,
         -3.2335e+00],
        [-4.4265e+00, -3.0154e+00, -1.1062e+00, -3.5117e-01, -3.0566e+00,
         -8.3602e-02],
        [-2.4720e+00, -2.6848e+00, -2.9261e+00, -1.4577e-01, -1.5761e+00,
         -5.3210e-01],
        [-9.8798e-01, -6.7371e-02, -2.5930e-01, -7.1976e-01, -3.8649e-01,
         -2.8352e+00],
        [-3.6502e+00, -1.9585e-01, -4.1622e+00, -4.8894e-01, -3.2556e+00,
         -4.4039e-01],
        [-3.3612e+00, -4.9976e-01, -1.8252e+00, -2.2847e-01, -3.1882e+00,
         -2.2142e-01],
        [-4.5001e+00, -1.7232e-02, -1.2961e+00, -2.6934e+00, -1.3112e+00,
         -2.2577e-02],
        [-3.4127e+00, -2.5449e+00, -3.1421e+00, -1.4093e+00, -1.9180e+00,
         -2.2495e+00],
        [-3.3338e+00, -3.4301e+00, -2.2678e+00, -1.0562e-04, -4.7459e+00,
         -2.4437e+00]], device='cuda:0')
tensor([-10.2769,  -3.4653,  -8.0802,  -8.2297, -13.8863, -10.0598, -11.4047,
        -15.1762,  -5.7815,  -6.7225,  -7.3311, -10.8444, -12.0396, -10.3370,
         -5.2561, -12.1932,  -9.3242,  -9.8405, -14.6764, -16.2213],
       device='cuda:0')
tensor([-3.1481, -4.1586, -3.1341, -3.2792,  0.2240, -1.9108, -2.7133,  0.0942,
        -3.1491, -3.6198, -4.7182, -1.9223, -2.9281, -2.8350, -4.0033, -1.1153,
        -3.1903, -3.9412, -0.1879,  0.8815], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.5002, -2.4057, -2.1359, -1.0777, -1.4459, -1.2842],
        [-2.7018, -1.9637, -1.2636, -1.0461, -1.1807, -1.8735],
        [-3.6349, -1.4779, -1.9169, -1.6754, -1.8041, -1.1465],
        [-4.2597, -1.2497, -1.4916, -1.1666, -1.3852, -1.4829],
        [-1.0781, -1.4701, -1.3811, -1.4489, -2.5679, -1.2663],
        [-5.3932, -3.6398, -1.6074, -1.5800, -2.1383, -3.0691],
        [-5.1620, -3.8852, -2.6735, -0.9389, -1.8651, -1.2930],
        [-1.2150, -1.3471, -2.2764, -0.8337, -1.5396, -1.1148],
        [-3.3588, -1.2762, -2.6421, -1.2951, -1.3764, -1.6474],
        [-7.3291, -1.5577, -2.0775, -1.1764, -1.4984, -2.1760],
        [-1.0927, -1.8825, -1.8603, -1.8184, -1.1661, -1.6817],
        [-4.4517, -1.5138, -3.3199, -2.7200, -2.0752, -2.7525],
        [-3.5660, -1.1497, -1.7326, -1.1235, -2.0019, -2.1459],
        [-7.0940, -2.6634, -1.7754, -1.7227, -1.9774, -3.7174],
        [-3.0618, -2.8357, -2.1034, -4.5212, -2.7488, -2.0124],
        [-1.2145, -1.2185, -1.5466, -1.0851, -1.3742, -1.2408],
        [-3.3121, -1.2659, -3.7724, -1.2929, -2.1249, -3.0023],
        [-4.5257, -1.4074, -1.6728, -1.1686, -1.4662, -1.0784],
        [-2.3296, -1.4861, -1.7905, -2.0462, -1.8415, -1.1719],
        [-7.0338, -1.9598, -2.0722, -1.4457, -2.4278, -1.4292]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.8496, -10.0294, -11.6557, -11.0356,  -9.2123, -17.4279, -15.8177,
         -8.3267, -11.5961, -15.8150,  -9.5017, -16.8331, -11.7195, -18.9503,
        -17.2833,  -7.6797, -14.7704, -11.3192, -10.6659, -16.3686],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.4466e+00, -2.2094e+00, -1.8926e+00, -1.0996e-01, -1.3103e-01,
         -7.0378e-02],
        [-3.1714e+00, -2.5914e+00, -1.6687e-01, -8.0633e-04, -2.0856e-01,
         -1.5715e+00],
        [-3.7097e+00, -2.6583e-02, -1.6297e+00, -1.0104e+00, -2.3989e+00,
         -3.8532e-01],
        [-3.7140e+00, -8.3861e-01, -3.0586e-01, -1.8179e-01, -1.2895e+00,
         -1.2566e+00],
        [-7.3189e-01, -2.6202e-04, -4.7054e-01, -8.2723e-01, -3.5442e+00,
         -4.8144e-01],
        [-4.9089e+00, -4.4056e+00, -1.1765e+00, -1.6123e+00, -2.3832e+00,
         -3.6330e+00],
        [-3.9287e+00, -3.7911e+00, -2.3880e+00, -9.2494e-04, -1.6151e+00,
         -2.3192e-02],
        [-8.5622e-01, -1.0501e-01, -2.4233e+00, -9.5367e-07, -1.3569e+00,
         -5.3189e-01],
        [-3.2647e+00, -1.3595e-01, -3.3931e+00, -1.3264e-01, -6.7325e-01,
         -1.3282e+00],
        [-4.3772e+00, -1.9372e+00, -2.2328e+00, -1.7303e-01, -1.7761e-01,
         -1.9762e+00],
        [-4.5307e-01, -2.6318e+00, -1.9182e+00, -1.2828e+00, -5.0068e-06,
         -5.3301e-01],
        [-5.6314e+00, -5.8974e-01, -3.8884e+00, -2.8038e+00, -1.7583e+00,
         -2.4977e+00],
        [-3.9584e+00, -8.8744e-01, -1.4063e+00, -1.1667e-01, -2.3823e+00,
         -3.2091e+00],
        [-4.8358e+00, -2.0050e+00, -1.6544e+00, -1.3308e+00, -2.1778e+00,
         -3.8753e+00],
        [-2.0488e+00, -3.0472e+00, -2.2886e+00, -2.9383e+00, -4.6979e+00,
         -2.3959e+00],
        [-4.8568e-01, -4.8580e-01, -7.6271e-01, -5.9814e-01, -4.9048e-01,
         -3.6180e-01],
        [-3.9197e+00, -2.0027e-05, -4.3802e+00, -1.3878e-02, -2.5398e+00,
         -2.0722e+00],
        [-3.6331e+00, -5.6231e-01, -8.0594e-01, -9.0580e-03, -1.3048e+00,
         -1.5213e-02],
        [-2.0203e+00, -5.5487e-02, -2.0438e+00, -1.7565e+00, -2.2795e+00,
         -1.1834e-01],
        [-4.9001e+00, -1.2780e+00, -2.7544e+00, -1.0536e+00, -3.4454e+00,
         -2.7490e-01]], device='cuda:0')
tensor([ -5.8601,  -7.7105,  -9.1605,  -7.5864,  -6.0556, -18.1194, -11.7471,
         -5.2733,  -8.9278, -10.8741,  -6.8189, -17.1693, -11.9602, -15.8792,
        -17.4168,  -3.1846, -12.9259,  -6.3304,  -8.2739, -13.7063],
       device='cuda:0')
tensor([-4.9895, -2.3189, -2.4951, -3.4492, -3.1567,  0.6916, -4.0706, -3.0534,
        -2.6683, -4.9409, -2.6828,  0.3361,  0.2407, -3.0712,  0.1335, -4.4951,
        -1.8445, -4.9888, -2.3919, -2.6623], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.6676, -1.6530, -3.4718, -1.3588, -1.8312, -1.3623],
        [-5.8816, -1.2391, -2.9484, -1.1385, -1.4148, -1.3617],
        [-3.3411, -1.3857, -1.2371, -1.9142, -1.9395, -1.5846],
        [-3.9206, -1.9910, -1.9458, -1.2169, -2.2410, -1.6386],
        [-2.9570, -3.3255, -2.4278, -2.0308, -2.1048, -1.5861],
        [-1.1841, -1.1918, -1.9584, -1.1427, -1.9351, -1.2843],
        [-2.5895, -1.4901, -2.4949, -1.3925, -1.3399, -1.2898],
        [-2.7979, -1.5034, -2.3703, -1.2492, -1.3842, -1.2483],
        [-2.3683, -1.2497, -4.1077, -1.1078, -1.6598, -2.2083],
        [-4.7411, -1.2219, -1.1391, -1.1336, -2.2571, -1.7936],
        [-4.4751, -2.1838, -2.7815, -1.0575, -2.4750, -1.2761],
        [-2.8188, -5.8629, -1.6267, -1.7738, -2.1025, -1.4400],
        [-1.2716, -1.3888, -1.4509, -1.1260, -1.7687, -1.2757],
        [-3.0181, -1.6495, -2.1752, -1.2799, -1.9672, -0.8849],
        [-3.3353, -1.2303, -3.0399, -1.2736, -1.4022, -1.5285],
        [-3.3000, -1.2512, -1.8123, -1.2130, -1.8274, -1.8876],
        [-3.6078, -1.1731, -1.5207, -1.2111, -2.6124, -1.8355],
        [-6.5244, -2.4984, -1.8250, -1.1581, -1.9987, -1.5666],
        [-2.3844, -5.0575, -2.7666, -1.5071, -2.4264, -1.3249],
        [-1.1672, -1.2460, -1.1121, -1.3576, -1.9364, -1.5022]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.3448, -13.9842, -11.4023, -12.9539, -14.4319,  -8.6965, -10.5967,
        -10.5534, -12.7015, -12.2863, -14.2490, -15.6248,  -8.2818, -10.9748,
        -11.8098, -11.2916, -11.9605, -15.5712, -15.4669,  -8.3215],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.9416e+00, -4.3423e-01, -5.1252e+00, -1.2434e-01, -1.9586e+00,
         -5.6000e-01],
        [-4.2325e+00, -1.3031e-02, -3.6504e+00, -3.9483e-02, -1.1672e+00,
         -7.6562e-01],
        [-3.1883e+00, -4.3408e-02, -4.7435e-02, -1.4892e+00, -2.3309e+00,
         -9.9499e-01],
        [-3.0380e+00, -1.4165e+00, -2.1420e+00, -4.2595e-01, -2.7147e+00,
         -1.2754e+00],
        [-2.0079e+00, -4.0966e+00, -3.5443e+00, -7.2758e-01, -2.4369e+00,
         -9.2813e-01],
        [-6.8511e-01, -8.6208e-02, -1.8304e+00, -1.2980e-01, -2.0282e+00,
         -1.2671e-01],
        [-3.2924e+00, -9.2085e-02, -3.0238e+00, -2.0257e-01, -1.2885e-01,
         -2.7348e-01],
        [-2.4396e+00, -1.5239e-01, -2.7455e+00, -5.7157e-02, -4.1866e-01,
         -2.5479e-01],
        [-1.9746e+00, -1.5421e-01, -3.8153e+00, -2.5008e-03, -1.7154e+00,
         -2.5896e+00],
        [-4.6418e+00, -1.8744e-02, -4.8072e-01, -8.3566e-04, -2.6606e+00,
         -1.1993e+00],
        [-3.0990e+00, -1.7603e+00, -4.8568e+00, -5.8616e-02, -3.5245e+00,
         -3.4034e-04],
        [-1.7178e+00, -5.7836e+00, -9.1950e-01, -9.8663e-01, -2.4745e+00,
         -7.2801e-01],
        [-7.1541e-01, -9.0940e-01, -8.6344e-01, -3.2935e-02, -1.3937e+00,
         -1.3203e-01],
        [-3.7760e+00, -4.5427e-01, -2.4963e+00, -7.9530e-03, -2.0152e+00,
         -2.0838e-01],
        [-2.9516e+00, -2.4596e-02, -3.7771e+00, -5.9177e-01, -9.6558e-01,
         -1.2671e+00],
        [-3.4276e+00, -3.0689e-01, -1.1168e+00, -1.1980e-01, -2.0463e+00,
         -1.5902e+00],
        [-3.6233e+00, -3.7055e-01, -1.3690e+00, -2.7455e-02, -3.4342e+00,
         -1.6023e+00],
        [-4.9765e+00, -2.2856e+00, -1.7392e+00, -1.0843e-02, -2.1381e+00,
         -1.3135e+00],
        [-1.3338e+00, -5.3394e+00, -3.3712e+00, -4.5289e-01, -3.5017e+00,
         -2.7752e-01],
        [-7.8126e-01, -2.6000e-02, -3.8286e-03, -9.2882e-01, -1.9709e+00,
         -9.1079e-01]], device='cuda:0')
tensor([-11.1439,  -9.8683,  -8.0942, -11.0126, -13.7414,  -4.8864,  -7.0132,
         -6.0681, -10.2516,  -9.0020, -13.2996, -12.6100,  -4.0469,  -8.9580,
         -9.5777,  -8.6076, -10.4268, -12.4639, -14.2765,  -4.6216],
       device='cuda:0')
tensor([-1.2009, -4.1159, -3.3081, -1.9413, -0.6906, -3.8100, -3.5836, -4.4852,
        -2.4500, -3.2843, -0.9495, -3.0148, -4.2348, -2.0168, -2.2321, -2.6839,
        -1.5338, -3.1073, -1.1904, -3.6998], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.4043, -1.8627, -2.3982, -1.3781, -1.8251, -3.2831],
        [-3.5007, -1.1770, -1.3628, -1.7822, -1.6821, -1.3013],
        [-2.3069, -1.8341, -2.3539, -3.3634, -2.8993, -3.3714],
        [-2.7047, -1.8796, -2.0463, -1.0846, -1.9798, -1.3236],
        [-2.1690, -1.5598, -1.5031, -0.8949, -1.3846, -2.2813],
        [-2.8810, -1.1580, -1.1018, -1.1780, -1.1257, -1.1118],
        [-1.1958, -1.9175, -2.9770, -1.3212, -1.3328, -1.1388],
        [-1.2860, -1.1423, -1.2281, -1.1406, -1.4173, -2.0495],
        [-1.5939, -1.6133, -1.5568, -1.4748, -1.1183, -1.4761],
        [-1.5023, -1.7544, -2.1531, -3.7913, -1.3359, -1.2889],
        [-2.2381, -1.6753, -3.1339, -5.0213, -1.8213, -1.6165],
        [-1.7983, -2.8734, -1.5926, -1.5669, -2.9941, -1.5680],
        [-4.9843, -4.4262, -2.2385, -1.3866, -2.5701, -2.1500],
        [-3.4583, -2.2152, -2.4303, -1.6068, -2.2948, -2.2268],
        [-1.6899, -1.2498, -1.2453, -0.9320, -1.9763, -1.9962],
        [-2.1228, -1.2466, -2.2862, -1.2047, -1.7855, -1.2721],
        [-3.9220, -1.5620, -2.0892, -1.2137, -2.2085, -1.6667],
        [-3.4027, -1.1869, -1.3133, -2.3350, -2.2106, -1.8350],
        [-0.9728, -1.3077, -1.6989, -1.3998, -1.4599, -1.0125],
        [-4.7447, -3.2177, -2.2878, -1.0147, -2.0715, -1.5115]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.1515, -10.8062, -16.1290, -11.0187,  -9.7927,  -8.5562,  -9.8831,
         -8.2637,  -8.8332, -11.8258, -15.5064, -12.3933, -17.7557, -14.2321,
         -9.0895,  -9.9178, -12.6621, -12.2834,  -7.8517, -14.8479],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.7597e+00, -6.8480e-01, -2.4333e+00, -1.9285e-01, -1.9242e+00,
         -2.0039e+00],
        [-2.8757e+00, -3.4763e-01, -3.3541e-02, -1.2742e+00, -1.9716e+00,
         -7.8143e-02],
        [-1.7137e+00, -1.0885e+00, -2.8351e+00, -4.1077e+00, -3.2806e+00,
         -3.8689e+00],
        [-1.2818e+00, -1.0749e+00, -1.6649e+00, -1.0490e-05, -1.7527e+00,
         -1.4459e-02],
        [-1.0375e+00, -5.0131e-01, -3.9737e-01, -1.9823e-01, -8.3573e-01,
         -2.8810e+00],
        [-3.2386e+00, -3.2568e-01, -4.6458e-03, -5.1210e-02, -5.4911e-02,
         -1.5843e-01],
        [-7.9635e-01, -1.1093e+00, -4.6920e+00, -1.3788e-01, -5.4196e-02,
         -5.3398e-01],
        [-6.4638e-01, -7.3743e-04, -9.7597e-02, -6.5348e-02, -4.2302e-01,
         -1.7780e+00],
        [-4.8733e-01, -1.2652e+00, -1.2298e+00, -3.3930e-01, -1.8303e-02,
         -1.2409e+00],
        [-1.7232e+00, -2.0702e+00, -2.5225e+00, -3.4513e+00, -1.3791e-01,
         -4.5658e-01],
        [-2.6074e+00, -9.8973e-01, -4.1620e+00, -5.3383e+00, -1.4578e+00,
         -1.0869e+00],
        [-1.8127e+00, -1.9969e+00, -1.0833e+00, -8.3723e-01, -3.4855e+00,
         -1.6154e+00],
        [-3.8765e+00, -3.5980e+00, -2.5843e+00, -4.1310e-01, -3.4206e+00,
         -1.9673e+00],
        [-3.1067e+00, -1.9571e+00, -3.7318e+00, -2.3365e-01, -2.6641e+00,
         -2.8784e+00],
        [-2.3010e+00, -7.7524e-01, -2.1159e-01, -1.3467e-01, -2.0850e+00,
         -1.9632e+00],
        [-2.4189e+00, -3.3606e-02, -2.5451e+00, -2.0991e-02, -1.5840e+00,
         -4.8855e-01],
        [-3.0771e+00, -2.3086e-01, -2.0440e+00, -1.5000e-02, -3.4106e+00,
         -1.6533e+00],
        [-3.1814e+00, -2.2115e-01, -8.3084e-02, -2.2372e+00, -2.8405e+00,
         -1.7823e+00],
        [-5.3592e-02, -6.7377e-02, -1.6822e+00, -5.7752e-01, -5.5689e-01,
         -2.9985e-02],
        [-3.7976e+00, -3.8411e+00, -2.8704e+00, -7.5298e-02, -2.3436e+00,
         -2.9934e-01]], device='cuda:0')
tensor([ -9.9988,  -6.5809, -16.8946,  -5.7888,  -5.8511,  -3.8335,  -7.3237,
         -3.0111,  -4.5808, -10.3617, -15.6421, -10.8311, -15.8597, -14.5717,
         -7.4707,  -7.0911, -10.4308, -10.3457,  -2.9676, -13.2273],
       device='cuda:0')
tensor([-3.1527, -4.2253,  0.7656, -5.2299, -3.9415, -4.7227, -2.5594, -5.2527,
        -4.2524, -1.4642,  0.1357, -1.5622, -1.8960,  0.3396, -1.6188, -2.8267,
        -2.2314, -1.9377, -4.8841, -1.6206], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.9108, -1.1299, -1.1685, -1.9554, -1.8065, -1.2903],
        [-3.3977, -2.0991, -2.2286, -1.3087, -3.1526, -1.2014],
        [-2.2761, -3.9526, -2.0696, -1.6295, -2.0229, -1.4383],
        [-2.0503, -1.1645, -1.1735, -1.0292, -1.7354, -1.2095],
        [-3.3527, -1.1760, -2.0847, -1.2307, -1.5420, -0.9290],
        [-5.3908, -1.5049, -2.0025, -1.5233, -1.6223, -2.5836],
        [-1.7599, -1.6999, -2.0007, -3.4035, -1.5468, -1.2616],
        [-4.6258, -2.2413, -2.0844, -3.9522, -2.6733, -2.5685],
        [-4.9389, -2.3892, -2.3606, -1.1548, -1.6888, -1.5169],
        [-1.2700, -2.0080, -2.5310, -3.5135, -1.3735, -1.2159],
        [-3.9214, -1.5719, -2.0258, -1.2649, -1.5468, -0.9459],
        [-2.1848, -1.5546, -1.8524, -1.4369, -1.4059, -1.2877],
        [-3.3371, -1.0867, -1.2711, -1.1803, -1.4068, -1.2129],
        [-1.2041, -1.3179, -1.4907, -1.0281, -1.7916, -2.2947],
        [-5.6290, -2.7183, -1.9739, -1.1311, -1.5564, -1.9467],
        [-3.3201, -4.0783, -2.4003, -1.2591, -2.0835, -1.4532],
        [-1.2709, -2.1808, -2.4175, -2.7234, -1.5616, -1.2777],
        [-2.3773, -1.3289, -2.9050, -1.3184, -1.6138, -1.3564],
        [-4.4089, -1.2447, -2.3653, -1.1008, -1.5614, -1.1946],
        [-2.6067, -1.1232, -1.1979, -1.2261, -1.7031, -2.3261]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.2613, -13.3881, -13.3891,  -8.3623, -10.3150, -14.6274, -11.6726,
        -18.1454, -14.0491, -11.9119, -11.2766,  -9.7223,  -9.4948,  -9.1271,
        -14.9555, -14.5945, -11.4319, -10.8999, -11.8756, -10.1831],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.0192e+00, -6.0669e-03, -7.4461e-03, -1.5541e+00, -1.8578e+00,
         -1.7650e-01],
        [-2.3327e+00, -1.6418e+00, -3.1212e+00, -6.6708e-01, -4.5886e+00,
         -9.0549e-02],
        [-8.2446e-01, -5.0408e+00, -2.3634e+00, -4.0590e-01, -2.1658e+00,
         -5.4927e-02],
        [-3.0639e+00, -1.2600e-04, -8.7746e-02, -2.5113e-01, -1.5101e+00,
         -4.9975e-03],
        [-3.9347e+00, -2.6314e-01, -2.1042e+00, -1.2545e-01, -1.0944e+00,
         -1.0848e-01],
        [-4.3058e+00, -8.4316e-02, -1.9176e+00, -7.6858e-01, -1.6965e+00,
         -2.5134e+00],
        [-1.5077e+00, -7.0758e-01, -2.5163e+00, -3.5316e+00, -1.3695e+00,
         -1.6051e-01],
        [-3.0934e+00, -2.1135e+00, -2.8995e+00, -4.3645e+00, -3.6733e+00,
         -3.3087e+00],
        [-4.2172e+00, -2.1631e+00, -2.7205e+00, -5.4646e-04, -1.0446e+00,
         -1.0094e+00],
        [-1.5652e-04, -2.5119e+00, -2.7345e+00, -2.9760e+00, -2.7140e-01,
         -3.4337e-01],
        [-3.8923e+00, -1.4857e+00, -1.8439e+00, -2.3924e-03, -1.4549e+00,
         -3.8042e-02],
        [-1.8445e+00, -3.6418e-02, -1.3752e+00, -6.9510e-01, -4.3195e-01,
         -6.7650e-01],
        [-3.7697e+00, -5.5581e-02, -1.7546e-01, -1.9014e-01, -7.9130e-01,
         -5.0080e-01],
        [-9.7639e-01, -1.6679e-02, -1.0236e+00, -2.7523e-02, -1.7483e+00,
         -2.4034e+00],
        [-3.9147e+00, -3.5144e+00, -2.5267e+00, -2.6387e-01, -2.2345e-01,
         -1.6988e+00],
        [-2.2342e+00, -3.7395e+00, -2.1577e+00, -5.7623e-01, -2.2243e+00,
         -7.0071e-01],
        [-8.6368e-02, -2.9479e+00, -2.7057e+00, -2.3940e+00, -8.8866e-01,
         -2.8414e-01],
        [-2.4814e+00, -2.5227e-02, -3.4634e+00, -8.2781e-02, -1.4770e+00,
         -6.4576e-01],
        [-3.6193e+00, -3.9340e-01, -2.9595e+00, -1.3164e-02, -1.7677e+00,
         -5.0613e-03],
        [-2.4892e+00, -7.4118e-02, -6.0548e-02, -2.1442e-01, -1.7649e+00,
         -2.7284e+00]], device='cuda:0')
tensor([ -6.6211, -12.4420, -10.8554,  -4.9180,  -7.6303, -11.2862,  -9.7932,
        -19.4529, -11.1554,  -8.8373,  -8.7172,  -5.0597,  -5.4830,  -6.1960,
        -12.1418, -11.6326,  -9.3067,  -8.1756,  -8.7582,  -7.3315],
       device='cuda:0')
tensor([-3.6403, -0.9461, -2.5337, -3.4443, -2.6847, -3.3412, -1.8794,  1.3075,
        -2.8938, -3.0746, -2.5594, -4.6626, -4.0119, -2.9311, -2.8136, -2.9619,
        -2.1252, -2.7243, -3.1174, -2.8515], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.1311, -1.4108, -3.1239, -1.4291, -1.5047, -1.2672],
        [-4.8063, -1.5572, -1.4544, -1.2551, -1.4640, -1.2667],
        [-1.1243, -1.5989, -1.5266, -2.2136, -1.1666, -1.3348],
        [-8.1190, -1.8197, -2.0462, -3.2489, -1.3373, -2.0313],
        [-2.8361, -1.0772, -1.5505, -0.9353, -2.5867, -1.9490],
        [-6.5415, -1.8382, -2.3053, -1.9650, -2.3048, -5.1970],
        [-3.7432, -2.5038, -1.8300, -1.8857, -2.4332, -2.5289],
        [-1.2459, -1.2631, -1.2883, -2.6777, -2.1036, -1.1957],
        [-2.0911, -2.2121, -2.4761, -1.5055, -1.2258, -0.9190],
        [-1.4399, -2.0484, -2.6748, -1.8735, -1.0814, -1.8016],
        [-6.1885, -1.8777, -2.6066, -1.2172, -2.1611, -1.2445],
        [-1.2393, -1.6833, -4.0037, -2.7410, -1.3944, -4.1730],
        [-2.4097, -1.9053, -4.2089, -5.9868, -1.2267, -1.9437],
        [-2.6857, -1.0376, -3.7054, -2.4929, -2.7693, -2.3418],
        [-4.5406, -1.6023, -2.0976, -1.6945, -2.8434, -4.6682],
        [-7.1029, -3.7762, -3.0848, -3.9556, -1.7558, -1.4474],
        [-1.2970, -1.3739, -1.5579, -1.1245, -1.4625, -1.4992],
        [-2.6126, -1.3690, -2.9111, -1.9371, -2.7766, -1.3505],
        [-4.2502, -1.3305, -2.0569, -1.4939, -2.1385, -1.1182],
        [-2.3723, -1.3064, -1.8870, -4.5152, -2.2804, -1.1305]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.8668, -11.8038,  -8.9648, -18.6024, -10.9347, -20.1517, -14.9248,
         -9.7743, -10.4295, -10.9195, -15.2956, -15.2346, -17.6811, -15.0328,
        -17.4467, -21.1227,  -8.3151, -12.9568, -12.3883, -13.4917],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.2924e+00, -5.9277e-01, -3.8809e+00, -4.0817e-01, -5.9206e-01,
         -1.9880e-01],
        [-2.7454e+00, -1.6585e+00, -7.5696e-01, -2.8005e-01, -3.1059e-02,
         -8.3554e-02],
        [-6.0293e-01, -1.1720e+00, -1.2980e+00, -1.9014e+00, -5.8456e-01,
         -3.8113e-01],
        [-6.7056e+00, -1.2753e+00, -2.0272e+00, -3.1427e+00, -5.2045e-01,
         -2.1241e+00],
        [-3.5250e+00, -7.4857e-01, -9.9241e-01, -8.0575e-02, -2.9645e+00,
         -2.3627e+00],
        [-4.5254e+00, -8.2568e-01, -3.0891e+00, -1.9014e+00, -3.0791e+00,
         -5.1011e+00],
        [-3.1312e+00, -2.4465e+00, -1.5278e+00, -6.8397e-01, -3.6551e+00,
         -3.5348e+00],
        [-7.0261e-01, -1.5845e-01, -8.9299e-02, -2.9556e+00, -2.5196e+00,
         -2.1661e-02],
        [-2.5241e+00, -1.8951e+00, -2.7034e+00, -7.6161e-01, -4.8759e-02,
         -1.1605e-01],
        [-9.7579e-01, -1.3234e+00, -3.6974e+00, -1.2062e+00, -1.2143e-01,
         -1.2552e+00],
        [-4.3459e+00, -2.4470e+00, -3.0937e+00, -4.1767e-02, -2.9819e+00,
         -4.3671e-01],
        [-1.9742e-01, -1.3741e+00, -4.6830e+00, -3.5118e+00, -9.4752e-01,
         -1.3921e+00],
        [-3.1795e+00, -1.1412e+00, -4.4039e+00, -5.1298e+00, -3.0965e-01,
         -1.5786e+00],
        [-3.2648e+00, -1.7888e-01, -4.0017e+00, -1.6769e+00, -3.0625e+00,
         -2.5434e+00],
        [-3.0529e+00, -1.5101e-01, -2.4234e+00, -1.2112e+00, -4.6528e+00,
         -4.1676e+00],
        [-4.8785e+00, -3.5403e+00, -2.7792e+00, -3.1149e+00, -6.3181e-01,
         -4.2243e-01],
        [-1.6739e-01, -8.8117e-01, -5.1808e-01, -1.1499e+00, -5.2570e-01,
         -1.3258e+00],
        [-2.9478e+00, -4.5714e-02, -2.9550e+00, -1.8525e+00, -4.3071e+00,
         -6.7482e-01],
        [-3.5123e+00, -2.6529e-03, -2.0349e+00, -8.9244e-01, -3.0311e+00,
         -1.1646e-01],
        [-2.1566e+00, -6.4429e-03, -2.1718e+00, -4.5002e+00, -3.1754e+00,
         -1.8690e-03]], device='cuda:0')
tensor([ -9.9651,  -5.5555,  -5.9400, -15.7954, -10.6738, -18.5219, -14.9793,
         -6.4473,  -8.0491,  -8.5793, -13.3470, -12.1059, -15.7426, -14.7282,
        -15.6590, -15.3671,  -4.5681, -12.7830,  -9.5899, -12.0123],
       device='cuda:0')
tensor([-2.9017, -6.2482, -3.0248, -2.8070, -0.2609, -1.6298,  0.0545, -3.3271,
        -2.3804, -2.3402, -1.9486, -3.1287, -1.9385, -0.3046, -1.7878, -5.7555,
        -3.7470, -0.1739, -2.7983, -1.4794], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.0817, -2.7798, -2.0159, -3.2262, -2.4160, -2.1484],
        [-4.5878, -3.3897, -2.5325, -0.9788, -1.8645, -1.4867],
        [-1.2269, -1.3923, -1.8774, -3.6725, -1.1112, -2.2255],
        [-3.6789, -1.7095, -3.3798, -1.8810, -1.4003, -1.6116],
        [-1.5786, -1.2581, -1.3021, -1.1150, -1.3740, -2.1240],
        [-2.0849, -1.4177, -1.0331, -1.1409, -1.1254, -1.0676],
        [-3.3540, -1.7272, -2.3496, -1.5322, -1.6982, -1.1084],
        [-0.9520, -1.0656, -2.6370, -2.5864, -1.2223, -1.5969],
        [-3.6789, -1.3601, -3.3654, -2.7255, -2.3505, -4.0650],
        [-3.0630, -1.0254, -1.4332, -1.2066, -1.9551, -2.0109],
        [-6.2503, -2.4603, -2.1127, -2.0920, -2.5426, -1.5084],
        [-4.8553, -3.1727, -2.6367, -1.3090, -2.2177, -2.1555],
        [-1.5148, -1.2294, -1.9283, -0.9592, -1.3583, -1.7149],
        [-2.6545, -1.2875, -2.4679, -1.3036, -1.8534, -2.2469],
        [-4.4545, -1.6553, -2.4684, -1.3818, -1.4120, -0.9750],
        [-2.7047, -1.1566, -1.1667, -1.3333, -2.0015, -2.3400],
        [-1.2528, -1.5772, -1.5725, -1.5194, -2.0887, -3.3942],
        [-5.8653, -4.2378, -3.0723, -1.0548, -1.9706, -1.5845],
        [-3.6046, -3.9875, -2.5357, -1.2894, -1.6943, -1.9607],
        [-1.2162, -1.3708, -1.6630, -1.4618, -1.4499, -1.2070]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-16.6680, -14.8400, -11.5059, -13.6609,  -8.7518,  -7.8696, -11.7696,
        -10.0603, -17.5454, -10.6942, -16.9664, -16.3470,  -8.7049, -11.8137,
        -12.3470, -10.7027, -11.4048, -17.7853, -15.0722,  -8.3688],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.8412e+00, -3.4492e+00, -2.5081e+00, -4.3666e+00, -3.3270e+00,
         -2.5820e+00],
        [-2.7686e+00, -2.9920e+00, -1.8985e+00, -4.5403e-02, -1.5529e+00,
         -4.7046e-01],
        [-2.6524e-01, -1.3503e+00, -1.5893e+00, -2.9410e+00, -5.7070e-03,
         -2.5295e+00],
        [-3.4828e+00, -2.3965e-01, -5.2641e+00, -1.6777e+00, -2.0341e-02,
         -1.5314e+00],
        [-9.5935e-01, -1.4069e-02, -3.4225e-04, -1.1921e-02, -1.8479e-01,
         -1.8312e+00],
        [-1.3070e+00, -1.0751e+00, -2.9033e-01, -1.1404e-02, -1.5415e-01,
         -2.9376e-01],
        [-2.2516e+00, -2.0671e+00, -3.1300e+00, -6.7928e-01, -1.6930e+00,
         -2.3925e-04],
        [-5.8470e-03, -3.5096e-01, -2.8641e+00, -2.7671e+00, -2.2167e-01,
         -9.9798e-01],
        [-3.5648e+00, -3.5052e-01, -3.7303e+00, -2.6783e+00, -2.6036e+00,
         -4.1079e+00],
        [-3.7118e+00, -2.8906e-01, -5.1506e-01, -1.8162e-01, -1.9938e+00,
         -2.1373e+00],
        [-3.7997e+00, -2.1505e+00, -2.8366e+00, -2.1601e+00, -3.6491e+00,
         -1.4854e+00],
        [-4.1972e+00, -3.6065e+00, -3.4131e+00, -1.2824e-01, -2.9063e+00,
         -2.6501e+00],
        [-2.1763e+00, -4.0254e-01, -1.7029e+00, -1.3646e-01, -9.2132e-03,
         -1.2448e+00],
        [-3.1551e+00, -3.2305e-01, -2.8239e+00, -6.5635e-02, -1.7858e+00,
         -1.4070e+00],
        [-3.5745e+00, -3.0497e-01, -2.8008e+00, -4.2767e-01, -8.0163e-02,
         -3.0718e-02],
        [-2.5782e+00, -9.2219e-03, -1.2716e-01, -5.4242e-01, -2.2711e+00,
         -2.8969e+00],
        [-1.3781e+00, -3.9653e-01, -1.3342e+00, -1.0179e+00, -2.6259e+00,
         -3.6449e+00],
        [-4.4607e+00, -4.6863e+00, -3.7445e+00, -1.4297e-01, -1.9298e+00,
         -1.2354e+00],
        [-2.4331e+00, -3.4855e+00, -2.8581e+00, -5.3889e-01, -1.3385e+00,
         -1.9904e+00],
        [-4.8711e-01, -7.5561e-01, -1.3379e+00, -4.5438e-01, -4.7317e-02,
         -8.5170e-02]], device='cuda:0')
tensor([-20.0740,  -9.7279,  -8.6811, -12.2159,  -3.0017,  -3.1318,  -9.8212,
         -7.2076, -17.0355,  -8.8287, -16.0815, -16.9015,  -5.6722,  -9.5605,
         -7.2189,  -8.4250, -10.3976, -16.1997, -12.6446,  -3.1675],
       device='cuda:0')
tensor([ 3.4060, -5.1121, -2.8249, -1.4450, -5.7501, -4.7378, -1.9485, -2.8527,
        -0.5099, -1.8655, -0.8849,  0.5545, -3.0327, -2.2531, -5.1281, -2.2776,
        -1.0072, -1.5856, -2.4276, -5.2013], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-5.8868, -1.5331, -2.3565, -1.2062, -1.5827, -1.8668],
        [-2.1434, -1.3499, -3.3119, -1.2175, -2.2969, -3.0930],
        [-4.5571, -1.2366, -1.6274, -1.2096, -1.9156, -1.2132],
        [-3.4323, -2.2206, -1.9007, -1.2709, -2.2340, -1.6496],
        [-2.6446, -2.0108, -2.6891, -2.1628, -2.1875, -2.0369],
        [-1.1943, -1.2279, -1.7013, -1.0566, -1.6282, -1.3230],
        [-4.7091, -1.2496, -2.7591, -1.2652, -1.2708, -1.0029],
        [-3.8236, -1.3737, -3.9839, -1.2697, -1.4529, -1.0909],
        [-1.7171, -1.1627, -2.8886, -1.1624, -2.0678, -1.5516],
        [-2.2061, -1.1767, -1.0470, -1.1841, -1.7195, -1.9124],
        [-3.9819, -2.2069, -1.7024, -1.1224, -2.1706, -1.2949],
        [-2.5406, -4.9267, -2.1242, -1.4319, -2.4548, -1.6363],
        [-1.2173, -1.2865, -1.1651, -1.2044, -1.6605, -1.2702],
        [-2.0729, -1.3291, -2.9633, -1.6563, -1.9302, -1.2164],
        [-3.5921, -1.3211, -2.0216, -1.3205, -1.9851, -1.9115],
        [-5.6755, -1.3353, -1.6095, -2.4529, -1.6222, -1.0410],
        [-4.5077, -2.1811, -2.4761, -2.7680, -2.7788, -2.1805],
        [-3.0601, -3.4601, -4.5532, -0.9647, -2.1512, -1.3598],
        [-1.1787, -1.3044, -2.6858, -1.1227, -1.2793, -1.1789],
        [-2.9701, -1.4678, -2.1048, -1.6238, -1.4538, -0.9131]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-14.4321, -13.4125, -11.7595, -12.7081, -13.7318,  -8.1313, -12.2568,
        -12.9947, -10.5503,  -9.2458, -12.4792, -15.1145,  -7.8041, -11.1683,
        -12.1519, -13.7363, -16.8922, -15.5491,  -8.7499, -10.5332],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.4898e+00, -1.9586e-01, -2.8140e+00, -3.6063e-02, -1.5683e+00,
         -1.8010e+00],
        [-2.1913e+00, -4.4762e-01, -3.6602e+00, -1.4904e-01, -3.1066e+00,
         -3.3149e+00],
        [-3.9908e+00, -1.9266e-01, -1.6331e+00, -1.2187e-02, -2.1587e+00,
         -5.0832e-01],
        [-2.3452e+00, -1.7697e+00, -1.9862e+00, -5.2033e-01, -3.0235e+00,
         -1.4027e+00],
        [-1.8563e+00, -1.5402e+00, -3.1656e+00, -1.2463e+00, -2.9498e+00,
         -2.3845e+00],
        [-9.0692e-01, -4.7439e-02, -1.3726e+00, -1.3758e-03, -1.1088e+00,
         -4.1911e-01],
        [-4.7554e+00, -3.6867e-02, -3.5293e+00, -1.6797e-02, -6.5911e-02,
         -1.6448e-01],
        [-3.1432e+00, -2.1186e-02, -5.5857e+00, -3.8585e-01, -1.2705e+00,
         -2.2181e-02],
        [-1.4929e+00, -6.4486e-01, -2.7612e+00, -8.2016e-03, -2.6215e+00,
         -1.1029e+00],
        [-2.5649e+00, -4.4071e-01, -7.0769e-02, -5.4872e-04, -1.5401e+00,
         -1.4412e+00],
        [-2.9617e+00, -1.7197e+00, -1.3230e+00, -1.1650e-01, -2.7790e+00,
         -5.6345e-02],
        [-1.3585e+00, -5.4484e+00, -2.1341e+00, -4.4315e-01, -3.5411e+00,
         -1.2290e+00],
        [-8.3922e-01, -4.6654e-01, -6.5017e-04, -1.3683e-01, -9.1965e-01,
         -8.5862e-02],
        [-2.3918e+00, -2.4974e-04, -3.7034e+00, -1.0012e+00, -2.1929e+00,
         -3.9425e-01],
        [-2.9361e+00, -2.9488e-01, -2.0269e+00, -4.1683e-01, -2.8639e+00,
         -2.2505e+00],
        [-4.5164e+00, -8.0117e-03, -1.6339e+00, -2.4685e+00, -1.6048e+00,
         -8.0515e-03],
        [-3.4869e+00, -1.8407e+00, -4.3056e+00, -3.5336e+00, -4.0568e+00,
         -2.7880e+00],
        [-1.6564e+00, -3.2310e+00, -3.9075e+00, -1.0375e-02, -2.5505e+00,
         -2.9876e-01],
        [-7.7645e-02, -9.5730e-01, -3.0992e+00, -2.4981e-01, -3.7533e-01,
         -2.1972e-01],
        [-3.1401e+00, -1.6759e-02, -2.2516e+00, -9.2612e-01, -1.0202e+00,
         -1.3484e-01]], device='cuda:0')
tensor([-10.9051, -12.8696,  -8.4957, -11.0477, -13.1427,  -3.8562,  -8.5688,
        -10.4286,  -8.6315,  -6.0582,  -8.9561, -14.1544,  -2.4488,  -9.6839,
        -10.7892, -10.2397, -20.0117, -11.6545,  -4.9790,  -7.4896],
       device='cuda:0')
tensor([-3.5271, -0.5429, -3.2638, -1.6604, -0.5891, -4.2751, -3.6880, -2.5661,
        -1.9188, -3.1876, -3.5230, -0.9601, -5.3553, -1.4844, -1.3627, -3.4967,
         3.1195, -3.8946, -3.7709, -3.0436], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.4534, -1.4851, -2.8692, -1.7141, -1.2823, -1.6234],
        [-2.0948, -3.5654, -3.4604, -3.7349, -1.5905, -3.2537],
        [-4.7921, -2.3831, -2.2473, -1.9847, -1.9754, -1.2407],
        [-4.5178, -2.9051, -1.7986, -4.1588, -2.1210, -1.2680],
        [-1.2619, -3.6670, -2.9091, -1.8270, -1.6478, -1.0936],
        [-2.5914, -1.4961, -1.6768, -1.8144, -1.7599, -1.3375],
        [-3.9188, -1.5828, -2.1417, -1.1694, -1.6167, -1.3561],
        [-3.3860, -1.6173, -1.1417, -1.2799, -1.6249, -1.8256],
        [-2.5080, -1.5060, -1.3369, -1.0589, -2.1447, -2.3229],
        [-3.6241, -1.7471, -1.4469, -1.0808, -1.7707, -1.3309],
        [-5.5942, -3.0855, -2.3513, -1.1164, -2.6438, -1.9224],
        [-1.4594, -1.2719, -1.1193, -1.0389, -1.8649, -1.6766],
        [-2.5523, -1.3532, -2.9944, -1.1208, -1.4738, -2.9081],
        [-2.8249, -1.6466, -2.4361, -1.3321, -1.7560, -0.9674],
        [-6.4318, -1.3252, -1.5176, -1.6698, -2.1279, -1.7727],
        [-2.3668, -1.4048, -1.7775, -1.2932, -2.5553, -1.2098],
        [-4.2024, -3.2339, -2.1391, -1.1120, -3.9598, -1.4157],
        [-4.1470, -3.4668, -1.7345, -1.2288, -1.9883, -1.3657],
        [-1.2018, -2.0100, -1.7386, -1.1901, -1.3203, -1.1900],
        [-2.4783, -1.4954, -2.8873, -2.2137, -1.8726, -0.9347]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.4275, -17.6997, -14.6232, -16.7693, -12.4063, -10.6761, -11.7854,
        -10.8754, -10.8775, -11.0005, -16.7135,  -8.4309, -12.4028, -10.9632,
        -14.8451, -10.6074, -16.0630, -13.9311,  -8.6509, -11.8821],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.4021e+00, -3.0376e-01, -3.6068e+00, -1.1583e+00, -6.1359e-01,
         -6.5169e-01],
        [-1.8524e+00, -3.3040e+00, -4.1790e+00, -4.1349e+00, -1.5574e+00,
         -3.9758e+00],
        [-2.5282e+00, -2.3012e+00, -3.8173e+00, -1.9451e+00, -2.0600e+00,
         -5.4795e-01],
        [-3.1499e+00, -2.5808e+00, -1.3179e+00, -3.5100e+00, -2.2206e+00,
         -1.4401e-01],
        [-5.3421e-03, -5.1226e+00, -3.5509e+00, -1.0562e+00, -1.1472e+00,
         -2.1836e-03],
        [-2.5136e+00, -1.3315e+00, -8.8346e-01, -1.6166e+00, -1.7831e+00,
         -7.2967e-01],
        [-3.5148e+00, -1.2160e+00, -2.4706e+00, -3.0059e-02, -1.9633e+00,
         -8.3288e-01],
        [-3.4803e+00, -1.7052e+00, -1.3585e-01, -6.7425e-01, -1.4891e+00,
         -1.9772e+00],
        [-3.2860e+00, -5.0243e-01, -1.9756e-01, -1.5785e-02, -2.6810e+00,
         -1.9724e+00],
        [-2.8279e+00, -7.6436e-01, -5.2230e-01, -9.2483e-04, -1.4461e+00,
         -2.4554e-01],
        [-4.4348e+00, -3.0164e+00, -2.3077e+00, -2.3339e-01, -3.6199e+00,
         -1.9222e+00],
        [-1.7386e+00, -4.2641e-02, -4.0436e-04, -3.3885e-02, -1.6703e+00,
         -1.1852e+00],
        [-3.1333e+00, -1.6975e-01, -3.6554e+00, -6.3112e-03, -8.7464e-01,
         -2.3199e+00],
        [-2.6341e+00, -1.2511e-01, -2.8163e+00, -2.7438e-01, -2.0075e+00,
         -1.0715e-01],
        [-4.3848e+00, -1.1362e-01, -5.9962e-01, -1.2112e+00, -3.0297e+00,
         -1.6812e+00],
        [-2.5299e+00, -3.3159e-01, -1.8913e+00, -4.7883e-01, -3.2547e+00,
         -3.9272e-01],
        [-2.9473e+00, -4.0761e+00, -2.7955e+00, -3.4189e-01, -6.1974e+00,
         -7.4884e-03],
        [-3.3595e+00, -3.2984e+00, -1.1719e+00, -2.7436e-01, -2.1436e+00,
         -6.5558e-01],
        [-6.6071e-01, -2.4379e+00, -1.4958e+00, -2.9227e-01, -1.4333e-02,
         -1.9346e-01],
        [-2.6799e+00, -6.3334e-02, -4.0681e+00, -2.1735e+00, -2.1538e+00,
         -1.1117e-01]], device='cuda:0')
tensor([ -8.7363, -19.0036, -13.1998, -12.9233, -10.8843,  -8.8580, -10.0276,
         -9.4621,  -8.6551,  -5.8071, -15.5345,  -4.6710, -10.1592,  -7.9645,
        -11.0201,  -8.8790, -16.3658, -10.9033,  -5.0945, -11.2498],
       device='cuda:0')
tensor([-2.6912,  1.3039, -1.4235, -3.8460, -1.5220, -1.8181, -1.7578, -1.4133,
        -2.2223, -5.1934, -1.1791, -3.7599, -2.2435, -2.9986, -3.8250, -1.7284,
         0.3028, -3.0278, -3.5564, -0.6323], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.7787, -1.0616, -2.1335, -1.2630, -1.4364, -1.3722],
        [-1.8536, -1.3394, -1.2894, -1.3562, -2.0132, -2.0340],
        [-4.6449, -3.0885, -1.7705, -1.0890, -1.9967, -2.4041],
        [-2.6806, -2.3303, -2.9823, -1.2665, -1.3747, -1.5793],
        [-1.3251, -2.0639, -1.8033, -2.2448, -1.4137, -1.2105],
        [-1.9575, -1.3362, -2.5779, -1.2499, -1.3452, -0.9764],
        [-0.8831, -1.2320, -2.6475, -1.0495, -1.2205, -1.2425],
        [-3.6281, -1.0626, -1.1665, -1.5024, -1.6474, -1.2375],
        [-2.6682, -1.5779, -1.5786, -1.1767, -1.3721, -1.1392],
        [-4.9440, -2.3842, -2.1681, -1.0078, -2.5594, -1.5594],
        [-1.7588, -1.2164, -1.2530, -1.0548, -1.6397, -1.4349],
        [-4.3336, -1.2515, -4.9048, -1.2201, -3.6162, -1.2651],
        [-2.8730, -1.5155, -1.5418, -1.2187, -1.7896, -1.3575],
        [-9.8775, -1.4753, -1.6575, -1.4512, -1.6910, -2.3369],
        [-3.9031, -1.1303, -1.5527, -1.1368, -2.3382, -1.2354],
        [-4.4019, -3.6315, -2.0528, -1.8702, -1.9943, -2.2068],
        [-3.6302, -2.8669, -4.0340, -1.0680, -1.8948, -1.2878],
        [-1.9536, -1.2194, -1.3491, -0.8656, -1.3204, -1.0718],
        [-3.9372, -1.3405, -2.2149, -1.4428, -1.8955, -1.0916],
        [-3.5151, -1.6729, -3.2475, -1.2604, -1.8130, -1.2426]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.0454,  -9.8859, -14.9937, -12.2137, -10.0614,  -9.4431,  -8.2751,
        -10.2444,  -9.5126, -14.6229,  -8.3576, -16.5914, -10.2960, -18.4893,
        -11.2965, -16.1574, -14.7818,  -7.7799, -11.9226, -12.7514],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.5598e+00, -5.4645e-01, -1.9654e+00, -2.6841e-01, -9.1882e-01,
         -8.0965e-01],
        [-2.4547e+00, -2.2746e-01, -9.6792e-01, -6.0380e-01, -2.0049e+00,
         -1.0931e+00],
        [-3.1138e+00, -3.8366e+00, -1.3073e+00, -1.5777e-02, -2.1939e+00,
         -3.2551e+00],
        [-1.6614e+00, -1.6998e+00, -3.4436e+00, -5.5132e-01, -7.9240e-01,
         -1.1785e+00],
        [-1.0476e+00, -2.8145e+00, -1.5855e+00, -1.7566e+00, -1.1539e-02,
         -9.1395e-02],
        [-2.2330e+00, -5.3406e-02, -3.2987e+00, -1.9515e-04, -2.2500e-01,
         -8.3136e-02],
        [-1.9988e-01, -2.7874e-01, -3.4990e+00, -2.3869e-02, -6.9787e-01,
         -5.1310e-01],
        [-3.5642e+00, -6.6517e-01, -6.2311e-02, -7.8829e-01, -1.5230e+00,
         -1.8787e-01],
        [-1.8983e+00, -6.2167e-01, -1.1027e+00, -4.5863e-02, -1.1483e-02,
         -2.4977e-02],
        [-3.4357e+00, -2.3739e+00, -2.2795e+00, -2.1129e-02, -3.9999e+00,
         -4.7347e-01],
        [-2.6398e+00, -6.2239e-04, -2.5931e-01, -1.6204e-01, -9.2512e-01,
         -5.5283e-01],
        [-5.0396e+00, -4.6565e-02, -5.6119e+00, -1.2813e-01, -4.3754e+00,
         -4.5405e-01],
        [-2.6668e+00, -4.8908e-02, -3.9331e-01, -1.1470e-02, -2.1968e+00,
         -7.6275e-01],
        [-6.2178e+00, -1.6802e-01, -7.9031e-01, -6.5716e-01, -1.7496e+00,
         -2.6939e+00],
        [-3.8209e+00, -4.4472e-02, -1.3543e+00, -1.8052e-01, -2.8770e+00,
         -5.7713e-01],
        [-3.2418e+00, -5.0036e+00, -2.7133e+00, -1.9934e+00, -2.1576e+00,
         -2.8979e+00],
        [-2.2143e+00, -2.4231e+00, -3.3547e+00, -2.4906e-02, -1.6802e+00,
         -1.8957e-02],
        [-2.9269e+00, -6.2080e-01, -3.5031e-01, -4.6912e-03, -7.3381e-01,
         -1.4007e-01],
        [-3.9550e+00, -4.6590e-01, -2.6587e+00, -4.6766e-01, -2.0856e+00,
         -1.1898e-01],
        [-2.6925e+00, -4.6126e-01, -4.2603e+00, -2.3892e-01, -2.5111e+00,
         -2.4928e-01]], device='cuda:0')
tensor([ -6.0685,  -7.3519, -13.7224,  -9.3270,  -7.3071,  -5.8934,  -5.2125,
         -6.7908,  -3.7051, -12.5837,  -4.5397, -15.6557,  -6.0801, -12.2767,
         -8.8543, -18.0076,  -9.7163,  -4.7766,  -9.7518, -10.4133],
       device='cuda:0')
tensor([-2.9769, -2.5340, -1.2713, -2.8866, -2.7543, -3.5497, -3.0627, -3.4536,
        -5.8075, -2.0392, -3.8178, -0.9357, -4.2160, -6.2126, -2.4422,  1.8501,
        -5.0655, -3.0033, -2.1707, -2.3381], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.1679, -1.3905, -1.3954, -1.1949, -1.4327, -1.4304],
        [-1.4002, -1.4818, -2.1000, -1.4119, -1.6644, -0.7929],
        [-1.9360, -1.3360, -3.0060, -1.1685, -1.8158, -1.3956],
        [-5.2740, -1.0552, -1.7413, -1.5475, -2.6363, -1.5475],
        [-2.1208, -2.8289, -2.8266, -1.3309, -2.1579, -0.9339],
        [-4.2315, -3.0092, -2.0408, -1.5198, -2.2846, -1.6318],
        [-1.2210, -1.3616, -1.4090, -1.0937, -1.6368, -1.2765],
        [-2.2716, -1.3852, -2.7746, -1.3289, -1.6393, -0.7859],
        [-2.4755, -1.5408, -3.6905, -1.3614, -1.4806, -1.5187],
        [-5.4050, -1.2631, -4.3807, -1.1129, -1.3988, -1.4783],
        [-3.1823, -1.2540, -1.2372, -1.2207, -1.9892, -3.0959],
        [-6.2208, -2.0387, -1.8716, -1.1346, -1.9120, -1.2456],
        [-4.3810, -2.1446, -2.2413, -1.1711, -2.1356, -1.7342],
        [-1.3499, -1.2628, -1.4181, -1.0516, -1.9382, -1.2280],
        [-2.0743, -1.3731, -1.9856, -1.2186, -1.5794, -1.9782],
        [-2.6705, -1.8163, -1.6568, -1.2940, -1.4635, -1.1263],
        [-3.8498, -1.1513, -1.2809, -1.9348, -2.8940, -1.4373],
        [-7.1497, -1.9706, -2.1426, -2.4622, -2.5510, -1.1447],
        [-3.8619, -2.9130, -2.2583, -1.6456, -2.1372, -1.6838],
        [-1.3259, -1.4655, -2.1597, -1.1770, -1.4175, -1.2735]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.0118,  -8.8512, -10.6579, -13.8019, -12.1989, -14.7177,  -7.9987,
        -10.1856, -12.0676, -15.0388, -11.9792, -14.4233, -13.8078,  -8.2487,
        -10.2094, -10.0274, -12.5481, -17.4209, -14.4998,  -8.8190],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.6106e-01, -9.2004e-01, -7.1646e-01, -2.5156e-01, -2.4576e-01,
         -7.7430e-01],
        [-1.5075e+00, -6.8355e-02, -2.1061e+00, -2.7710e-01, -1.4913e+00,
         -1.7279e-01],
        [-1.6826e+00, -1.0612e-01, -3.7594e+00, -3.4016e-02, -2.2629e+00,
         -4.4546e-02],
        [-4.8330e+00, -4.2316e-02, -1.1764e+00, -9.4771e-01, -3.3421e+00,
         -1.2749e+00],
        [-1.0755e+00, -2.7440e+00, -3.7651e+00, -4.7956e-01, -2.4333e+00,
         -3.7785e-01],
        [-2.9994e+00, -4.0984e+00, -2.2203e+00, -3.1511e-01, -3.0218e+00,
         -8.6694e-01],
        [-2.6350e-02, -8.3546e-01, -6.9020e-01, -4.1722e-02, -9.5000e-01,
         -4.2519e-02],
        [-2.7939e+00, -9.0009e-02, -3.1354e+00, -1.2380e-01, -1.4120e+00,
         -2.7758e-02],
        [-2.0659e+00, -4.1663e-01, -4.4984e+00, -5.1529e-01, -9.0054e-01,
         -1.3989e+00],
        [-5.3136e+00, -3.9968e-01, -5.0529e+00, -5.1359e-02, -7.8224e-01,
         -7.0157e-01],
        [-3.1285e+00, -1.5468e-02, -4.1807e-01, -1.7492e-02, -2.1785e+00,
         -2.7488e+00],
        [-4.6507e+00, -1.3314e+00, -2.0529e+00, -1.5721e-02, -1.8612e+00,
         -1.6698e-02],
        [-3.5705e+00, -1.6904e+00, -2.3572e+00, -1.9511e-01, -2.7032e+00,
         -1.5513e+00],
        [-1.3899e+00, -5.2339e-02, -7.2782e-01, -2.5536e-02, -2.0005e+00,
         -1.1325e-05],
        [-2.6194e+00, -8.0224e-03, -2.0835e+00, -5.4109e-04, -9.8460e-01,
         -1.3075e+00],
        [-2.3521e+00, -8.2306e-01, -7.8583e-01, -2.4959e-01, -1.0486e+00,
         -3.4463e-01],
        [-3.3772e+00, -4.1008e-05, -1.8197e-01, -1.7393e+00, -4.0796e+00,
         -7.9572e-01],
        [-5.2814e+00, -1.4652e+00, -2.4862e+00, -2.7242e+00, -3.0695e+00,
         -2.5995e-01],
        [-3.0925e+00, -3.5124e+00, -3.2209e+00, -4.0475e-01, -2.4616e+00,
         -1.4441e+00],
        [-1.3554e+00, -1.1815e+00, -2.2929e+00, -1.1499e-01, -7.0235e-03,
         -1.6999e-03]], device='cuda:0')
tensor([ -3.0692,  -5.6233,  -7.8896, -11.6163, -10.8754, -13.5220,  -2.5862,
         -7.5829,  -9.7956, -12.3014,  -8.5068,  -9.9287, -12.0677,  -4.1961,
         -7.0036,  -5.6038, -10.1738, -15.2864, -14.1362,  -4.9535],
       device='cuda:0')
tensor([-4.9426, -3.2280, -2.7682, -2.1856, -1.3235, -1.1957, -5.4125, -2.6026,
        -2.2720, -2.7374, -3.4724, -4.4946, -1.7401, -4.0525, -3.2057, -4.4236,
        -2.3742, -2.1344, -0.3636, -3.8655], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.6879, -1.6172, -2.1405, -1.7666, -1.2916, -1.0810],
        [-1.5497, -1.1802, -1.3230, -1.1061, -1.2886, -1.7727],
        [-1.9981, -5.1046, -1.2731, -2.3850, -1.3363, -1.1596],
        [-1.2991, -1.7474, -2.3103, -3.6055, -1.6340, -1.7532],
        [-2.1853, -1.6968, -1.8497, -1.6149, -2.9651, -2.5513],
        [-4.7271, -0.9628, -1.9427, -2.4202, -3.1776, -2.3958],
        [-4.1628, -1.7596, -2.2818, -1.3529, -2.2487, -2.6557],
        [-5.1470, -1.9020, -5.6928, -2.2499, -1.6158, -1.4074],
        [-1.3628, -2.2647, -2.9669, -0.8819, -1.2398, -1.0427],
        [-2.0399, -1.8762, -1.4524, -1.4024, -1.9881, -2.3798],
        [-1.2582, -1.5094, -2.0508, -1.2539, -1.4678, -3.1494],
        [-3.4665, -0.9746, -4.5997, -1.5662, -2.3166, -1.3283],
        [-3.4935, -1.4367, -1.1014, -1.0677, -3.1809, -3.8354],
        [-2.4518, -2.0809, -2.3373, -1.2154, -1.3674, -2.6256],
        [-5.6586, -1.9900, -2.0513, -1.1262, -3.3501, -1.4415],
        [-1.3769, -1.2039, -1.3442, -1.9206, -1.6217, -1.1959],
        [-5.1416, -1.2372, -2.8645, -1.2276, -1.1728, -1.0722],
        [-3.7382, -1.2829, -2.5940, -1.7712, -2.7644, -1.0145],
        [-0.9115, -1.8826, -2.2894, -2.6846, -1.6614, -1.1101],
        [-5.6643, -2.0795, -2.7210, -2.7953, -2.2913, -1.7363]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.5848,  -8.2203, -13.2568, -12.3493, -12.8631, -15.6262, -14.4616,
        -18.0149,  -9.7587, -11.1388, -10.6895, -14.2519, -14.1157, -12.0783,
        -15.6178,  -8.6633, -12.7159, -13.1651, -10.5397, -17.2876],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.7138e+00, -3.2284e-01, -2.4319e+00, -1.2118e+00, -3.2585e-01,
         -1.2584e-01],
        [-7.9477e-01, -4.6682e-04, -1.9904e-02, -2.0334e-03, -5.6301e-03,
         -1.1303e+00],
        [-1.2638e+00, -6.3515e+00, -7.2508e-01, -2.1591e+00, -8.8469e-01,
         -4.8204e-01],
        [-1.2627e+00, -1.6960e+00, -2.9487e+00, -3.3731e+00, -1.7024e+00,
         -1.1031e+00],
        [-2.4075e+00, -1.0039e+00, -1.8735e+00, -1.3721e+00, -3.2740e+00,
         -2.5402e+00],
        [-6.0622e+00, -2.5713e-02, -1.7645e+00, -1.7699e+00, -3.6940e+00,
         -2.5982e+00],
        [-3.0230e+00, -4.6337e-01, -3.0502e+00, -2.3327e-01, -2.9491e+00,
         -2.9128e+00],
        [-3.6609e+00, -1.2080e+00, -4.9039e+00, -1.4603e+00, -2.3952e-01,
         -3.9985e-01],
        [-1.2147e-02, -2.8701e+00, -3.6378e+00, -1.1790e-02, -1.0005e-01,
         -5.5996e-03],
        [-2.0241e+00, -2.8158e+00, -4.6048e-01, -2.7507e-01, -2.7685e+00,
         -2.0654e+00],
        [-1.2417e+00, -9.6798e-05, -2.1848e+00, -3.0966e-02, -9.3609e-01,
         -2.5930e+00],
        [-4.5323e+00, -9.2142e-03, -5.5028e+00, -8.3459e-01, -2.2404e+00,
         -1.8835e-01],
        [-3.6241e+00, -1.0955e+00, -6.9463e-02, -2.2434e-02, -4.3256e+00,
         -4.2746e+00],
        [-1.3064e+00, -1.5534e+00, -3.3602e+00, -2.3669e-01, -2.5525e-02,
         -2.2632e+00],
        [-3.8599e+00, -1.2601e+00, -2.0267e+00, -3.3654e-01, -5.3858e+00,
         -1.0903e-01],
        [-1.3222e-01, -1.5131e-02, -3.8150e-01, -1.2476e+00, -1.1008e+00,
         -3.5429e-02],
        [-5.3944e+00, -9.3570e-01, -3.2353e+00, -4.2915e-06, -2.2571e-01,
         -6.6078e-04],
        [-3.3497e+00, -2.1385e-01, -3.0530e+00, -1.4468e+00, -4.3298e+00,
         -2.6086e-01],
        [-3.8318e-01, -1.1211e+00, -3.0515e+00, -2.7366e+00, -1.8436e+00,
         -5.6206e-02],
        [-3.6061e+00, -1.6595e+00, -4.3882e+00, -3.0627e+00, -2.6227e+00,
         -1.4767e+00]], device='cuda:0')
tensor([ -7.1319,  -1.9531, -11.8662, -12.0862, -12.4712, -15.9145, -12.6316,
        -11.8724,  -6.6375, -10.4093,  -6.9867, -13.3077, -13.4117,  -8.7454,
        -12.9780,  -2.9127,  -9.7917, -12.6539,  -9.1921, -16.8159],
       device='cuda:0')
tensor([-3.4529, -6.2672, -1.3905, -0.2632, -0.3919,  0.2883, -1.8299, -6.1424,
        -3.1213, -0.7295, -3.7028, -0.9442, -0.7040, -3.3329, -2.6398, -5.7506,
        -2.9242, -0.5112, -1.3476, -0.4718], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-5.3118, -1.3060, -1.1340, -1.2429, -2.0305, -2.5788],
        [-3.3621, -2.6067, -2.1872, -1.0643, -2.7011, -1.4728],
        [-3.7047, -2.9041, -2.1409, -2.4230, -3.0342, -1.4805],
        [-1.2803, -1.2487, -1.2256, -1.5987, -1.5044, -1.3573],
        [-2.2012, -1.1886, -2.7772, -1.2124, -2.1546, -1.7081],
        [-6.6895, -1.5223, -1.9502, -1.6288, -1.9482, -1.5590],
        [-2.4178, -1.8333, -1.8690, -2.3994, -1.5492, -1.2618],
        [-4.2686, -2.2001, -2.2146, -4.0188, -3.0247, -2.4281],
        [-3.9488, -2.7761, -4.1620, -1.1238, -2.2242, -1.3759],
        [-1.1795, -1.4964, -1.8419, -1.7187, -1.1749, -1.1688],
        [-1.6297, -1.5146, -2.2457, -1.5097, -1.4091, -1.6542],
        [-3.5079, -1.1301, -2.5927, -1.2840, -1.7043, -1.4264],
        [-2.9890, -1.1916, -1.3014, -1.2386, -1.7946, -2.6768],
        [-1.2800, -1.3984, -1.4665, -1.3132, -1.9245, -2.8066],
        [-4.5795, -3.1999, -2.2055, -1.0579, -1.9486, -1.3556],
        [-4.6356, -4.3499, -2.0761, -1.2628, -1.7916, -1.3027],
        [-1.3274, -1.2296, -1.3932, -0.9938, -1.4616, -1.1132],
        [-3.3701, -1.4587, -2.3456, -1.2905, -2.1101, -1.0929],
        [-5.6537, -1.4275, -3.6097, -1.4605, -1.5609, -1.2179],
        [-3.3104, -1.2917, -1.3953, -2.0626, -2.1253, -1.3069]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.6039, -13.3943, -15.6874,  -8.2151, -11.2420, -15.2981, -11.3306,
        -18.1550, -15.6108,  -8.5802,  -9.9630, -11.6454, -11.1919, -10.1892,
        -14.3470, -15.4187,  -7.5188, -11.6679, -14.9303, -11.4922],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.7391e+00, -6.1348e-02, -7.1548e-02, -6.1178e-04, -2.3125e+00,
         -2.3118e+00],
        [-2.4276e+00, -2.7531e+00, -2.9048e+00, -1.1874e-01, -3.7310e+00,
         -6.4620e-01],
        [-2.4747e+00, -2.9103e+00, -2.2066e+00, -1.5435e+00, -4.4441e+00,
         -5.2724e-01],
        [-6.4463e-02, -4.9989e-02, -6.7122e-02, -6.7738e-01, -1.2863e-01,
         -4.3343e-01],
        [-2.4313e+00, -1.7034e-01, -3.0319e+00, -3.7274e-03, -2.6715e+00,
         -1.1186e+00],
        [-4.6505e+00, -1.0376e-01, -1.6291e+00, -1.1912e+00, -2.6663e+00,
         -1.1493e+00],
        [-2.0102e+00, -1.0403e+00, -1.8857e+00, -2.2650e+00, -1.5097e+00,
         -2.6847e-02],
        [-3.2285e+00, -2.0086e+00, -3.1402e+00, -4.8161e+00, -4.4356e+00,
         -3.1645e+00],
        [-2.4674e+00, -2.5610e+00, -3.6246e+00, -5.1388e-02, -2.5472e+00,
         -2.1954e-01],
        [-1.5314e-01, -1.5819e+00, -1.4337e+00, -1.1634e+00, -4.3751e-01,
         -2.3346e-01],
        [-1.8946e+00, -1.6253e-03, -2.5408e+00, -6.2172e-01, -4.2494e-01,
         -1.7654e+00],
        [-3.3823e+00, -9.2396e-02, -3.3932e+00, -9.1847e-02, -1.9079e+00,
         -2.5219e-01],
        [-2.9126e+00, -1.6698e-02, -1.8858e-01, -1.6068e-01, -1.9726e+00,
         -3.3171e+00],
        [-1.4530e+00, -3.6765e-02, -1.0213e+00, -6.1450e-01, -1.9451e+00,
         -2.3051e+00],
        [-3.1567e+00, -3.9193e+00, -2.9751e+00, -5.2190e-04, -2.0157e+00,
         -1.1719e-02],
        [-3.9728e+00, -4.2013e+00, -1.9934e+00, -3.3768e-01, -1.6260e+00,
         -5.7939e-01],
        [-1.2650e+00, -3.3811e-01, -7.9628e-01, -6.6519e-05, -4.7535e-01,
         -6.5049e-03],
        [-3.9198e+00, -6.8799e-02, -2.8124e+00, -1.0549e-03, -2.5833e+00,
         -1.8689e-01],
        [-4.1049e+00, -9.6590e-02, -4.7294e+00, -7.6746e-01, -1.6669e+00,
         -7.7244e-02],
        [-3.0388e+00, -1.3536e-01, -1.0790e+00, -1.9004e+00, -2.6910e+00,
         -5.0956e-01]], device='cuda:0')
tensor([ -9.4969, -12.5815, -14.1064,  -1.4210,  -9.4274, -11.3901,  -8.7376,
        -20.7933, -11.4711,  -5.0031,  -7.2491,  -9.1198,  -8.5682,  -7.3757,
        -12.0790, -12.7105,  -2.8814,  -9.5723, -11.4425,  -9.3542],
       device='cuda:0')
tensor([-4.1070, -0.8128, -1.5810, -6.7941, -1.8146, -3.9081, -2.5930,  2.6384,
        -4.1397, -3.5770, -2.7138, -2.5256, -2.6237, -2.8135, -2.2680, -2.7082,
        -4.6374, -2.0956, -3.4878, -2.1380], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-5.5339, -2.4787, -1.7248, -1.8082, -1.9642, -2.0252],
        [-4.2677, -1.9809, -2.5587, -1.2487, -1.8342, -1.4928],
        [-1.1121, -1.5016, -1.7650, -1.0534, -1.5607, -1.4503],
        [-1.6914, -1.4250, -3.0147, -1.3059, -1.3488, -0.8391],
        [-1.5360, -1.2721, -2.2055, -1.3028, -1.4413, -1.4759],
        [-5.8765, -1.2606, -3.0848, -1.2110, -1.6564, -1.6403],
        [-3.1379, -1.0947, -1.5949, -1.2263, -1.8237, -1.3907],
        [-5.0074, -2.6973, -2.1445, -1.1962, -2.0346, -1.3536],
        [-2.9811, -4.5637, -3.0689, -0.9224, -1.6022, -1.3857],
        [-1.3283, -1.4566, -1.5498, -1.0707, -1.4238, -1.2702],
        [-4.5531, -1.6558, -1.6771, -1.3946, -1.5006, -1.2737],
        [-2.6179, -1.1270, -1.3912, -1.2018, -1.7847, -1.4192],
        [-6.9992, -1.1820, -5.0900, -1.3039, -1.7347, -1.5867],
        [-1.2065, -1.0600, -1.8260, -1.1675, -2.2705, -1.6317],
        [-5.0989, -2.4317, -1.8946, -1.1329, -2.0645, -1.4452],
        [-2.3295, -3.3555, -2.2028, -1.3323, -2.7730, -1.3867],
        [-1.9710, -1.1642, -1.2315, -0.9804, -1.7115, -1.5603],
        [-3.8402, -1.3502, -1.8980, -1.2397, -2.0013, -0.9494],
        [-7.9571, -1.4064, -1.8563, -1.1769, -1.7003, -1.2861],
        [-1.9828, -1.2138, -1.7236, -1.2883, -1.8709, -1.7738]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-15.5351, -13.3831,  -8.4431,  -9.6249,  -9.2336, -14.7297, -10.2681,
        -14.4335, -14.5240,  -8.0994, -12.0550,  -9.5417, -17.8965,  -9.1622,
        -14.0678, -13.3797,  -8.6189, -11.2788, -15.3832,  -9.8532],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.3645e+00, -2.5860e+00, -1.3504e+00, -1.5647e+00, -2.0437e+00,
         -2.5193e+00],
        [-3.3159e+00, -1.3740e+00, -3.0438e+00, -2.0827e-01, -1.5683e+00,
         -6.6365e-01],
        [-2.6431e-01, -1.3785e+00, -1.4833e+00, -1.0163e-02, -8.4170e-01,
         -7.7595e-01],
        [-1.9081e+00, -6.0680e-02, -3.9591e+00, -1.6243e-02, -4.3866e-01,
         -7.5953e-02],
        [-1.1004e+00, -7.8015e-03, -2.3474e+00, -6.5450e-01, -1.1427e+00,
         -1.1844e+00],
        [-5.0733e+00, -1.0190e-01, -3.0795e+00, -1.0538e-01, -1.5660e+00,
         -1.1497e+00],
        [-3.1870e+00, -7.1993e-02, -1.3584e+00, -1.5793e-01, -1.8695e+00,
         -9.3416e-01],
        [-3.9361e+00, -2.5762e+00, -2.9139e+00, -1.2074e-02, -2.1973e+00,
         -3.6110e-02],
        [-2.1937e+00, -4.5270e+00, -3.9618e+00, -2.3524e-03, -1.3345e+00,
         -6.1421e-01],
        [-1.3593e+00, -1.1718e+00, -1.0713e+00, -2.2795e-01, -8.9405e-02,
         -2.6276e-01],
        [-4.7125e+00, -3.6461e-01, -9.0947e-01, -1.7333e-01, -8.8695e-01,
         -1.0209e+00],
        [-2.4013e+00, -7.0210e-02, -4.5799e-01, -1.7090e-02, -2.1527e+00,
         -5.7907e-02],
        [-5.1327e+00, -9.4121e-02, -4.7222e+00, -2.5895e-01, -1.9019e+00,
         -1.1982e+00],
        [-9.6284e-01, -4.1799e-02, -1.8720e+00, -1.5099e-01, -2.5507e+00,
         -8.6676e-01],
        [-3.6052e+00, -2.4219e+00, -1.7720e+00, -1.4586e-01, -2.4366e+00,
         -2.3805e-01],
        [-1.3112e+00, -3.4719e+00, -2.3034e+00, -2.8036e-01, -4.0146e+00,
         -6.2290e-01],
        [-2.8624e+00, -5.1999e-02, -1.7625e-01, -1.5736e-04, -1.4890e+00,
         -1.1512e+00],
        [-4.0080e+00, -3.6087e-02, -1.8581e+00, -2.2352e-02, -2.1891e+00,
         -6.0547e-02],
        [-5.0762e+00, -4.9791e-02, -1.5111e+00, -1.2836e-02, -1.9545e+00,
         -4.2603e-01],
        [-1.9309e+00, -4.0333e-01, -9.4633e-01, -2.4655e-01, -2.1182e+00,
         -1.5317e+00]], device='cuda:0')
tensor([-13.4287, -10.1739,  -4.7540,  -6.4587,  -6.4372, -11.0758,  -7.5790,
        -11.6716, -12.6335,  -4.1825,  -8.0677,  -5.1572, -13.3080,  -6.4451,
        -10.6197, -12.0043,  -5.7310,  -8.1742,  -9.0304,  -7.1769],
       device='cuda:0')
tensor([-2.1064, -3.2092, -3.6892, -3.1662, -2.7964, -3.6539, -2.6891, -2.7619,
        -1.8905, -3.9169, -3.9872, -4.3846, -4.5885, -2.7171, -3.4481, -1.3754,
        -2.8879, -3.1046, -6.3528, -2.6763], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.0531, -1.1977, -1.3520, -1.2418, -1.5951, -3.3278],
        [-3.3688, -1.7757, -1.7855, -1.1261, -2.7157, -1.3550],
        [-2.3842, -4.3538, -1.7108, -2.3480, -2.6450, -1.2789],
        [-1.2553, -1.1945, -1.2224, -1.0452, -1.7222, -1.5653],
        [-2.9730, -1.3240, -1.9662, -1.2583, -1.5581, -1.0188],
        [-5.3195, -1.4312, -2.4943, -1.2574, -1.6016, -1.2572],
        [-5.1918, -1.5339, -1.6599, -1.4399, -2.1116, -1.7322],
        [-1.6538, -1.1994, -1.6894, -1.2126, -2.2409, -1.1335],
        [-4.4369, -3.1629, -1.8789, -1.2145, -2.1224, -2.8427],
        [-4.9958, -2.1597, -1.8168, -0.9252, -1.8023, -1.3610],
        [-1.2680, -1.3381, -1.2510, -1.0483, -1.6492, -1.1068],
        [-1.8356, -1.2580, -2.0899, -1.6459, -2.5914, -1.4895],
        [-5.9815, -1.8142, -3.2424, -1.7814, -1.6212, -1.2341],
        [-1.9746, -1.7996, -1.8170, -3.1521, -1.8084, -1.3556],
        [-6.5691, -2.2414, -2.0417, -2.0412, -2.3831, -1.6350],
        [-6.2841, -2.7950, -1.9291, -1.7844, -1.9598, -1.5901],
        [-1.2066, -1.0835, -2.1456, -0.8387, -1.1474, -1.0801],
        [-3.7487, -1.2538, -1.7823, -1.4320, -1.4436, -1.0155],
        [-4.4672, -1.4473, -3.2208, -1.2525, -1.5098, -1.0658],
        [-3.4568, -1.4861, -3.3378, -1.1131, -1.4248, -2.0605]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.7675, -12.1268, -14.7207,  -8.0049, -10.0985, -13.3612, -13.6693,
         -9.1295, -15.6582, -13.0609,  -7.6614, -10.9102, -15.6747, -11.9073,
        -16.9115, -16.3424,  -7.5020, -10.6759, -12.9635, -12.8790],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.3276e+00, -4.4795e-01, -1.0956e+00, -2.0447e-01, -1.1647e+00,
         -2.8306e+00],
        [-2.0061e+00, -1.1226e-01, -9.5845e-01, -8.4212e-03, -4.3067e+00,
         -1.4878e-02],
        [-1.2816e+00, -4.5060e+00, -1.1345e+00, -1.6343e+00, -3.7913e+00,
         -6.4020e-02],
        [-8.0600e-01, -3.7134e-04, -2.4742e-01, -3.1812e-03, -1.2997e+00,
         -9.5449e-01],
        [-3.6014e+00, -1.6065e-01, -1.9304e+00, -1.1955e-01, -9.4771e-01,
         -1.8650e-01],
        [-4.2591e+00, -7.6122e-01, -2.9408e+00, -2.3005e-02, -1.3347e+00,
         -7.6514e-01],
        [-4.3655e+00, -2.9522e-01, -7.7787e-01, -5.6235e-01, -3.1528e+00,
         -1.0401e+00],
        [-1.4424e+00, -4.0253e-01, -1.6361e+00, -3.4315e-01, -2.7268e+00,
         -2.8729e-02],
        [-3.8132e+00, -3.9128e+00, -1.7988e+00, -5.8271e-01, -2.6019e+00,
         -4.3225e+00],
        [-3.9216e+00, -1.6940e+00, -1.3516e+00, -7.3610e-02, -1.4851e+00,
         -3.8350e-01],
        [-1.1436e+00, -1.7326e-01, -1.5736e-01, -2.5145e-01, -1.5777e+00,
         -2.4061e-01],
        [-1.9849e+00, -8.3455e-02, -2.3334e+00, -9.7699e-01, -3.0386e+00,
         -8.5109e-01],
        [-4.0900e+00, -5.1743e-01, -4.2217e+00, -1.5538e+00, -1.5111e+00,
         -5.3998e-01],
        [-1.7594e+00, -7.6311e-01, -2.1249e+00, -3.3827e+00, -2.0009e+00,
         -8.5413e-02],
        [-4.4656e+00, -2.0369e+00, -2.7712e+00, -2.3239e+00, -3.2503e+00,
         -1.1249e+00],
        [-5.1686e+00, -2.5875e+00, -1.7670e+00, -7.5238e-01, -1.8539e+00,
         -1.2765e+00],
        [-6.3909e-01, -9.7876e-02, -1.9562e+00, -5.9832e-02, -3.4294e-01,
         -2.6445e-01],
        [-3.8858e+00, -1.3712e-01, -1.3822e+00, -4.4294e-01, -7.3340e-01,
         -1.3221e-01],
        [-3.5567e+00, -8.4733e-02, -4.5549e+00, -7.9842e-02, -1.1862e+00,
         -2.6179e-01],
        [-3.3664e+00, -1.5686e+00, -3.6039e+00, -8.9128e-02, -9.7000e-01,
         -2.2155e+00]], device='cuda:0')
tensor([ -8.0708,  -7.4069, -12.4116,  -3.3112,  -6.9462, -10.0840, -10.1939,
         -6.5796, -17.0319,  -8.9095,  -3.5440,  -9.2684, -12.4340, -10.1164,
        -15.9728, -13.4058,  -3.3604,  -6.7136,  -9.7242, -11.8137],
       device='cuda:0')
tensor([-2.6968, -4.7199, -2.3090, -4.6937, -3.1524, -3.2772, -3.4754, -2.5499,
         1.3737, -4.1514, -4.1174, -1.6418, -3.2407, -1.7910, -0.9387, -2.9365,
        -4.1415, -3.9623, -3.2393, -1.0654], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.9326, -3.4099, -3.6039, -2.0367, -1.6055, -1.4076],
        [-2.0520, -2.1669, -1.4372, -2.9600, -1.5494, -1.1037],
        [-7.7466, -1.1846, -3.1395, -5.4361, -2.3236, -1.0588],
        [-1.8874, -1.1550, -1.5621, -0.9834, -1.9585, -1.2998],
        [-0.9841, -1.4700, -1.4189, -1.3096, -1.8125, -1.5448],
        [-7.3303, -2.2419, -1.9887, -1.1323, -3.2772, -2.8727],
        [-2.8108, -3.8468, -2.5927, -2.1952, -2.1459, -2.1526],
        [-1.4663, -1.1654, -1.2138, -1.3649, -1.2570, -1.2573],
        [-2.8354, -1.1915, -2.0564, -1.2145, -2.3700, -2.4176],
        [-3.6350, -1.5960, -2.6798, -2.0252, -2.0019, -1.9752],
        [-3.0130, -1.2175, -2.0292, -2.7273, -1.8220, -1.2171],
        [-3.6829, -2.1445, -2.1380, -2.3183, -2.6597, -2.8634],
        [-3.1821, -2.9201, -3.0482, -1.2671, -1.8849, -1.6958],
        [-1.2688, -1.1659, -2.1552, -0.8340, -1.3262, -0.9969],
        [-2.9398, -1.4311, -2.7672, -1.3573, -1.4755, -1.5049],
        [-5.3551, -1.3132, -2.0914, -1.1425, -1.8908, -1.2639],
        [-3.4723, -1.3293, -1.1970, -1.2674, -2.0603, -1.3177],
        [-2.4521, -1.9132, -2.1784, -1.2906, -1.9083, -1.2021],
        [-3.2903, -2.2435, -2.0149, -1.3404, -2.4362, -1.4524],
        [-1.6475, -1.4297, -1.6228, -1.0843, -1.4791, -1.4305]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-14.9961, -11.2692, -20.8891,  -8.8462,  -8.5399, -18.8431, -15.7439,
         -7.7246, -12.0854, -13.9131, -12.0261, -15.8068, -13.9982,  -7.7469,
        -11.4759, -13.0569, -10.6439, -10.9447, -12.7777,  -8.6939],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.0433e+00, -4.9446e+00, -4.2433e+00, -1.6317e+00, -1.2019e+00,
         -7.7529e-01],
        [-2.5510e+00, -2.9657e+00, -9.8237e-01, -2.5160e+00, -1.2672e+00,
         -7.9356e-02],
        [-6.4520e+00, -3.4083e-03, -3.7180e+00, -5.8059e+00, -2.5551e+00,
         -5.3776e-02],
        [-2.0178e+00, -6.4081e-01, -8.9631e-01, -8.3208e-05, -1.9561e+00,
         -8.3672e-01],
        [-9.1953e-01, -1.2774e-03, -2.5828e-01, -8.6632e-02, -1.6754e+00,
         -5.4579e-01],
        [-5.3918e+00, -1.9308e+00, -1.8165e+00, -4.6345e-01, -4.4338e+00,
         -3.0884e+00],
        [-2.0347e+00, -4.3290e+00, -3.1404e+00, -1.2055e+00, -2.6627e+00,
         -2.6070e+00],
        [-1.7331e+00, -1.4292e-02, -2.5638e-01, -7.1086e-01, -7.8234e-01,
         -7.5090e-01],
        [-3.1492e+00, -5.8156e-01, -2.1971e+00, -1.9543e-01, -2.9197e+00,
         -2.3552e+00],
        [-3.3732e+00, -1.4525e-02, -3.5506e+00, -1.9539e+00, -2.8922e+00,
         -1.5148e+00],
        [-2.5614e+00, -8.5208e-03, -2.3244e+00, -3.0311e+00, -2.1702e+00,
         -1.7338e-02],
        [-2.5909e+00, -1.6939e+00, -3.1981e+00, -2.8532e+00, -3.7740e+00,
         -4.1654e+00],
        [-2.2433e+00, -2.9881e+00, -3.9740e+00, -1.5815e-01, -1.8356e+00,
         -1.5793e+00],
        [-9.1526e-01, -6.4218e-01, -1.9721e+00, -3.2473e-02, -9.7931e-01,
         -1.0876e-01],
        [-3.5801e+00, -1.7506e-02, -3.6429e+00, -1.2087e-01, -7.5960e-01,
         -5.6814e-01],
        [-3.9899e+00, -1.4000e-01, -2.1608e+00, -2.6071e-02, -2.6658e+00,
         -5.3099e-01],
        [-3.0067e+00, -4.6619e-02, -3.9855e-02, -1.4614e-01, -2.7810e+00,
         -3.6864e-01],
        [-1.6226e+00, -1.3714e+00, -2.6529e+00, -7.5269e-01, -1.8220e+00,
         -2.1073e-03],
        [-2.0492e+00, -1.9208e+00, -2.1487e+00, -1.8576e-01, -3.5108e+00,
         -1.4720e-03],
        [-2.4772e+00, -1.2692e+00, -1.1148e+00, -7.3546e-02, -4.1727e-01,
         -6.0581e-01]], device='cuda:0')
tensor([-14.8402, -10.3616, -18.5883,  -6.3479,  -3.4869, -17.1249, -15.9794,
         -4.2479, -11.3982, -13.2992, -10.1129, -18.2756, -12.7785,  -4.6500,
         -8.6892,  -9.5136,  -6.3889,  -8.2238,  -9.8168,  -5.9579],
       device='cuda:0')
tensor([-0.1559, -0.9076, -2.3009, -2.4983, -5.0530, -1.7183,  0.2355, -3.4768,
        -0.6873, -0.6138, -1.9132,  2.4688, -1.2197, -3.0968, -2.7867, -3.5433,
        -4.2550, -2.7209, -2.9609, -2.7360], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.2209, -1.5534, -1.9838, -0.9876, -1.3539, -1.1098],
        [-1.7305, -1.2508, -3.8566, -1.3612, -1.6810, -0.9440],
        [-4.7589, -1.3914, -2.4710, -1.2868, -1.9116, -1.2757],
        [-1.4586, -1.2830, -3.2301, -2.2038, -2.3717, -1.2986],
        [-3.7685, -2.2850, -2.1954, -3.6712, -2.5684, -1.8897],
        [-4.0898, -4.1180, -2.7774, -1.0803, -1.6405, -1.4399],
        [-1.1621, -1.2213, -1.8250, -1.5505, -1.1497, -1.1907],
        [-4.2543, -1.4829, -2.7605, -1.4541, -1.5383, -0.9215],
        [-1.6122, -1.2492, -3.1104, -1.5043, -1.6602, -1.5850],
        [-3.6687, -1.5508, -1.6607, -1.5829, -1.6616, -2.3357],
        [-6.9042, -2.6377, -2.0765, -1.0411, -2.0133, -1.8650],
        [-6.3564, -3.9661, -2.1782, -3.5206, -1.6806, -1.2526],
        [-1.1215, -2.0127, -1.4344, -6.5961, -1.2758, -1.5051],
        [-1.3779, -2.0450, -3.2870, -1.3007, -1.7865, -2.4964],
        [-3.2984, -1.1986, -1.8635, -1.2337, -2.5299, -1.4189],
        [-1.7082, -1.7318, -2.0949, -3.1957, -2.3094, -1.1965],
        [-5.4507, -2.8056, -2.0008, -1.5452, -2.4129, -1.7492],
        [-4.1172, -3.2168, -2.4739, -0.9783, -1.7220, -1.3277],
        [-1.1497, -1.3481, -1.7324, -0.9519, -1.1793, -1.0574],
        [-3.5431, -1.5237, -2.0482, -1.8283, -1.3146, -1.3910]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.2095, -10.8240, -13.0955, -11.8457, -16.3782, -15.1459,  -8.0994,
        -12.4115, -10.7212, -12.4605, -16.5378, -18.9545, -13.9457, -12.2936,
        -11.5430, -12.2365, -15.9644, -13.8359,  -7.4187, -11.6489],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.9771e-02, -1.5019e+00, -1.9859e+00, -1.4019e-04, -1.2270e-01,
         -7.8917e-05],
        [-1.8530e+00, -1.8576e-01, -3.9337e+00, -1.0739e-01, -1.3751e+00,
         -1.2995e-01],
        [-3.8203e+00, -4.8230e-02, -3.0927e+00, -3.5462e-01, -2.6295e+00,
         -6.8583e-01],
        [-1.1220e+00, -1.4144e-01, -4.3200e+00, -1.8591e+00, -2.8205e+00,
         -2.2560e-01],
        [-2.9854e+00, -2.1350e+00, -3.2286e+00, -4.6645e+00, -3.8392e+00,
         -1.8295e+00],
        [-2.5381e+00, -3.8804e+00, -2.1986e+00, -6.1095e-02, -8.2507e-01,
         -4.9773e-01],
        [-3.4877e-01, -6.6247e-01, -1.4133e+00, -9.2911e-01, -3.7670e-04,
         -2.4623e-01],
        [-4.3155e+00, -4.2816e-02, -3.8690e+00, -4.3794e-01, -1.0371e+00,
         -1.2505e-04],
        [-1.2586e+00, -1.4509e-01, -3.7792e+00, -8.2651e-01, -1.9068e+00,
         -6.1327e-01],
        [-3.4736e+00, -5.4667e-01, -1.3504e+00, -8.8899e-01, -1.7540e+00,
         -3.0878e+00],
        [-4.6288e+00, -2.8967e+00, -2.9066e+00, -8.8534e-02, -2.1974e+00,
         -1.4162e+00],
        [-4.5893e+00, -3.6901e+00, -2.1924e+00, -2.6881e+00, -1.3405e+00,
         -3.8267e-02],
        [-9.3194e-02, -2.4323e+00, -7.1729e-01, -4.9365e+00, -1.7738e-02,
         -1.0011e+00],
        [-1.3086e+00, -1.1817e+00, -4.2277e+00, -3.0204e-01, -1.8905e+00,
         -1.8776e+00],
        [-2.6272e+00, -6.2157e-01, -1.4520e+00, -2.2726e-01, -3.3502e+00,
         -7.8973e-01],
        [-1.4604e+00, -8.0035e-01, -2.5947e+00, -3.5932e+00, -3.3904e+00,
         -5.0842e-01],
        [-4.1719e+00, -2.9519e+00, -2.4041e+00, -1.4019e+00, -3.4845e+00,
         -1.4000e+00],
        [-2.4544e+00, -3.0396e+00, -1.9327e+00, -2.6230e-03, -1.1818e+00,
         -4.9821e-02],
        [-3.8592e-01, -1.0461e+00, -1.2523e+00, -9.3446e-02, -5.3003e-03,
         -7.2109e-03],
        [-3.4942e+00, -7.0949e-02, -2.1252e+00, -1.3934e+00, -8.3804e-02,
         -5.9138e-01]], device='cuda:0')
tensor([ -3.6305,  -7.5849, -10.6311, -10.4886, -18.6821, -10.0010,  -3.6002,
         -9.7025,  -8.5295, -11.1014, -14.1343, -14.5386,  -9.1980, -10.7880,
         -9.0680, -12.3476, -15.8144,  -8.6609,  -2.7902,  -7.7589],
       device='cuda:0')
tensor([-4.5790, -3.2392, -2.4643, -1.3572,  2.3039, -5.1449, -4.4991, -2.7090,
        -2.1916, -1.3590, -2.4035, -4.4159, -4.7477, -1.5055, -2.4750,  0.1110,
        -0.1500, -5.1750, -4.6285, -3.8900], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.3513, -1.3392, -2.5879, -2.5426, -2.2176, -2.9185],
        [-2.4202, -0.9596, -2.1504, -2.3052, -2.1067, -1.5206],
        [-4.0538, -2.1616, -2.3334, -2.1826, -3.0418, -2.2082],
        [-3.3396, -3.6627, -2.2303, -1.6325, -1.7348, -1.3373],
        [-1.2736, -1.1853, -1.9394, -1.6597, -1.4689, -1.0535],
        [-3.0038, -1.3619, -2.6920, -1.4448, -1.7447, -2.9682],
        [-2.9988, -1.2754, -2.6072, -1.1682, -2.0115, -1.3589],
        [-3.1962, -1.4089, -1.3853, -1.7591, -1.8575, -2.1414],
        [-1.3449, -1.2822, -1.5841, -1.1518, -1.4139, -1.1481],
        [-8.0377, -2.3368, -2.0443, -1.0456, -2.2974, -1.9451],
        [-4.2489, -3.7842, -2.7213, -1.0112, -2.0026, -1.2821],
        [-1.1923, -1.0818, -1.9237, -0.8285, -1.9047, -0.9661],
        [-2.2786, -1.4578, -2.3575, -1.3859, -1.3488, -1.1221],
        [-1.6015, -1.3046, -1.2225, -1.5559, -1.3409, -1.5839],
        [-2.7243, -3.0501, -1.5938, -1.8969, -1.2302, -1.2393],
        [-1.1893, -1.6000, -2.3617, -2.7883, -1.5123, -4.3842],
        [-2.4689, -2.5169, -2.1596, -1.7634, -1.9020, -2.5079],
        [-4.6043, -1.0107, -2.4423, -2.2240, -3.1938, -1.9669],
        [-6.3826, -2.0514, -2.3504, -1.5584, -2.2626, -2.3354],
        [-5.7184, -4.3776, -1.8585, -3.2783, -2.3509, -1.9025]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-14.9571, -11.4626, -15.9815, -13.9373,  -8.5803, -13.2153, -11.4200,
        -11.7484,  -7.9250, -17.7070, -15.0504,  -7.8971,  -9.9507,  -8.6093,
        -11.7346, -13.8358, -13.3187, -15.4420, -16.9408, -19.4862],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.8683e+00, -3.3405e-01, -3.1106e+00, -2.7799e+00, -2.1070e+00,
         -2.8101e+00],
        [-3.1146e+00, -4.9694e-03, -2.1739e+00, -1.6904e+00, -2.0559e+00,
         -1.3731e+00],
        [-3.0854e+00, -1.2182e+00, -3.0936e+00, -2.4918e+00, -4.7253e+00,
         -2.3111e+00],
        [-1.8811e+00, -3.2891e+00, -1.6236e+00, -7.1880e-01, -7.8307e-01,
         -9.2366e-02],
        [-7.0311e-02, -6.2634e-01, -1.4505e+00, -1.3166e+00, -1.3024e+00,
         -1.9862e-01],
        [-3.5188e+00, -8.5541e-02, -3.1820e+00, -3.9551e-01, -1.6734e+00,
         -2.8000e+00],
        [-2.4998e+00, -4.6709e-01, -3.1346e+00, -2.6530e-03, -2.7784e+00,
         -9.0502e-01],
        [-2.9612e+00, -6.4441e-02, -1.7138e-01, -1.1693e+00, -2.3972e+00,
         -1.9659e+00],
        [-1.2112e+00, -5.0391e-02, -1.3689e+00, -2.0240e-01, -5.3963e-01,
         -3.5125e-02],
        [-6.0434e+00, -2.3171e+00, -2.4246e+00, -3.0531e-01, -3.1083e+00,
         -2.0205e+00],
        [-3.0060e+00, -3.7879e+00, -2.4491e+00, -3.5715e-04, -2.0625e+00,
         -5.0201e-02],
        [-5.2553e-01, -3.9750e-02, -1.6449e+00, -9.0481e-03, -2.0706e+00,
         -3.1126e-04],
        [-2.4721e+00, -5.5796e-02, -2.9695e+00, -2.3547e-01, -4.1314e-01,
         -6.0478e-01],
        [-9.5800e-01, -6.1904e-02, -3.0243e-04, -7.7668e-01, -7.3586e-02,
         -4.9348e-01],
        [-1.9576e+00, -4.0078e+00, -1.4375e+00, -1.4465e+00, -3.0683e-01,
         -5.1177e-01],
        [-1.2266e+00, -9.3239e-01, -2.9883e+00, -2.6750e+00, -1.5201e+00,
         -3.8296e+00],
        [-3.0613e+00, -2.3014e+00, -2.5415e+00, -1.6919e+00, -1.5739e+00,
         -2.4018e+00],
        [-5.1567e+00, -1.2003e-02, -2.4897e+00, -1.4448e+00, -3.7481e+00,
         -2.4614e+00],
        [-4.2747e+00, -8.8257e-01, -3.2405e+00, -8.3281e-01, -3.0682e+00,
         -2.4891e+00],
        [-3.7902e+00, -4.4531e+00, -1.2531e+00, -2.4061e+00, -3.1144e+00,
         -1.6975e+00]], device='cuda:0')
tensor([-15.0100, -10.4130, -16.9254,  -8.3879,  -4.9648, -11.6552,  -9.7876,
         -8.7295,  -3.4076, -16.2192, -11.3561,  -4.2901,  -6.7508,  -2.3639,
         -9.6680, -13.1720, -13.5718, -15.3127, -14.7878, -16.7144],
       device='cuda:0')
tensor([ 0.0529, -1.0496,  0.9439, -5.5494, -3.6156, -1.5601, -1.6324, -3.0189,
        -4.5174, -1.4878, -3.6942, -3.6070, -3.1999, -6.2453, -2.0666, -0.6639,
         0.2531, -0.1294, -2.1530, -2.7718], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.7729, -1.2538, -1.4321, -1.8757, -1.6810, -1.7663],
        [-4.2104, -2.6375, -1.6447, -1.3608, -2.4767, -1.1913],
        [-2.2495, -4.7121, -2.0740, -1.0871, -2.2850, -1.4277],
        [-1.1949, -2.6428, -1.5487, -1.8857, -1.3472, -1.2726],
        [-4.2420, -1.3631, -2.0635, -1.3070, -1.7805, -0.9357],
        [-2.5575, -1.1898, -1.5046, -1.2860, -1.4211, -1.3534],
        [-3.0134, -1.2557, -1.8318, -1.7186, -1.3193, -1.4034],
        [-1.6859, -2.1526, -1.3905, -2.0247, -1.5453, -1.5406],
        [-1.8041, -2.4055, -2.2300, -3.1568, -1.4564, -2.4223],
        [-2.2709, -1.0255, -4.2550, -2.3527, -2.9665, -1.0230],
        [-2.6879, -1.9127, -2.3772, -1.4795, -2.1383, -1.4335],
        [-3.3167, -2.7530, -4.5626, -7.4960, -1.5822, -1.4430],
        [-1.4068, -2.3094, -3.4674, -0.9362, -1.1662, -1.1652],
        [-1.4691, -1.7177, -1.6684, -1.3232, -2.0958, -1.9893],
        [-3.5176, -1.4713, -1.9572, -1.3344, -1.8337, -2.8898],
        [-2.0979, -1.2326, -3.5682, -5.0946, -1.1567, -1.2414],
        [-6.0738, -2.3797, -2.0558, -1.8655, -2.4900, -1.4229],
        [-2.4641, -2.5331, -1.9608, -1.3435, -2.3145, -1.6290],
        [-1.1594, -2.0104, -2.1849, -1.2989, -1.2834, -1.2226],
        [-1.6968, -1.4417, -3.3607, -1.2967, -1.4847, -1.9237]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.7817, -13.5213, -13.8354,  -9.8920, -11.6918,  -9.3124, -10.5421,
        -10.3396, -13.4751, -13.8936, -12.0291, -21.1534, -10.4512, -10.2635,
        -13.0040, -14.3913, -16.2876, -12.2450,  -9.1596, -11.2043],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.4487e+00, -1.4845e-01, -7.5683e-01, -1.2163e+00, -1.6643e+00,
         -1.8051e+00],
        [-2.7945e+00, -2.7710e+00, -1.3916e+00, -7.3667e-01, -3.4146e+00,
         -1.0169e-03],
        [-7.2270e-01, -5.0482e+00, -2.0955e+00, -2.9546e-02, -2.8532e+00,
         -4.4604e-01],
        [-9.0562e-01, -3.3908e+00, -1.0859e+00, -1.3686e+00, -6.4170e-03,
         -2.1169e-01],
        [-4.1911e+00, -4.4497e-01, -2.1275e+00, -1.7564e-01, -1.9232e+00,
         -1.6788e-01],
        [-1.9187e+00, -3.8676e-02, -9.4886e-01, -1.7066e-01, -2.3000e-02,
         -3.9555e-02],
        [-2.1359e+00, -7.5983e-01, -1.6955e+00, -1.3710e+00, -6.0641e-01,
         -1.1109e+00],
        [-1.9650e+00, -2.8299e+00, -9.5173e-01, -1.4466e+00, -1.1034e+00,
         -9.0486e-01],
        [-1.7778e+00, -2.3928e+00, -2.5527e+00, -3.7138e+00, -8.0694e-01,
         -2.3049e+00],
        [-2.6072e+00, -4.6364e-02, -4.7258e+00, -1.7930e+00, -3.3055e+00,
         -4.4796e-02],
        [-1.8790e+00, -6.4040e-01, -2.8130e+00, -6.0845e-01, -2.3519e+00,
         -1.1386e+00],
        [-2.4801e+00, -2.6357e+00, -6.1417e+00, -5.8360e+00, -6.1290e-01,
         -7.4328e-01],
        [-1.3535e-01, -2.8325e+00, -3.9977e+00, -1.4845e-03, -6.5393e-03,
         -1.9512e-01],
        [-1.5049e+00, -2.3531e+00, -9.3259e-01, -6.4202e-02, -2.8048e+00,
         -1.3566e+00],
        [-2.8199e+00, -4.7024e-01, -1.6789e+00, -4.6841e-01, -2.5316e+00,
         -3.3168e+00],
        [-1.8005e+00, -7.3746e-02, -4.5020e+00, -4.1809e+00, -1.6342e-01,
         -6.7214e-02],
        [-4.1461e+00, -2.4297e+00, -2.6589e+00, -2.0430e+00, -3.5771e+00,
         -1.2075e-01],
        [-9.2689e-01, -2.3618e+00, -1.7331e+00, -2.9592e-01, -2.9769e+00,
         -1.2574e+00],
        [-6.6139e-01, -2.4095e+00, -2.2793e+00, -3.8361e-01, -6.1296e-02,
         -2.0637e-01],
        [-1.6342e+00, -3.6322e-02, -4.5141e+00, -1.2697e-01, -1.0844e+00,
         -1.7952e+00]], device='cuda:0')
tensor([ -7.0397, -11.1095, -11.1952,  -6.9690,  -9.0304,  -3.1394,  -7.6795,
         -9.2015, -13.5488, -12.5227,  -9.4313, -18.4497,  -7.1687,  -9.0162,
        -11.2859, -10.7878, -14.9756,  -9.5519,  -6.0014,  -9.1912],
       device='cuda:0')
tensor([-2.7420, -2.4118, -2.6403, -2.9229, -2.6615, -6.1730, -2.8626, -1.1381,
         0.0738, -1.3709, -2.5978, -2.7038, -3.2825, -1.2474, -1.7182, -3.6035,
        -1.3120, -2.6931, -3.1582, -2.0131], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.4955, -2.3339, -2.1336, -1.3887, -2.8060, -1.6132],
        [-1.1539, -1.1913, -1.1871, -0.9828, -1.7075, -1.5441],
        [-2.6277, -1.1941, -1.8722, -1.1319, -2.2044, -0.8897],
        [-5.5069, -1.4582, -2.4493, -1.2201, -2.1370, -1.1829],
        [-2.4695, -1.2131, -1.2147, -1.3863, -1.7605, -1.4032],
        [-4.0164, -2.6776, -1.8030, -1.2148, -2.8014, -1.6387],
        [-4.1419, -2.8822, -2.9070, -1.3281, -2.5541, -1.4097],
        [-1.1922, -1.4008, -1.6803, -1.7160, -1.7483, -1.2827],
        [-4.6539, -1.3377, -3.4156, -1.3980, -1.4929, -2.8661],
        [-1.3190, -2.2315, -1.6202, -1.6193, -1.4833, -1.7828],
        [-6.3944, -5.9167, -3.2340, -5.0843, -1.2505, -1.0517],
        [-4.1096, -1.9658, -1.2000, -5.0975, -1.2306, -1.0371],
        [-1.9852, -1.0286, -2.7435, -1.4590, -1.6787, -1.9537],
        [-2.9124, -1.0087, -3.6934, -8.3504, -1.9053, -1.6098],
        [-6.0072, -1.5824, -1.8607, -1.1159, -2.3930, -2.3656],
        [-4.2988, -2.0468, -2.1238, -1.1947, -2.2892, -2.4422],
        [-4.9571, -2.2469, -2.1707, -1.1187, -2.5550, -2.4362],
        [-1.0808, -1.2338, -1.1134, -1.4465, -1.8502, -1.1683],
        [-4.1011, -2.5939, -2.4861, -1.3375, -1.2457, -0.8345],
        [-2.5384, -1.2625, -1.4082, -1.1791, -2.3422, -2.9543]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.7710,  -7.7668,  -9.9200, -13.9543,  -9.4473, -14.1519, -15.2231,
         -9.0204, -15.1641, -10.0561, -22.9316, -14.6406, -10.8488, -19.4801,
        -15.3248, -14.3955, -15.4846,  -7.8931, -12.5987, -11.6847],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.6178e+00, -1.9318e+00, -2.1109e+00, -5.0076e-01, -3.8259e+00,
         -1.2895e+00],
        [-3.2117e-01, -4.8991e-03, -3.8805e-02, -1.0067e-01, -1.3668e+00,
         -1.0826e+00],
        [-2.9089e+00, -2.9941e-01, -1.7609e+00, -8.1004e-03, -2.6367e+00,
         -1.8563e-03],
        [-3.9922e+00, -5.3709e-03, -3.0320e+00, -5.1021e-02, -3.2099e+00,
         -4.0807e-01],
        [-2.4897e+00, -1.8546e-01, -8.6655e-02, -5.4515e-01, -1.7979e+00,
         -5.5337e-01],
        [-2.7314e+00, -2.7703e+00, -1.9827e+00, -2.8884e-01, -4.2096e+00,
         -1.3549e+00],
        [-3.3545e+00, -3.0858e+00, -3.9389e+00, -1.9999e-01, -4.0875e+00,
         -7.7832e-04],
        [-1.1667e+00, -3.5035e-01, -1.1146e+00, -1.2698e+00, -1.5576e+00,
         -2.0907e-01],
        [-4.4480e+00, -6.5774e-01, -4.6196e+00, -5.9010e-01, -9.6332e-01,
         -3.2343e+00],
        [-1.0742e+00, -1.1579e+00, -1.2866e+00, -7.0257e-01, -1.7362e-01,
         -1.0554e+00],
        [-5.7932e+00, -6.5084e+00, -3.7692e+00, -5.2188e+00, -3.9883e-01,
         -1.0990e-01],
        [-7.1823e+00, -2.8563e+00, -7.2725e-02, -5.6846e+00, -1.1457e+00,
         -5.4430e-03],
        [-1.8881e+00, -3.4351e-01, -2.7678e+00, -1.1757e+00, -1.4257e+00,
         -2.0841e+00],
        [-3.5841e+00, -1.0452e-01, -4.9382e+00, -7.2730e+00, -1.9016e+00,
         -1.5443e+00],
        [-5.8114e+00, -1.3648e+00, -1.5816e+00, -4.2335e-01, -2.4402e+00,
         -2.6089e+00],
        [-2.7881e+00, -8.9670e-01, -3.0345e+00, -1.4831e-02, -3.1332e+00,
         -3.6791e+00],
        [-4.6628e+00, -2.0361e+00, -2.1644e+00, -3.9417e-02, -3.7808e+00,
         -3.1765e+00],
        [-2.0004e-01, -1.9894e-01, -4.7086e-01, -8.4790e-01, -1.6912e+00,
         -1.5783e-01],
        [-4.7763e+00, -2.4788e+00, -2.5174e+00, -3.9303e-01, -1.4852e-01,
         -5.9309e-02],
        [-2.2007e+00, -7.5741e-01, -5.3566e-01, -3.1944e-02, -2.8922e+00,
         -3.1669e+00]], device='cuda:0')
tensor([-12.2765,  -2.9149,  -7.6159, -10.6985,  -5.6583, -13.3377, -14.6675,
         -5.6682, -14.5131,  -5.4502, -21.7983, -16.9471,  -9.6849, -19.3457,
        -14.2303, -13.5465, -15.8601,  -3.5668, -10.3734,  -9.5848],
       device='cuda:0')
tensor([-1.4945, -4.8519, -2.3041, -3.2558, -3.7890, -0.8141, -0.5556, -3.3522,
        -0.6510, -4.6060, -1.1333,  2.3065, -1.1638, -0.1344, -1.0945, -0.8490,
         0.3754, -4.3263, -2.2253, -2.1000], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.1946, -1.3018, -2.3096, -1.4969, -1.4791, -1.3741],
        [-1.9258, -1.4977, -2.7829, -1.3693, -1.5063, -1.1642],
        [-4.5233, -1.3160, -4.5970, -1.1963, -1.5791, -1.2581],
        [-5.2504, -1.3232, -1.1170, -1.8318, -1.6640, -1.3254],
        [-3.9562, -1.9713, -2.2211, -1.1249, -1.8969, -1.3797],
        [-2.0748, -2.1767, -2.1120, -1.8109, -1.9026, -1.4346],
        [-1.2023, -1.1711, -1.1128, -1.0172, -1.7582, -1.2952],
        [-4.8924, -1.3556, -2.1101, -1.1731, -2.0182, -0.9614],
        [-8.0587, -1.4180, -2.7029, -1.2177, -1.3616, -1.1394],
        [-3.5663, -1.7562, -1.8809, -1.4436, -1.9934, -2.2094],
        [-1.8733, -1.1945, -1.7782, -1.0995, -1.7604, -2.2119],
        [-3.1422, -2.4364, -1.8508, -1.1397, -2.2150, -1.3480],
        [-3.5821, -2.4417, -2.0586, -2.0219, -2.3947, -1.7816],
        [-1.1826, -1.5954, -1.8343, -1.6150, -1.4309, -1.1498],
        [-2.3033, -1.4534, -2.6654, -1.3776, -1.5355, -0.9634],
        [-2.7692, -1.3209, -1.8350, -1.1917, -1.3755, -1.2319],
        [-4.5440, -1.2292, -1.8815, -1.0957, -2.3067, -1.4243],
        [-2.0666, -1.2053, -1.1114, -1.1061, -2.0469, -2.4683],
        [-4.2310, -2.1986, -2.8706, -1.0669, -2.0290, -1.3757],
        [-4.0910, -3.2917, -2.5184, -1.7167, -1.9827, -1.4222]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.1560, -10.2461, -14.4698, -12.5118, -12.5501, -11.5116,  -7.5569,
        -12.5108, -15.8982, -12.8498,  -9.9178, -12.1322, -14.2806,  -8.8080,
        -10.2986,  -9.7241, -12.4814, -10.0046, -13.7718, -15.0227],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-6.9653e-01, -5.7739e-01, -2.6471e+00, -6.9823e-01, -2.8989e-01,
         -5.6145e-01],
        [-2.2931e+00, -3.7545e-02, -3.8246e+00, -1.3017e-01, -8.2892e-01,
         -1.8154e-01],
        [-3.3610e+00, -7.8774e-02, -5.6952e+00, -1.6956e-01, -1.6184e+00,
         -2.8698e-01],
        [-4.3457e+00, -4.7165e-02, -8.2872e-03, -1.4672e+00, -1.6930e+00,
         -4.5693e-01],
        [-3.1368e+00, -1.3020e+00, -2.7608e+00, -5.8783e-02, -1.8447e+00,
         -6.1955e-01],
        [-5.4904e-01, -1.7974e+00, -2.4404e+00, -5.4872e-01, -1.6059e+00,
         -4.5195e-02],
        [-9.4552e-01, -8.3837e-02, -5.2281e-02, -1.0018e-02, -1.4880e+00,
         -2.4132e-01],
        [-5.2443e+00, -7.1097e-02, -2.1474e+00, -7.5543e-02, -2.1968e+00,
         -1.2331e-01],
        [-5.1328e+00, -1.7121e-02, -3.3317e+00, -1.8722e-03, -9.8048e-02,
         -2.7826e-01],
        [-3.3594e+00, -7.3418e-01, -1.4350e+00, -6.0768e-01, -2.5545e+00,
         -1.7729e+00],
        [-1.8824e+00, -3.5014e-02, -1.7263e+00, -1.3234e-01, -1.7714e+00,
         -2.5041e+00],
        [-1.7968e+00, -2.5032e+00, -1.7116e+00, -2.3310e-01, -2.9077e+00,
         -3.5149e-02],
        [-2.6776e+00, -2.0913e+00, -2.0049e+00, -1.2577e+00, -2.9751e+00,
         -1.7748e+00],
        [-6.9925e-01, -1.5884e+00, -1.7147e+00, -8.1886e-01, -3.7012e-01,
         -7.1409e-02],
        [-2.6390e+00, -4.1115e-04, -3.4683e+00, -1.9305e-01, -1.0267e+00,
         -3.2647e-02],
        [-2.2411e+00, -8.2809e-02, -1.4544e+00, -1.2469e-01, -5.9619e-01,
         -2.1311e-02],
        [-4.2111e+00, -5.3998e-01, -1.3828e+00, -3.2740e-02, -2.9722e+00,
         -8.3390e-01],
        [-2.3561e+00, -2.6377e-03, -2.3179e-01, -1.6911e-03, -2.1641e+00,
         -2.0775e+00],
        [-3.2668e+00, -1.6515e+00, -4.1706e+00, -5.4252e-04, -2.2956e+00,
         -2.0967e-01],
        [-3.5827e+00, -3.4879e+00, -2.9150e+00, -7.5307e-01, -2.2299e+00,
         -6.4490e-01]], device='cuda:0')
tensor([ -5.4705,  -7.2959, -11.2098,  -8.0183,  -9.7227,  -6.9867,  -2.8209,
         -9.8584,  -8.8598, -10.4638,  -8.0515,  -9.1876, -12.7814,  -5.2627,
         -7.3601,  -4.5205,  -9.9727,  -6.8339, -11.5947, -13.6134],
       device='cuda:0')
tensor([-3.6855, -2.9502, -3.2600, -4.4935, -2.8274, -4.5250, -4.7360, -2.6524,
        -7.0384, -2.3861, -1.8662, -2.9445, -1.4992, -3.5453, -2.9386, -5.2036,
        -2.5087, -3.1707, -2.1771, -1.4093], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-5.1434, -2.1112, -1.8027, -2.3032, -2.2199, -1.8843],
        [-7.6901, -3.2502, -2.3696, -0.9895, -2.2547, -1.3386],
        [-1.1772, -1.1511, -2.8503, -0.8542, -1.4365, -1.0308],
        [-2.5157, -1.4479, -2.6557, -1.4386, -1.3694, -1.7924],
        [-5.2632, -1.4196, -2.3040, -1.1454, -1.4537, -1.2141],
        [-3.1349, -1.3953, -3.1564, -1.1851, -1.9233, -2.0501],
        [-2.7959, -1.1918, -1.6084, -1.2133, -2.7633, -1.7810],
        [-6.5755, -2.7800, -1.9934, -1.2355, -1.7203, -1.3051],
        [-3.0760, -3.6918, -2.1158, -1.4800, -1.9360, -1.6255],
        [-1.1369, -1.6794, -1.8154, -1.3977, -1.4343, -1.3432],
        [-5.1116, -1.4348, -4.6546, -1.5151, -1.9528, -0.9338],
        [-9.2472, -1.2713, -1.9326, -1.3348, -3.4940, -1.2736],
        [-1.8879, -2.3564, -1.3703, -3.0254, -3.2617, -1.2571],
        [-6.8686, -2.3417, -1.8622, -2.6302, -2.4113, -2.0605],
        [-2.8077, -2.7996, -3.0544, -1.4373, -2.2429, -1.4454],
        [-1.2267, -1.6023, -2.0216, -1.4659, -1.2887, -1.6025],
        [-2.0533, -1.5097, -2.3227, -1.6161, -1.4013, -1.1456],
        [-3.2830, -1.1350, -3.3557, -1.8004, -1.5093, -1.3344],
        [-2.6918, -1.6075, -1.3211, -1.8367, -2.1133, -1.2595],
        [-7.5055, -2.5430, -1.7897, -1.1417, -2.1046, -1.4733]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-15.4647, -17.8927,  -8.5002, -11.2198, -12.7999, -12.8449, -11.3536,
        -15.6098, -13.9251,  -8.8070, -15.6027, -18.5535, -13.1587, -18.1746,
        -13.7873,  -9.2078, -10.0487, -12.4177, -10.8298, -16.5578],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.6145e+00, -1.6533e+00, -1.4036e+00, -2.8570e+00, -2.8315e+00,
         -1.9655e+00],
        [-5.3607e+00, -3.2233e+00, -1.9900e+00, -3.7348e-04, -2.8126e+00,
         -1.1959e-01],
        [-2.1594e-01, -4.5496e-01, -3.0423e+00, -3.3774e-02, -1.1282e+00,
         -7.0668e-03],
        [-3.0462e+00, -3.4383e-02, -3.3417e+00, -3.5127e-01, -2.9320e-01,
         -7.8655e-01],
        [-3.9056e+00, -5.4075e-02, -2.6537e+00, -5.6131e-02, -1.2991e+00,
         -2.2305e-02],
        [-2.8285e+00, -1.0834e+00, -2.7742e+00, -1.4154e-01, -2.3172e+00,
         -2.2697e+00],
        [-3.1795e+00, -3.9513e-01, -1.6699e+00, -1.6635e-02, -3.4613e+00,
         -1.0316e+00],
        [-4.3342e+00, -3.3656e+00, -2.4067e+00, -3.8355e-01, -9.3216e-01,
         -8.2961e-02],
        [-2.5587e+00, -3.8127e+00, -1.9291e+00, -6.0959e-01, -2.2634e+00,
         -1.2214e+00],
        [-3.7061e-01, -1.8425e+00, -1.7534e+00, -3.9839e-01, -6.5255e-02,
         -2.4370e-01],
        [-5.3734e+00, -2.9206e-05, -6.4658e+00, -5.1591e-01, -2.3072e+00,
         -2.0661e-02],
        [-5.9836e+00, -9.5474e-03, -1.8417e+00, -4.5810e-01, -5.5085e+00,
         -3.5769e-02],
        [-1.6455e+00, -1.7175e+00, -1.0033e+00, -3.2928e+00, -4.4261e+00,
         -3.8174e-01],
        [-4.5890e+00, -2.3028e+00, -1.9841e+00, -3.0300e+00, -3.1687e+00,
         -2.4309e+00],
        [-1.7285e+00, -2.6523e+00, -4.1098e+00, -2.6807e-01, -2.6592e+00,
         -7.7339e-01],
        [-9.4910e-01, -1.7163e+00, -1.9536e+00, -6.4604e-01, -8.2952e-02,
         -1.3607e+00],
        [-1.9712e+00, -6.9882e-02, -2.7629e+00, -8.8547e-01, -6.4097e-01,
         -4.9815e-01],
        [-3.0992e+00, -8.1588e-02, -4.5488e+00, -1.3013e+00, -1.4806e+00,
         -2.4056e-04],
        [-2.8082e+00, -6.0278e-01, -8.1095e-01, -1.5118e+00, -2.6541e+00,
         -6.0792e-01],
        [-4.6880e+00, -2.7694e+00, -1.2502e+00, -3.6796e-01, -2.3783e+00,
         -4.3023e-01]], device='cuda:0')
tensor([-14.3254, -13.5067,  -4.8823,  -7.8533,  -7.9908, -11.4145,  -9.7540,
        -11.5052, -12.3950,  -4.6739, -14.6829, -13.8372, -12.4669, -17.5055,
        -12.1913,  -6.7086,  -6.8286, -10.5118,  -8.9958, -11.8840],
       device='cuda:0')
tensor([-1.1393, -4.3860, -3.6179, -3.3665, -4.8091, -1.4304, -1.5996, -4.1047,
        -1.5300, -4.1331, -0.9198, -4.7163, -0.6918, -0.6691, -1.5960, -2.4992,
        -3.2201, -1.9059, -1.8340, -4.6738], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.1365, -1.5845, -1.5174, -2.0897, -1.4327, -1.4448],
        [-4.3449, -1.4906, -1.8675, -2.2609, -2.4625, -3.7773],
        [-1.3856, -0.9437, -1.7306, -1.4814, -2.2093, -2.4067],
        [-8.6418, -2.5541, -1.9561, -1.5564, -2.0981, -0.8990],
        [-3.9810, -2.2044, -2.8866, -2.1342, -1.7074, -1.9831],
        [-1.4627, -1.5749, -1.7366, -0.9953, -1.3776, -1.7539],
        [-3.0480, -1.2009, -2.9855, -1.3726, -1.9135, -1.2251],
        [-3.2139, -1.3641, -2.2002, -1.3086, -1.5672, -1.3950],
        [-1.8496, -1.4689, -1.2394, -2.6066, -1.9276, -1.1919],
        [-2.2982, -3.5345, -1.3336, -2.6907, -2.0980, -1.2906],
        [-2.9131, -2.7303, -1.6954, -1.1784, -1.8874, -1.5264],
        [-1.3420, -1.2630, -1.4625, -0.9867, -1.9866, -1.2179],
        [-2.9726, -1.3183, -2.4652, -1.2613, -2.0212, -0.9961],
        [-2.3386, -1.4819, -2.3591, -1.2421, -1.4840, -1.5639],
        [-3.6939, -0.9918, -1.7697, -1.4047, -1.2942, -2.0428],
        [-2.3525, -1.3439, -1.2904, -1.2076, -1.7762, -2.9215],
        [-7.2198, -2.3360, -1.9693, -1.1458, -1.6898, -1.5732],
        [-4.1419, -3.7567, -1.9734, -1.1598, -2.6212, -1.8637],
        [-1.5958, -1.7860, -2.1738, -1.7312, -1.6356, -1.2059],
        [-3.5090, -1.5963, -1.8966, -1.4615, -1.6844, -1.0992]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.2056, -16.2037, -10.1573, -17.7056, -14.8966,  -8.9009, -11.7456,
        -11.0491, -10.2840, -13.2456, -11.9310,  -8.2587, -11.0347, -10.4697,
        -11.1971, -10.8921, -15.9339, -15.5167, -10.1283, -11.2470],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-6.6109e-01, -1.3448e+00, -1.1157e+00, -1.6190e+00, -9.5747e-01,
         -7.7527e-01],
        [-4.5308e+00, -6.3956e-01, -1.6889e+00, -2.4946e+00, -2.5249e+00,
         -3.6311e+00],
        [-9.2403e-01, -8.6788e-03, -1.5661e+00, -7.3599e-01, -2.3415e+00,
         -3.0056e+00],
        [-5.7133e+00, -1.7468e+00, -2.1142e+00, -8.5270e-01, -2.4418e+00,
         -7.8298e-02],
        [-2.9478e+00, -1.8335e+00, -3.9182e+00, -1.1142e+00, -9.8340e-01,
         -2.2940e+00],
        [-1.6909e+00, -1.4155e+00, -1.3608e+00, -2.9753e-01, -4.6626e-01,
         -1.6867e+00],
        [-3.5079e+00, -2.0386e-02, -3.4550e+00, -3.6143e-01, -1.9729e+00,
         -4.1105e-01],
        [-2.7311e+00, -1.7345e-04, -2.1808e+00, -4.1096e-01, -1.4447e+00,
         -9.7333e-01],
        [-1.7132e+00, -3.8541e-01, -4.1652e-04, -2.4273e+00, -2.2052e+00,
         -1.1808e-01],
        [-1.5314e+00, -4.1784e+00, -5.5253e-01, -2.8856e+00, -2.2215e+00,
         -3.3540e-01],
        [-1.9551e+00, -2.9654e+00, -1.0705e+00, -3.5402e-02, -1.6178e+00,
         -1.0102e+00],
        [-1.4366e+00, -5.8647e-01, -8.0107e-01, -7.4580e-03, -2.1384e+00,
         -3.2635e-02],
        [-3.5108e+00, -1.4709e-01, -2.9552e+00, -6.0040e-02, -2.1275e+00,
         -5.9822e-01],
        [-2.0487e+00, -3.5519e-01, -2.5204e+00, -8.6561e-03, -6.4755e-01,
         -1.3343e+00],
        [-3.7540e+00, -6.7646e-02, -1.2698e+00, -5.5529e-01, -5.9914e-01,
         -2.5540e+00],
        [-2.4797e+00, -3.2089e-01, -6.7948e-01, -1.2231e-01, -1.6367e+00,
         -2.6307e+00],
        [-4.9753e+00, -2.2044e+00, -2.3715e+00, -3.1403e-01, -9.7931e-01,
         -7.8461e-01],
        [-2.5685e+00, -3.4299e+00, -1.6580e+00, -2.4565e-01, -3.1734e+00,
         -1.7157e+00],
        [-2.1147e+00, -1.9379e+00, -2.3463e+00, -1.0387e+00, -9.7743e-01,
         -1.7657e-01],
        [-3.6336e+00, -3.0119e-01, -1.7733e+00, -4.0476e-01, -1.5463e+00,
         -1.3149e-01]], device='cuda:0')
tensor([ -6.4732, -15.5099,  -8.5818, -12.9472, -13.0911,  -6.9177,  -9.7287,
         -7.7410,  -6.8496, -11.7049,  -8.6544,  -5.0027,  -9.3989,  -6.9148,
         -8.7998,  -7.8697, -11.6292, -12.7911,  -8.5916,  -7.7906],
       device='cuda:0')
tensor([-2.7323, -0.6938, -1.5755, -4.7584, -1.8055, -1.9832, -2.0169, -3.3080,
        -3.4344, -1.5407, -3.2766, -3.2560, -1.6358, -3.5548, -2.3973, -3.0223,
        -4.3048, -2.7256, -1.5367, -3.4564], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.9821, -1.5981, -2.5411, -1.4199, -1.3639, -1.0248],
        [-2.0010, -1.1782, -3.0542, -1.1497, -1.5569, -1.4691],
        [-4.1254, -1.5378, -3.1855, -1.3233, -1.3143, -1.0022],
        [-2.2706, -1.1893, -1.4402, -1.0121, -1.9453, -2.1018],
        [-4.2436, -2.1966, -1.8774, -1.1606, -1.7329, -1.5886],
        [-3.1970, -3.7936, -1.6720, -1.2139, -3.1306, -1.8685],
        [-1.2310, -1.3918, -2.3350, -1.7400, -1.4223, -1.2803],
        [-2.6450, -1.2903, -1.7783, -1.3017, -1.2936, -0.8788],
        [-5.0173, -1.4513, -2.8710, -1.2273, -1.4338, -1.1074],
        [-2.0938, -0.9805, -1.3467, -1.1504, -2.2838, -1.4822],
        [-2.6782, -1.2295, -1.8685, -1.1569, -3.1007, -1.9572],
        [-1.8885, -2.6407, -1.7148, -1.1074, -1.6357, -1.4046],
        [-3.9389, -4.1499, -1.6782, -1.1121, -2.6527, -2.0348],
        [-1.1779, -1.5985, -1.3767, -1.2892, -1.4716, -1.3790],
        [-2.7682, -1.3317, -2.3534, -1.5807, -1.7017, -1.1930],
        [-3.7290, -1.3576, -3.0928, -1.2141, -1.6192, -1.2314],
        [-0.8521, -1.2208, -1.2488, -1.6015, -2.3328, -1.7239],
        [-5.9675, -2.0996, -2.1405, -1.8955, -1.6823, -1.5425],
        [-2.9380, -3.6606, -2.2576, -1.2933, -2.0337, -1.5480],
        [-1.6870, -1.2016, -2.3227, -1.0288, -1.4866, -1.5790]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.9299, -10.4092, -12.4885,  -9.9594, -12.7997, -14.8756,  -9.4005,
         -9.1877, -13.1080,  -9.3374, -11.9911, -10.3916, -15.5665,  -8.2929,
        -10.9287, -12.2442,  -8.9799, -15.3279, -13.7312,  -9.3056],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.1118e+00, -3.5213e-01, -3.4000e+00, -3.1036e-01, -3.3124e-01,
         -2.9699e-01],
        [-1.7081e+00, -9.8596e-02, -3.8760e+00, -3.0385e-02, -1.4540e+00,
         -1.2009e+00],
        [-3.6696e+00, -1.6186e-01, -3.5279e+00, -2.8125e-01, -1.6444e-01,
         -3.4471e-02],
        [-2.8596e+00, -2.5921e-03, -9.2504e-01, -6.5501e-02, -1.8953e+00,
         -1.9584e+00],
        [-2.6606e+00, -1.8468e+00, -1.7752e+00, -3.1140e-01, -1.0301e+00,
         -7.3863e-01],
        [-1.9611e+00, -3.4341e+00, -1.0388e+00, -3.7500e-01, -3.9966e+00,
         -1.6330e+00],
        [-4.5308e-02, -9.4993e-01, -2.6305e+00, -1.1461e+00, -4.4526e-02,
         -2.5245e-01],
        [-3.0648e+00, -3.6615e-01, -1.2883e+00, -5.1603e-02, -2.9059e-01,
         -2.6209e-03],
        [-3.8474e+00, -4.0133e-02, -3.9758e+00, -1.5676e-01, -1.0791e+00,
         -2.4753e-01],
        [-2.1035e+00, -2.9036e-01, -2.7969e-01, -6.9340e-02, -2.5474e+00,
         -6.6907e-01],
        [-3.0582e+00, -3.1426e-02, -1.9702e+00, -6.8078e-02, -4.1490e+00,
         -1.7167e+00],
        [-6.4748e-01, -2.5604e+00, -1.2728e+00, -1.1963e-01, -5.9569e-01,
         -2.2103e-01],
        [-3.0791e+00, -3.9617e+00, -1.0562e+00, -2.0493e-01, -3.4003e+00,
         -2.1575e+00],
        [-6.5174e-01, -1.5590e+00, -7.1589e-01, -2.2426e-01, -1.9975e-01,
         -3.7136e-01],
        [-3.1121e+00, -3.4237e-03, -2.7689e+00, -7.8396e-01, -1.4452e+00,
         -2.9143e-01],
        [-2.9167e+00, -1.6465e-01, -3.8864e+00, -7.2795e-02, -1.7868e+00,
         -5.4728e-01],
        [-1.2646e-01, -4.9590e-02, -4.5245e-01, -9.1176e-01, -3.1864e+00,
         -1.4827e+00],
        [-3.9541e+00, -1.6147e+00, -2.9402e+00, -2.0438e+00, -7.1403e-01,
         -9.7895e-01],
        [-1.9252e+00, -4.2071e+00, -2.4973e+00, -1.9656e-01, -2.2973e+00,
         -7.7551e-01],
        [-2.3506e+00, -4.1970e-01, -2.5255e+00, -2.6764e-03, -8.1989e-01,
         -1.2078e+00]], device='cuda:0')
tensor([ -6.8026,  -8.3679,  -7.8395,  -7.7064,  -8.3627, -12.4386,  -5.0689,
         -5.0640,  -9.3467,  -5.9593, -10.9935,  -5.4171, -13.8597,  -3.7220,
         -8.4050,  -9.3746,  -6.2094, -12.2458, -11.8990,  -7.3262],
       device='cuda:0')
tensor([-3.1273, -2.0412, -4.6490, -2.2530, -4.4369, -2.4370, -4.3316, -4.1237,
        -3.7613, -3.3781, -0.9976, -4.9745, -1.7068, -4.5709, -2.5237, -2.8696,
        -2.7705, -3.0821, -1.8322, -1.9795], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.9486, -2.3205, -3.1287, -1.2170, -2.4316, -1.3163],
        [-1.2283, -1.1471, -1.2730, -2.3906, -1.6317, -1.1822],
        [-3.0862, -1.2703, -2.2997, -1.2895, -1.5029, -1.2748],
        [-3.1573, -1.2975, -1.7156, -1.4745, -1.3299, -1.1529],
        [-1.2015, -1.4229, -1.8901, -2.6041, -1.7559, -1.6930],
        [-4.6216, -3.6855, -1.9801, -2.1063, -2.4400, -1.3839],
        [-4.4584, -3.6793, -2.3358, -1.0103, -2.2666, -1.2145],
        [-1.8163, -1.9813, -2.2457, -1.7098, -1.2911, -1.3284],
        [-2.8314, -1.7335, -1.9211, -1.3477, -1.1501, -1.0956],
        [-2.5554, -1.3892, -1.4789, -1.1989, -1.3007, -1.0033],
        [-3.3475, -2.1532, -1.6791, -1.8333, -1.2366, -1.0832],
        [-1.5998, -2.2471, -2.0693, -2.2646, -1.1752, -2.5036],
        [-1.5965, -2.5368, -3.3945, -1.2874, -2.0499, -1.3601],
        [-2.8434, -1.0440, -1.7585, -1.2026, -2.1823, -1.4734],
        [-3.5019, -2.4795, -1.5526, -2.2880, -2.1597, -3.6784],
        [-6.4184, -4.6787, -1.9992, -1.2355, -2.9789, -2.6746],
        [-1.2019, -1.4208, -1.5540, -1.1986, -1.3766, -1.5730],
        [-3.2281, -1.1794, -4.4589, -1.5242, -1.7908, -2.1265],
        [-1.4003, -1.3615, -1.7080, -1.7673, -1.5756, -1.5246],
        [-0.8591, -1.3361, -2.5404, -4.4494, -1.3591, -1.6764]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.3626,  -8.8529, -10.7234, -10.1276, -10.5674, -16.2174, -14.9650,
        -10.3726, -10.0793,  -8.9264, -11.3329, -11.8595, -12.2252, -10.5042,
        -15.6602, -19.9854,  -8.3250, -14.3080,  -9.3373, -12.2205],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.0285e+00, -2.0981e+00, -3.0993e+00, -2.8725e-01, -3.1555e+00,
         -4.5269e-01],
        [-7.6306e-01, -3.3929e-01, -2.6828e-02, -2.6518e+00, -1.3421e+00,
         -1.0991e-01],
        [-3.4776e+00, -5.7045e-02, -2.3081e+00, -3.0465e-01, -1.0464e+00,
         -4.8438e-01],
        [-2.7207e+00, -7.5288e-02, -1.1888e+00, -8.2567e-01, -2.3093e-01,
         -1.1670e-02],
        [-5.2781e-01, -4.0076e-01, -1.9679e+00, -2.2934e+00, -1.8775e+00,
         -1.5423e+00],
        [-2.9554e+00, -4.5931e+00, -2.5077e+00, -1.9874e+00, -3.0259e+00,
         -1.0025e+00],
        [-3.6126e+00, -3.7960e+00, -2.5729e+00, -3.0800e-01, -2.7307e+00,
         -1.5156e-01],
        [-2.4791e+00, -2.1861e+00, -2.0306e+00, -1.2171e+00, -1.8728e-02,
         -4.3936e-01],
        [-2.6803e+00, -2.2526e+00, -1.7676e+00, -3.5845e-01, -2.2331e-03,
         -2.1648e-03],
        [-1.8842e+00, -1.3220e-04, -4.9238e-01, -1.1776e-01, -9.2773e-02,
         -8.8584e-04],
        [-2.4353e+00, -2.4309e+00, -1.4604e+00, -1.5616e+00, -3.1938e-01,
         -1.9430e-02],
        [-1.9325e+00, -3.0194e+00, -1.9340e+00, -1.8398e+00, -2.5825e-01,
         -2.1036e+00],
        [-1.3190e+00, -2.5499e+00, -4.1856e+00, -7.9432e-01, -2.0409e+00,
         -6.6885e-01],
        [-2.9674e+00, -5.9283e-02, -1.4975e+00, -3.2646e-01, -2.3749e+00,
         -1.3828e+00],
        [-2.4930e+00, -1.7661e+00, -9.7063e-01, -2.4751e+00, -2.6703e+00,
         -3.6240e+00],
        [-4.1866e+00, -4.5617e+00, -1.7013e+00, -1.4140e-01, -3.9225e+00,
         -3.1321e+00],
        [-4.3657e-01, -1.0972e+00, -8.3985e-01, -8.7363e-01, -4.9333e-01,
         -1.3821e+00],
        [-3.7146e+00, -6.5280e-02, -5.1071e+00, -7.2869e-01, -1.6058e+00,
         -1.3613e+00],
        [-1.2917e+00, -1.9091e-02, -1.0689e+00, -1.5970e+00, -1.3683e+00,
         -1.1444e+00],
        [-6.8903e-05, -1.4668e-01, -3.4166e+00, -4.3504e+00, -9.3341e-01,
         -1.2138e+00]], device='cuda:0')
tensor([-11.1213,  -5.2330,  -7.6782,  -5.0530,  -8.6097, -16.0721, -13.1719,
         -8.3709,  -7.0633,  -2.5881,  -8.2270, -11.0877, -11.5586,  -8.6083,
        -13.9991, -17.6456,  -5.1227, -12.5828,  -6.4893, -10.0609],
       device='cuda:0')
tensor([-2.2413, -3.6199, -3.0451, -5.0746, -1.9577, -0.1453, -1.7931, -2.0016,
        -3.0160, -6.3382, -3.1059, -0.7719, -0.6667, -1.8959, -1.6610, -2.3397,
        -3.2023, -1.7251, -2.8480, -2.1596], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-5.6796, -2.2509, -3.1371, -4.2056, -2.6119, -2.2246],
        [-3.0866, -1.5819, -3.4563, -1.0467, -1.5030, -1.3339],
        [-1.4321, -2.1407, -2.1366, -1.0840, -1.1243, -1.1503],
        [-4.0922, -1.5430, -1.8145, -1.9839, -1.4221, -1.3550],
        [-3.5584, -1.0849, -2.1432, -1.5783, -1.4080, -1.3163],
        [-0.8994, -1.3954, -1.0425, -1.5419, -2.2322, -1.3252],
        [-4.4308, -2.1041, -1.9137, -1.2609, -1.8633, -1.3904],
        [-4.7388, -3.1367, -1.6783, -1.3155, -1.7947, -1.4217],
        [-1.1427, -2.1485, -2.2500, -1.7870, -1.3630, -1.0872],
        [-2.3565, -1.6072, -2.8777, -1.5190, -1.2958, -1.1493],
        [-2.8221, -1.1642, -1.6702, -1.1920, -1.3031, -1.4524],
        [-1.7695, -1.2396, -1.4252, -1.4293, -1.4982, -1.7906],
        [-3.3807, -1.7732, -1.8483, -1.1089, -1.3882, -1.2429],
        [-7.3328, -2.4729, -1.8848, -1.1454, -2.7581, -1.5774],
        [-1.3010, -1.1898, -2.0944, -2.6221, -1.4432, -1.3910],
        [-3.1872, -1.2980, -2.8061, -1.2457, -1.3071, -0.9391],
        [-5.3154, -1.2564, -1.7264, -1.1492, -1.3755, -1.2471],
        [-2.2500, -1.2320, -4.2012, -1.1194, -2.9834, -1.3623],
        [-4.0836, -1.1327, -1.0459, -1.1269, -1.6908, -1.2626],
        [-5.5486, -1.3509, -2.4617, -1.1172, -1.8967, -1.2709]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-20.1098, -12.0084,  -9.0680, -12.2107, -11.0890,  -8.4365, -12.9632,
        -14.0858,  -9.7784, -10.8055,  -9.6041,  -9.1524, -10.7422, -17.1714,
        -10.0416, -10.7832, -12.0700, -13.1483, -10.3426, -13.6461],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.8626e+00, -1.9662e+00, -4.9913e+00, -4.9259e+00, -4.0873e+00,
         -2.5093e+00],
        [-1.6816e+00, -6.4777e-01, -2.7588e+00, -6.8435e-02, -1.3613e-01,
         -2.9062e-02],
        [-1.6091e+00, -3.7593e+00, -2.7408e+00, -1.5496e-03, -3.6944e-02,
         -5.5215e-01],
        [-3.7897e+00, -1.3122e-01, -1.8284e+00, -1.7969e+00, -1.1534e+00,
         -5.4948e-01],
        [-2.8418e+00, -1.9763e-01, -2.0836e+00, -9.9806e-01, -1.3097e+00,
         -4.4395e-01],
        [-4.0966e-01, -2.4415e-01, -5.6979e-02, -8.8078e-01, -2.9264e+00,
         -6.9186e-01],
        [-2.9036e+00, -1.7323e+00, -2.2410e+00, -8.2595e-01, -1.5996e+00,
         -2.6007e-01],
        [-3.4783e+00, -3.6634e+00, -7.7202e-01, -2.9433e-01, -1.4702e+00,
         -4.9345e-01],
        [-2.2538e-01, -2.6389e+00, -2.4340e+00, -1.1566e+00, -2.4991e-01,
         -4.6463e-02],
        [-2.3095e+00, -1.6035e-01, -4.2247e+00, -6.5093e-01, -3.3022e-01,
         -1.9969e-01],
        [-2.5308e+00, -1.4085e-01, -1.1167e+00, -7.3480e-02, -5.6389e-01,
         -9.8914e-01],
        [-1.7579e+00, -2.8809e-02, -3.9865e-01, -5.8642e-01, -1.1312e+00,
         -1.3235e+00],
        [-2.5763e+00, -1.1100e+00, -1.9163e+00, -1.2638e-01, -3.4051e-01,
         -3.8891e-01],
        [-4.7719e+00, -2.5400e+00, -1.7488e+00, -1.5484e-01, -4.4181e+00,
         -1.7215e-01],
        [-1.0454e+00, -2.2217e-01, -2.0993e+00, -2.2446e+00, -2.8173e-02,
         -5.7267e-01],
        [-3.3775e+00, -9.2962e-03, -3.7004e+00, -9.9853e-03, -3.1508e-01,
         -2.9204e-02],
        [-4.0634e+00, -2.6474e-03, -1.3630e+00, -4.1582e-02, -7.3481e-01,
         -3.1435e-01],
        [-2.1404e+00, -9.0142e-01, -4.4189e+00, -1.7535e-02, -3.9043e+00,
         -5.6540e-01],
        [-4.1985e+00, -4.2709e-02, -9.5165e-04, -5.9187e-04, -1.5174e+00,
         -4.7019e-01],
        [-4.3922e+00, -1.9329e-03, -3.2264e+00, -2.0986e-01, -1.8871e+00,
         -3.8568e-02]], device='cuda:0')
tensor([-22.3426,  -5.3218,  -8.6999,  -9.2491,  -7.8748,  -5.2099,  -9.5625,
        -10.1717,  -6.7513,  -7.8754,  -5.4149,  -5.2264,  -6.4584, -13.8058,
         -6.2122,  -7.4415,  -6.5197, -11.9480,  -6.2304,  -9.7561],
       device='cuda:0')
tensor([ 2.2327, -6.6866, -0.3681, -2.9616, -3.2142, -3.2267, -3.4007, -3.9141,
        -3.0272, -2.9301, -4.1892, -3.9260, -4.2838, -3.3656, -3.8293, -3.3417,
        -5.5502, -1.2003, -4.1122, -3.8901], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.3892, -2.3568, -2.5113, -1.5627, -2.0729, -1.7244],
        [-1.5509, -1.1448, -1.3136, -0.9139, -1.9007, -1.2899],
        [-3.1166, -1.6245, -3.0037, -1.3531, -1.9990, -2.2737],
        [-4.3982, -1.2703, -2.2582, -1.2046, -1.6632, -1.6659],
        [-2.2615, -1.3386, -1.3685, -2.9587, -1.5788, -1.6907],
        [-7.9512, -3.2340, -1.9294, -2.2512, -2.7282, -2.0437],
        [-4.1285, -3.6978, -2.0904, -1.1645, -1.8082, -1.6061],
        [-1.2921, -1.2111, -1.4157, -1.0504, -1.3559, -1.2320],
        [-3.3006, -1.3221, -1.8876, -1.6585, -1.4865, -0.9978],
        [-1.1434, -1.6077, -2.8065, -1.3644, -1.4142, -2.4668],
        [-3.5003, -1.0337, -2.1232, -1.1367, -1.3637, -1.2786],
        [-2.1623, -1.8664, -1.4343, -1.3551, -1.6689, -1.6329],
        [-3.7948, -3.2165, -2.1551, -1.1523, -1.9602, -1.8012],
        [-3.6435, -3.5102, -2.4023, -1.3670, -2.5676, -1.6004],
        [-1.2067, -1.6676, -2.4136, -2.4365, -1.6049, -1.4906],
        [-3.2386, -1.2844, -2.7744, -1.3485, -1.7256, -1.2775],
        [-3.1578, -1.3495, -1.9535, -1.2460, -1.6912, -1.9868],
        [-1.3796, -1.2575, -1.5052, -2.5351, -1.4599, -1.3337],
        [-3.8924, -3.3522, -1.5553, -1.1077, -1.9160, -1.3676],
        [-6.3642, -3.4567, -2.2131, -1.2676, -1.9696, -1.3009]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.6173,  -8.1137, -13.3705, -12.4605, -11.1969, -20.1377, -14.4956,
         -7.5572, -10.6531, -10.8032, -10.4362, -10.1200, -14.0801, -15.0911,
        -10.8200, -11.6492, -11.3848,  -9.4711, -13.1912, -16.5720],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.4033e+00, -2.1732e+00, -3.0160e+00, -4.6767e-01, -2.5100e+00,
         -1.5645e+00],
        [-1.9675e+00, -3.6794e-01, -4.4925e-01, -1.2244e-03, -1.9188e+00,
         -3.7094e-01],
        [-3.7254e+00, -5.2020e-01, -4.0414e+00, -2.4510e-01, -2.1186e+00,
         -1.5603e+00],
        [-3.4060e+00, -1.8092e-01, -2.2273e+00, -8.5407e-02, -1.9124e+00,
         -1.6113e+00],
        [-1.8692e+00, -2.5603e-03, -1.0458e+00, -3.1719e+00, -1.4960e+00,
         -1.5106e+00],
        [-5.1031e+00, -4.3784e+00, -2.3656e+00, -2.4778e+00, -4.0463e+00,
         -2.4559e+00],
        [-3.5309e+00, -4.1461e+00, -2.0514e+00, -7.7252e-02, -1.5516e+00,
         -1.2406e+00],
        [-1.3078e+00, -5.7158e-03, -6.1823e-01, -3.6144e-04, -9.2381e-02,
         -1.7918e-02],
        [-3.5517e+00, -3.0352e-02, -1.8425e+00, -9.9839e-01, -8.1171e-01,
         -2.0780e-02],
        [-6.9477e-01, -3.7847e-01, -3.6169e+00, -4.0757e-01, -8.6384e-01,
         -3.1860e+00],
        [-3.8377e+00, -3.8758e-02, -1.8532e+00, -5.7408e-03, -6.9566e-01,
         -1.8348e-01],
        [-2.2880e+00, -1.2966e+00, -1.0720e+00, -8.2796e-01, -1.6114e+00,
         -9.8911e-01],
        [-2.7932e+00, -3.4576e+00, -2.8254e+00, -8.8600e-02, -2.0633e+00,
         -1.6867e+00],
        [-2.1977e+00, -2.8330e+00, -2.4997e+00, -6.4913e-01, -2.8682e+00,
         -1.1975e+00],
        [-2.5468e-01, -1.7509e+00, -2.7594e+00, -1.9459e+00, -9.5311e-01,
         -8.8189e-01],
        [-3.0359e+00, -3.6221e-01, -3.0304e+00, -2.2257e-01, -1.8627e+00,
         -5.9362e-01],
        [-2.5710e+00, -1.5795e-04, -1.9128e+00, -2.0570e-01, -2.0475e+00,
         -2.4219e+00],
        [-1.2694e+00, -1.7031e-01, -8.5418e-01, -2.1373e+00, -7.9232e-01,
         -4.6577e-01],
        [-2.6418e+00, -4.3796e+00, -8.2492e-02, -1.9865e-01, -1.9270e+00,
         -2.2131e-03],
        [-5.0120e+00, -3.4938e+00, -2.3670e+00, -2.4699e-01, -2.0149e+00,
         -3.1517e-01]], device='cuda:0')
tensor([-12.1347,  -5.0757, -12.2111,  -9.4233,  -9.0960, -20.8272, -12.5979,
         -2.0424,  -7.2554,  -9.1476,  -6.6145,  -8.0850, -12.9148, -12.2451,
         -8.5458,  -9.1074,  -9.1590,  -5.6893,  -9.2317, -13.4499],
       device='cuda:0')
tensor([-1.4826, -3.0380, -1.1594, -3.0371, -2.1008,  0.6895, -1.8977, -5.5148,
        -3.3977, -1.6556, -3.8217, -2.0350, -1.1653, -2.8460, -2.2742, -2.5418,
        -2.2258, -3.7817, -3.9595, -3.1220], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.3667, -1.2145, -1.4406, -1.8324, -1.8790, -1.7453],
        [-2.6179, -2.3461, -1.8623, -1.6884, -1.5712, -1.0537],
        [-3.5507, -2.6587, -1.7494, -1.2609, -2.1673, -1.5472],
        [-1.6107, -1.5669, -1.5985, -1.0000, -1.4232, -1.3050],
        [-1.8860, -1.2349, -2.9309, -1.2451, -1.6093, -2.2465],
        [-4.1630, -1.4439, -2.6406, -1.2245, -1.4865, -1.5397],
        [-4.5705, -1.5041, -1.6615, -2.9969, -2.0749, -1.1630],
        [-4.0191, -1.7852, -2.1011, -2.9789, -2.1277, -1.4337],
        [-3.7840, -2.7909, -1.6222, -1.1288, -2.2914, -1.7773],
        [-1.4619, -1.4917, -1.8453, -1.1056, -1.2763, -1.4638],
        [-1.4853, -1.4319, -2.6684, -1.3405, -1.5305, -1.1661],
        [-3.5913, -1.3398, -2.0840, -1.2150, -1.4728, -1.1490],
        [-2.6565, -1.4353, -1.3459, -1.4358, -1.5294, -1.5328],
        [-1.1634, -1.1596, -1.6288, -1.1291, -1.7293, -2.1071],
        [-4.3280, -1.8862, -1.7340, -1.0922, -2.3202, -1.3443],
        [-2.1084, -3.6209, -2.5771, -1.5424, -1.9544, -1.8198],
        [-1.1193, -1.2147, -1.3819, -1.0165, -1.6133, -1.0924],
        [-3.1695, -1.3327, -1.8531, -1.2148, -1.4723, -2.9733],
        [-4.9272, -1.4726, -1.7969, -1.2323, -1.7856, -1.1403],
        [-2.6929, -1.4204, -1.8294, -1.6929, -1.4896, -1.2254]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.4785, -11.1396, -12.9341,  -8.5042, -11.1528, -12.4982, -13.9708,
        -14.4456, -13.3946,  -8.6446,  -9.6227, -10.8521,  -9.9358,  -8.9174,
        -12.7049, -13.6229,  -7.4381, -12.0156, -12.3550, -10.3506],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.2269e+00, -3.4792e-02, -4.3573e-01, -1.3885e+00, -2.0261e+00,
         -1.5988e+00],
        [-2.0021e+00, -2.0216e+00, -1.7868e+00, -1.3790e+00, -9.4274e-01,
         -1.0684e-03],
        [-2.4342e+00, -2.9978e+00, -9.6592e-01, -9.2478e-02, -2.5348e+00,
         -8.1647e-01],
        [-2.1702e+00, -1.4517e+00, -1.1135e+00, -8.4234e-02, -2.5931e-01,
         -2.6442e-01],
        [-2.1978e+00, -1.2025e-02, -3.4811e+00, -6.7804e-02, -1.2085e+00,
         -1.5592e+00],
        [-3.3386e+00, -1.5897e-01, -3.1409e+00, -1.0345e-01, -1.0047e+00,
         -1.1188e+00],
        [-3.8171e+00, -3.0219e-01, -1.7008e+00, -2.8633e+00, -2.7524e+00,
         -1.8688e-03],
        [-3.2588e+00, -9.5049e-01, -2.7265e+00, -3.6289e+00, -2.3689e+00,
         -3.6512e-01],
        [-2.3530e+00, -2.6179e+00, -8.8835e-01, -2.8262e-02, -2.7951e+00,
         -1.6708e+00],
        [-1.8382e+00, -1.3098e+00, -1.5393e+00, -1.3178e-01, -6.5757e-02,
         -9.5089e-01],
        [-1.4395e+00, -1.9135e-02, -3.6323e+00, -1.8375e-01, -9.0132e-01,
         -1.9753e-01],
        [-2.8211e+00, -7.9067e-02, -1.9429e+00, -6.9767e-02, -1.1239e+00,
         -2.5200e-03],
        [-2.3246e+00, -1.8650e-01, -3.2892e-01, -6.7157e-01, -1.3434e+00,
         -1.1508e+00],
        [-7.8395e-01, -1.5772e-02, -1.4735e+00, -2.7475e-01, -1.4895e+00,
         -2.0533e+00],
        [-3.0198e+00, -8.4537e-01, -1.2560e+00, -7.3416e-02, -3.2590e+00,
         -2.2383e-02],
        [-8.9976e-01, -3.4555e+00, -3.0090e+00, -7.0884e-01, -1.9919e+00,
         -1.8242e+00],
        [-4.7968e-01, -1.1291e-01, -7.1210e-01, -2.0688e-02, -1.0119e+00,
         -3.5465e-03],
        [-3.4965e+00, -1.1867e-02, -1.5999e+00, -3.2774e-03, -8.5705e-01,
         -2.2230e+00],
        [-3.4709e+00, -1.0106e-01, -1.1220e+00, -1.4029e-01, -2.2518e+00,
         -2.1626e-01],
        [-2.2764e+00, -2.1228e-01, -2.0242e+00, -1.1263e+00, -1.3028e+00,
         -1.8536e-01]], device='cuda:0')
tensor([ -7.7108,  -8.1334,  -9.8417,  -5.3433,  -8.5264,  -8.8655, -11.4376,
        -13.2988, -10.3535,  -5.8357,  -6.3735,  -6.0392,  -6.0058,  -6.0908,
         -8.4760, -11.8892,  -2.3408,  -8.1916,  -7.3022,  -7.1273],
       device='cuda:0')
tensor([-2.7678, -3.0062, -3.0924, -3.1609, -2.6264, -3.6327, -2.5332, -1.1469,
        -3.0411, -2.8088, -3.2491, -4.8128, -3.9300, -2.8265, -4.2289, -1.7337,
        -5.0973, -3.8240, -5.0527, -3.2233], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.1588, -1.1530, -1.7280, -1.3921, -1.8021, -1.4010],
        [-5.0996, -4.3217, -2.9337, -1.0581, -3.6342, -2.4298],
        [-5.1124, -2.2498, -3.6494, -0.9794, -2.3464, -1.2714],
        [-2.0530, -1.2349, -1.1742, -2.1064, -1.2982, -1.5064],
        [-2.1071, -1.0851, -2.1093, -1.1550, -1.9056, -1.0750],
        [-4.8220, -1.7566, -2.8736, -1.2050, -1.4829, -1.8430],
        [-5.6719, -1.1555, -3.4772, -1.3076, -1.9841, -1.4715],
        [-1.4662, -1.1231, -1.2759, -1.2707, -2.3179, -1.3895],
        [-4.7290, -2.6971, -2.0020, -1.1351, -2.5541, -1.3873],
        [-2.3995, -3.9589, -2.2086, -2.2397, -1.9738, -1.6450],
        [-1.3382, -1.0818, -1.0828, -0.9925, -1.4437, -1.1609],
        [-4.6639, -1.3424, -2.0958, -1.2705, -2.1862, -1.6999],
        [-2.2794, -1.4104, -2.1007, -1.7030, -1.3775, -1.1302],
        [-1.8433, -1.4952, -1.2478, -2.3455, -1.8918, -1.4060],
        [-3.8261, -2.4381, -2.0240, -5.6283, -2.1408, -2.0517],
        [-4.8807, -1.6894, -2.2967, -1.4074, -1.8112, -1.5164],
        [-1.2413, -1.6402, -2.2302, -0.9018, -1.2745, -1.8189],
        [-2.2264, -1.5878, -2.1423, -1.9766, -1.6618, -0.8631],
        [-3.1597, -1.2291, -1.7251, -1.1242, -1.5581, -1.2577],
        [-3.2964, -1.3051, -1.1097, -1.2511, -1.7036, -1.9984]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.6351, -19.4772, -15.6088,  -9.3731,  -9.4370, -13.9831, -15.0678,
         -8.8433, -14.5045, -14.4255,  -7.0998, -13.2587, -10.0014, -10.2295,
        -18.1090, -13.6019,  -9.1068, -10.4579, -10.0539, -10.6643],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-6.8372e-01, -1.7946e-03, -1.6460e+00, -7.8715e-01, -1.8148e+00,
         -9.2607e-01],
        [-3.8085e+00, -4.9424e+00, -4.0029e+00, -1.8039e-01, -5.7706e+00,
         -2.9284e+00],
        [-3.8101e+00, -1.9152e+00, -3.5077e+00, -5.8210e-04, -3.0325e+00,
         -1.2550e-02],
        [-3.0262e+00, -3.5995e-02, -1.1963e-02, -1.7950e+00, -6.2425e-01,
         -1.1619e+00],
        [-2.3138e+00, -3.5307e-02, -2.6031e+00, -2.1540e-02, -2.0940e+00,
         -2.6724e-01],
        [-3.7085e+00, -6.6996e-01, -3.8384e+00, -1.0876e-02, -1.2804e+00,
         -1.6739e+00],
        [-5.0617e+00, -2.3137e-01, -3.5972e+00, -3.2330e-01, -2.2197e+00,
         -5.9905e-01],
        [-1.2854e+00, -5.2803e-02, -7.9894e-01, -3.0546e-01, -2.7620e+00,
         -7.9170e-01],
        [-3.5695e+00, -2.7370e+00, -2.4330e+00, -2.1111e-02, -4.1007e+00,
         -6.2206e-01],
        [-1.3577e+00, -4.1391e+00, -2.3148e+00, -1.3177e+00, -2.2456e+00,
         -1.4552e+00],
        [-1.2029e+00, -3.0951e-01, -4.8196e-04, -1.7000e-01, -1.1484e+00,
         -3.2310e-01],
        [-5.1141e+00, -3.2685e-01, -1.9735e+00, -3.0325e-02, -2.8663e+00,
         -6.9693e-01],
        [-1.9249e+00, -1.4194e-01, -2.3101e+00, -1.2271e+00, -9.0386e-01,
         -2.2700e-03],
        [-1.5138e+00, -4.9513e-01, -6.5339e-01, -2.3085e+00, -2.2679e+00,
         -4.3781e-01],
        [-3.0818e+00, -2.5893e+00, -2.4243e+00, -6.1511e+00, -2.4349e+00,
         -2.1468e+00],
        [-3.4329e+00, -9.0668e-01, -1.8006e+00, -5.1402e-01, -1.3451e+00,
         -6.9107e-01],
        [-3.1689e-02, -1.9003e+00, -2.1789e+00, -2.2775e-01, -3.1696e-01,
         -1.1599e+00],
        [-2.5749e+00, -5.9911e-02, -1.8119e+00, -1.8684e+00, -5.2344e-01,
         -2.4226e-02],
        [-2.6728e+00, -4.8653e-01, -1.3672e+00, -1.8127e-02, -1.5474e+00,
         -5.9474e-02],
        [-2.8071e+00, -3.6882e-01, -3.9481e-02, -3.1198e-01, -1.8312e+00,
         -2.0391e+00]], device='cuda:0')
tensor([ -5.8595, -21.6332, -12.2786,  -6.6554,  -7.3351, -11.1821, -12.0323,
         -5.9964, -13.4834, -12.8302,  -3.1544, -11.0080,  -6.5102,  -7.6766,
        -18.8282,  -8.6904,  -5.8155,  -6.8628,  -6.1516,  -7.3977],
       device='cuda:0')
tensor([-2.7756,  2.1560, -3.3303, -2.7177, -2.1020, -2.8010, -3.0355, -2.8469,
        -1.0211, -1.5953, -3.9454, -2.2507, -3.4912, -2.5530,  0.7192, -4.9114,
        -3.2913, -3.5951, -3.9023, -3.2666], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.7052, -2.1187, -1.3883, -2.0626, -1.3596, -1.3149],
        [-1.3025, -1.7374, -1.3357, -3.4371, -1.4477, -2.9821],
        [-1.8504, -3.0581, -1.4994, -1.8563, -2.3877, -2.5327],
        [-4.3461, -0.9870, -1.8008, -1.6631, -1.8598, -2.5036],
        [-2.5183, -2.3109, -2.4720, -1.3739, -2.5831, -1.2333],
        [-6.1044, -3.8870, -1.6330, -1.1438, -1.8182, -2.1699],
        [-1.2580, -1.6888, -2.1253, -1.0932, -1.3943, -1.2144],
        [-3.6479, -1.4658, -3.1031, -1.4268, -1.6864, -1.0174],
        [-2.9689, -1.4626, -2.5355, -1.2453, -2.0568, -1.3989],
        [-7.0303, -1.2884, -2.9892, -1.0972, -2.0310, -1.7743],
        [-4.4306, -1.3101, -1.2166, -1.1480, -3.9955, -2.0169],
        [-1.8774, -2.8980, -2.8912, -1.1498, -2.1526, -1.1613],
        [-4.2325, -2.7609, -2.1189, -1.8593, -2.2541, -1.5502],
        [-1.2936, -1.2048, -1.1224, -1.1206, -1.6342, -1.2336],
        [-2.1199, -1.2797, -2.9299, -1.2555, -1.2337, -0.7879],
        [-4.4346, -1.4281, -1.3464, -1.3275, -1.3625, -1.0313],
        [-2.1112, -3.1057, -2.8668, -1.7420, -1.4852, -1.2563],
        [-1.3147, -1.6337, -2.0392, -2.0814, -1.3145, -2.2220],
        [-3.0405, -1.7166, -2.9922, -2.0095, -2.2424, -3.0377],
        [-2.4727, -1.1531, -2.8042, -1.6714, -3.6757, -1.8927]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.9493, -12.2426, -13.1846, -13.1603, -12.4915, -16.7564,  -8.7739,
        -12.3476, -11.6681, -16.2105, -14.1177, -12.1302, -14.7759,  -7.6092,
         -9.6068, -10.9303, -12.5672, -10.6055, -15.0388, -13.6698],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.3813e+00, -2.8318e+00, -1.1878e+00, -1.9512e+00, -7.0851e-01,
         -2.8850e-01],
        [-1.5399e+00, -8.2808e-01, -3.5031e-01, -3.2367e+00, -1.2765e+00,
         -2.7891e+00],
        [-2.0700e+00, -2.8897e+00, -8.5751e-01, -1.8179e+00, -2.3784e+00,
         -2.6085e+00],
        [-5.3648e+00, -2.2247e-01, -1.5782e+00, -8.5945e-01, -1.7775e+00,
         -2.8597e+00],
        [-1.3120e+00, -1.4445e+00, -3.4505e+00, -1.8727e-01, -3.9101e+00,
         -1.0296e+00],
        [-4.2059e+00, -3.5663e+00, -8.8435e-01, -1.2004e-01, -1.1658e+00,
         -2.4585e+00],
        [-8.2905e-01, -1.6969e+00, -2.1499e+00, -6.2394e-01, -3.2150e-01,
         -1.0373e-01],
        [-4.3512e+00, -1.0551e-01, -3.8908e+00, -2.9140e-01, -1.3265e+00,
         -5.7815e-02],
        [-2.4754e+00, -1.8941e-01, -3.1783e+00, -4.3145e-01, -2.8961e+00,
         -1.0057e+00],
        [-5.9969e+00, -7.4742e-03, -2.9066e+00, -1.9453e-03, -2.4655e+00,
         -1.4468e+00],
        [-4.1976e+00, -6.8101e-01, -1.5166e-01, -3.1523e-02, -4.9350e+00,
         -1.8898e+00],
        [-7.6788e-01, -2.5499e+00, -3.8249e+00, -2.0318e-02, -2.5246e+00,
         -5.2812e-01],
        [-3.1857e+00, -3.0071e+00, -2.1558e+00, -6.7713e-01, -3.0614e+00,
         -7.3926e-01],
        [-1.2266e+00, -1.6010e-01, -1.6758e-03, -1.0800e-01, -1.0001e+00,
         -4.5257e-03],
        [-2.5622e+00, -7.2466e-02, -3.4969e+00, -1.2856e-01, -2.5581e-01,
         -1.4641e-03],
        [-3.1168e+00, -2.2341e-02, -6.5619e-02, -2.0772e-01, -2.8592e-01,
         -6.1400e-02],
        [-1.0448e+00, -4.8020e+00, -3.3274e+00, -1.2021e+00, -1.0214e+00,
         -6.9987e-01],
        [-1.3428e+00, -1.3132e+00, -2.4745e+00, -1.4015e+00, -7.2465e-01,
         -1.8536e+00],
        [-3.6373e+00, -1.0812e+00, -3.8665e+00, -2.1249e+00, -2.1759e+00,
         -2.9899e+00],
        [-3.1862e+00, -8.2168e-01, -3.2901e+00, -8.6881e-01, -4.2681e+00,
         -2.0081e+00]], device='cuda:0')
tensor([ -9.3491, -10.0206, -12.6221, -12.6621, -11.3339, -12.4008,  -5.7251,
        -10.0233, -10.1764, -12.8252, -11.8866, -10.2157, -12.8265,  -2.5011,
         -6.5174,  -3.7598, -12.0976,  -9.1103, -15.8757, -14.4430],
       device='cuda:0')
tensor([-2.6002, -2.2220, -0.5625, -0.4982, -1.1576, -4.3556, -3.0489, -2.3243,
        -1.4918, -3.3852, -2.2311, -1.9145, -1.9494, -5.1081, -3.0894, -7.1705,
        -0.4696, -1.4952,  0.8368,  0.7732], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.3662, -1.1833, -1.1851, -1.2162, -1.5316, -1.7632],
        [-2.9991, -1.2492, -2.9287, -1.0997, -2.0587, -1.2101],
        [-4.5534, -1.9075, -2.0694, -1.3245, -1.5207, -1.7549],
        [-1.5676, -1.5303, -1.4073, -3.7333, -1.6555, -1.6480],
        [-1.0920, -1.3592, -1.3324, -1.2463, -2.3542, -1.7582],
        [-3.8918, -3.3771, -1.8499, -1.1844, -1.6143, -1.2685],
        [-3.9942, -4.4988, -2.4114, -1.3881, -2.3008, -2.4203],
        [-1.1351, -1.8736, -2.0476, -1.4391, -1.4912, -1.1546],
        [-3.0242, -1.3901, -3.1280, -1.4818, -1.7051, -0.8815],
        [-4.2753, -1.1973, -1.7788, -1.1416, -1.5972, -1.2596],
        [-3.5688, -1.1008, -1.2457, -1.1589, -2.0406, -1.3175],
        [-5.8550, -1.9023, -2.0180, -1.3281, -1.6924, -1.1720],
        [-4.4834, -2.7639, -2.1902, -1.0342, -2.1140, -1.5332],
        [-1.4461, -1.4608, -1.6551, -1.8373, -1.4278, -1.6879],
        [-2.6827, -1.4429, -2.5406, -1.3341, -1.5660, -0.9556],
        [-8.3120, -1.3651, -2.9811, -1.2034, -1.5919, -1.2796],
        [-5.9371, -1.0617, -1.7785, -1.2110, -1.7426, -1.9779],
        [-2.0542, -1.2120, -1.1590, -1.1991, -2.5734, -4.5855],
        [-5.1467, -3.0717, -1.5973, -1.1900, -2.2376, -1.3464],
        [-2.9915, -2.4420, -2.3207, -1.8644, -2.1901, -1.4462]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.2456, -11.5454, -13.1304, -11.5421,  -9.1422, -13.1861, -17.0136,
         -9.1412, -11.6107, -11.2497, -10.4324, -13.9677, -14.1189,  -9.5149,
        -10.5219, -16.7331, -13.7087, -12.7832, -14.5897, -13.2549],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.5685e+00, -2.8577e-01, -1.0992e-02, -7.8829e-01, -9.5815e-01,
         -1.6536e+00],
        [-3.4829e+00, -5.7612e-01, -3.2577e+00, -1.0978e-01, -2.1124e+00,
         -4.7408e-01],
        [-3.7878e+00, -1.1077e+00, -1.9167e+00, -2.4646e-01, -9.9122e-01,
         -1.2112e+00],
        [-1.1751e+00, -5.0201e-01, -2.4741e-01, -3.7911e+00, -1.5494e+00,
         -1.4411e+00],
        [-5.8892e-01, -3.7896e-01, -7.7798e-01, -4.0669e-01, -2.9820e+00,
         -1.6191e+00],
        [-2.6147e+00, -4.6758e+00, -2.0089e+00, -4.5285e-01, -5.3531e-01,
         -1.8824e-02],
        [-3.0203e+00, -4.4210e+00, -2.6644e+00, -5.0732e-01, -2.7129e+00,
         -2.8863e+00],
        [-8.8184e-02, -2.2892e+00, -2.1234e+00, -5.1448e-01, -6.4110e-01,
         -7.3604e-02],
        [-2.9650e+00, -3.8170e-03, -3.4316e+00, -5.6310e-01, -1.6309e+00,
         -1.0110e-02],
        [-3.4096e+00, -3.8556e-01, -1.4065e+00, -1.6400e-01, -1.7662e+00,
         -1.0793e-01],
        [-3.7029e+00, -9.4995e-02, -3.5644e-04, -8.3417e-02, -2.3210e+00,
         -6.2063e-01],
        [-4.2020e+00, -1.2159e+00, -2.6370e+00, -7.8055e-01, -1.2297e+00,
         -9.1204e-02],
        [-3.1607e+00, -3.0244e+00, -2.4178e+00, -9.5413e-03, -2.5844e+00,
         -2.6109e-01],
        [-1.8409e+00, -1.2049e+00, -1.1830e+00, -1.3846e+00, -2.2578e-01,
         -1.3226e+00],
        [-2.9306e+00, -1.9312e-05, -3.4045e+00, -9.6796e-02, -1.1990e+00,
         -1.9133e-01],
        [-5.5491e+00, -1.0029e-01, -3.4475e+00, -1.0204e-01, -1.4468e+00,
         -5.9382e-03],
        [-5.2503e+00, -3.8108e-02, -1.1544e+00, -1.5449e-01, -1.6975e+00,
         -2.2781e+00],
        [-2.6341e+00, -4.8140e-01, -3.7826e-01, -1.0195e-01, -3.1448e+00,
         -3.2192e+00],
        [-3.8697e+00, -3.7093e+00, -4.6901e-01, -2.0746e-01, -2.9396e+00,
         -1.6155e-01],
        [-2.0623e+00, -2.1047e+00, -2.4842e+00, -1.0692e+00, -2.8304e+00,
         -6.6467e-01]], device='cuda:0')
tensor([ -5.2653, -10.0129,  -9.2611,  -8.7061,  -6.7536, -10.3063, -16.2122,
         -5.7300,  -8.6045,  -7.2397,  -6.8233, -10.1564, -11.4580,  -7.1617,
         -7.8223, -10.6517, -10.5729,  -9.9597, -11.3567, -11.2155],
       device='cuda:0')
tensor([-2.9803, -1.5325, -3.8693, -2.8359, -2.3886, -2.8798, -0.8014, -3.4112,
        -3.0062, -4.0100, -3.6091, -3.8113, -2.6610, -2.3532, -2.6996, -6.0813,
        -3.1358, -2.8236, -3.2331, -2.0394], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.0984, -1.4075, -2.6986, -1.5024, -2.3220, -1.7180],
        [-5.3822, -1.3341, -1.7231, -1.7601, -1.6839, -1.4256],
        [-6.3934, -1.4580, -2.2568, -2.6418, -1.3016, -1.1677],
        [-6.0834, -1.8173, -2.0131, -4.2885, -2.1631, -3.0027],
        [-7.8575, -3.1338, -3.1115, -1.1168, -1.5959, -1.4665],
        [-1.1877, -1.1836, -1.8537, -2.7970, -1.2464, -1.1457],
        [-4.6700, -1.4900, -2.1196, -1.3964, -1.2903, -0.9289],
        [-9.0405, -1.5238, -1.6925, -1.1417, -1.4587, -1.7096],
        [-2.4063, -1.9585, -1.8760, -1.7592, -1.1265, -1.7182],
        [-2.8580, -1.4717, -1.9735, -1.2211, -2.3284, -1.4387],
        [-6.1798, -1.3080, -1.5398, -1.0393, -3.0199, -3.1569],
        [-7.3950, -2.4262, -1.6308, -1.6813, -2.0918, -1.5284],
        [-2.8264, -1.9268, -2.2911, -2.0268, -2.0939, -1.4812],
        [-1.8550, -1.1996, -1.2185, -0.9847, -1.8564, -1.6169],
        [-2.2735, -1.3505, -3.7901, -1.3221, -1.5859, -0.8481],
        [-8.2246, -1.3579, -2.8170, -1.2110, -1.9297, -1.7657],
        [-5.3987, -1.3929, -2.0673, -1.2108, -2.2290, -1.3606],
        [-6.6599, -1.2429, -2.2981, -1.2142, -2.1549, -0.9961],
        [-4.4345, -2.7250, -2.3875, -1.2187, -2.3192, -2.2597],
        [-1.9130, -2.9078, -2.0983, -0.9321, -1.7889, -1.3357]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.7470, -13.3090, -15.2194, -19.3681, -18.2820,  -9.4140, -11.8952,
        -16.5668, -10.8446, -11.2914, -16.2437, -16.7535, -12.6463,  -8.7311,
        -11.1701, -17.3058, -13.6593, -14.5661, -15.3446, -10.9757],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.6211e+00, -6.3674e-03, -3.3443e+00, -5.9710e-01, -2.9211e+00,
         -7.4759e-01],
        [-3.8986e+00, -4.2629e-01, -1.1859e+00, -1.5617e+00, -2.0272e+00,
         -1.1088e+00],
        [-4.5500e+00, -2.3517e-01, -2.6195e+00, -2.6520e+00, -4.8016e-01,
         -8.0289e-03],
        [-4.1628e+00, -5.7235e-01, -2.6011e+00, -4.8906e+00, -2.4303e+00,
         -4.4904e+00],
        [-5.0904e+00, -2.8370e+00, -2.6178e+00, -1.1546e-01, -6.0039e-01,
         -5.8779e-01],
        [-1.1984e-01, -4.4951e-01, -1.4769e+00, -2.2240e+00, -3.4006e-01,
         -3.0010e-01],
        [-4.2910e+00, -1.4277e-01, -2.2545e+00, -4.7676e-01, -1.3756e-01,
         -1.8935e-02],
        [-5.0373e+00, -1.7334e+00, -1.4643e+00, -2.8417e-02, -1.3988e-01,
         -9.4871e-01],
        [-3.3228e+00, -2.8173e+00, -1.9746e+00, -1.1251e+00, -2.4437e-03,
         -7.3708e-01],
        [-3.5324e+00, -4.4077e-01, -1.9636e+00, -7.9167e-01, -2.3846e+00,
         -6.2365e-01],
        [-5.9126e+00, -1.2612e+00, -1.1107e+00, -1.7883e-02, -3.9856e+00,
         -4.7129e+00],
        [-5.1648e+00, -2.0863e+00, -1.3820e+00, -1.2857e+00, -2.4846e+00,
         -1.2824e+00],
        [-1.7056e+00, -1.2450e+00, -2.6146e+00, -9.0553e-01, -2.4400e+00,
         -3.6869e-01],
        [-2.7933e+00, -2.7302e-01, -8.7039e-02, -9.0307e-02, -1.8297e+00,
         -1.2391e+00],
        [-2.6588e+00, -1.1419e-01, -4.5062e+00, -2.9826e-01, -1.2205e+00,
         -1.0324e-03],
        [-5.3896e+00, -4.8046e-02, -3.4400e+00, -1.1681e-01, -2.6033e+00,
         -2.0611e+00],
        [-4.6129e+00, -2.9973e-01, -1.6971e+00, -1.3475e-01, -3.0037e+00,
         -4.8214e-01],
        [-5.3725e+00, -1.4787e-01, -2.7611e+00, -2.7455e-01, -2.4999e+00,
         -7.4542e-04],
        [-3.7215e+00, -2.7448e+00, -3.6861e+00, -1.0019e-01, -3.2891e+00,
         -3.0252e+00],
        [-5.0874e-01, -2.6873e+00, -1.6886e+00, -2.2229e-03, -1.4770e+00,
         -2.7280e-01]], device='cuda:0')
tensor([-10.2376, -10.2084, -10.5448, -19.1475, -11.8489,  -4.9104,  -7.3215,
         -9.3521,  -9.9793,  -9.7368, -17.0010, -13.6858,  -9.2794,  -6.3126,
         -8.7990, -13.6589, -10.2303, -11.0567, -16.5669,  -6.6366],
       device='cuda:0')
tensor([-1.5093, -3.1006, -4.6745, -0.2206, -6.4331, -4.5036, -4.5737, -7.2147,
        -0.8653, -1.5546,  0.7572, -3.0677, -3.3669, -2.4185, -2.3711, -3.6469,
        -3.4291, -3.5094,  1.2223, -4.3391], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-8.6308, -2.7845, -1.9715, -2.2420, -2.6187, -1.7347],
        [-2.6259, -4.2039, -2.2830, -2.3891, -1.7903, -1.2787],
        [-1.1336, -1.2906, -2.5868, -3.1745, -1.1225, -1.0155],
        [-2.1066, -1.7107, -1.9099, -1.3891, -1.3229, -1.2351],
        [-7.8450, -1.4990, -1.8861, -1.2639, -1.9185, -1.0683],
        [-4.0394, -2.5055, -1.7033, -3.2980, -1.6979, -1.5476],
        [-2.6862, -3.3886, -2.6749, -7.5786, -2.0234, -1.4813],
        [-5.4057, -1.1766, -2.7515, -2.9145, -1.9800, -1.7273],
        [-3.2237, -1.0236, -2.3730, -1.7096, -4.0684, -2.1597],
        [-6.1296, -4.8684, -2.0215, -1.3164, -2.0028, -1.0464],
        [-2.6500, -3.0878, -2.0012, -2.0114, -1.8666, -1.5142],
        [-1.2050, -1.7735, -3.3013, -1.5365, -1.3790, -1.2598],
        [-1.9503, -1.3609, -2.1653, -1.3667, -1.6462, -0.9268],
        [-2.9470, -1.3643, -3.6787, -1.2425, -1.4077, -1.2132],
        [-4.7415, -1.2653, -2.5393, -1.0845, -1.9506, -1.3879],
        [-1.6127, -1.3464, -1.4007, -1.1421, -1.7385, -2.7577],
        [-6.7562, -1.8812, -1.9927, -1.1606, -1.7969, -1.4274],
        [-4.5789, -3.3764, -2.3594, -1.3756, -3.2137, -1.9635],
        [-1.2415, -1.2151, -1.9016, -1.2697, -1.5548, -1.2671],
        [-4.1257, -1.5117, -2.9032, -1.2462, -1.4470, -1.0477]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-19.9823, -14.5709, -10.3235,  -9.6742, -15.4808, -14.7919, -19.8330,
        -15.9556, -14.5580, -17.3850, -13.1312, -10.4551,  -9.4163, -11.8534,
        -12.9691,  -9.9981, -15.0150, -16.8676,  -8.4498, -12.2815],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-5.3942e+00, -3.6914e+00, -2.5019e+00, -2.6281e+00, -3.6516e+00,
         -1.4849e+00],
        [-1.6412e+00, -4.3130e+00, -2.6487e+00, -1.5453e+00, -1.4241e+00,
         -5.8787e-02],
        [-3.9622e-01, -8.9441e-01, -2.8638e+00, -2.5541e+00, -2.8962e-02,
         -1.1743e-02],
        [-1.9963e+00, -2.0032e+00, -1.6400e+00, -3.9107e-01, -2.7419e-01,
         -3.9285e-01],
        [-4.1683e+00, -5.8930e-01, -2.1413e+00, -1.7836e-01, -1.1371e+00,
         -4.1804e-02],
        [-2.8139e+00, -4.1516e+00, -1.6950e+00, -3.2409e+00, -1.6295e+00,
         -1.4791e+00],
        [-3.4026e+00, -5.0665e+00, -3.1611e+00, -5.0081e+00, -2.2743e+00,
         -9.0577e-01],
        [-4.9549e+00, -3.4290e-02, -3.2937e+00, -3.3376e+00, -1.9293e+00,
         -1.5922e+00],
        [-4.2358e+00, -1.1930e-01, -2.8289e+00, -9.5216e-01, -4.6542e+00,
         -2.2212e+00],
        [-4.5444e+00, -4.2803e+00, -2.1861e+00, -8.7075e-02, -2.2085e+00,
         -3.2982e-01],
        [-1.4385e+00, -3.5570e+00, -2.0054e+00, -9.0222e-01, -1.6237e+00,
         -9.3137e-01],
        [-7.2977e-01, -1.9515e+00, -3.9593e+00, -6.6795e-01, -4.6652e-02,
         -1.4517e-01],
        [-2.1094e+00, -5.0737e-02, -2.2044e+00, -2.3078e-01, -1.5697e+00,
         -3.1776e-02],
        [-2.3236e+00, -1.7560e-01, -4.7135e+00, -1.6795e-01, -6.0583e-01,
         -2.6520e-01],
        [-4.6486e+00, -7.1543e-01, -2.7188e+00, -3.7238e-02, -2.2147e+00,
         -6.7420e-01],
        [-1.5340e+00, -7.7412e-01, -9.9035e-01, -4.1889e-02, -1.6234e+00,
         -2.7078e+00],
        [-4.6762e+00, -6.6247e-01, -2.3882e+00, -3.4362e-02, -1.1975e+00,
         -1.5241e-01],
        [-3.9750e+00, -3.3157e+00, -2.3663e+00, -5.7224e-01, -4.6311e+00,
         -2.0531e+00],
        [-8.0974e-01, -1.1855e-01, -1.7996e+00, -3.7775e-01, -4.7375e-01,
         -4.2262e-02],
        [-4.6252e+00, -2.4793e-01, -3.8606e+00, -8.4281e-05, -6.8468e-01,
         -1.3177e-01]], device='cuda:0')
tensor([-19.3523, -11.6311,  -6.7492,  -6.6976,  -8.2562, -15.0099, -19.8184,
        -15.1419, -15.0116, -13.6362, -10.4583,  -7.5004,  -6.1969,  -8.2516,
        -11.0090,  -7.6715,  -9.1111, -16.9135,  -3.6216,  -9.5503],
       device='cuda:0')
tensor([-0.6301, -2.9398, -3.5743, -2.9767, -7.2247,  0.2180, -0.0146, -0.8137,
         0.4536, -3.7489, -2.6730, -2.9548, -3.2194, -3.6018, -1.9601, -2.3266,
        -5.9039,  0.0459, -4.8282, -2.7312], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[ -4.7632,  -3.0001,  -1.0055,  -1.7305,  -1.5697,  -1.7059],
        [ -1.2801,  -1.6478,  -1.5569,  -1.8748,  -1.4513,  -2.2296],
        [ -2.5608,  -1.8401,  -1.9018,  -3.3853,  -2.7041,  -1.2337],
        [ -1.4901,  -1.0403,  -1.4561,  -1.1829,  -3.1429,  -2.8248],
        [ -6.8540,  -2.3707,  -2.2755,  -1.6673,  -2.4435,  -5.6561],
        [ -4.0427,  -4.0161,  -3.5835,  -1.2402,  -1.9193,  -1.3449],
        [ -1.2085,  -1.1310,  -1.7668,  -1.0391,  -1.2215,  -1.2135],
        [ -2.1594,  -1.2260,  -3.5244,  -1.4612,  -1.8422,  -1.3769],
        [ -4.9622,  -1.3937,  -1.8573,  -1.1995,  -1.8888,  -1.1990],
        [-10.6074,  -1.2183,  -1.6097,  -1.6297,  -2.5145,  -1.6694],
        [ -2.0102,  -1.2117,  -1.3482,  -1.2382,  -1.7880,  -0.9393],
        [ -6.1154,  -2.8826,  -2.0138,  -1.1130,  -2.6660,  -1.8877],
        [ -4.1601,  -1.6181,  -2.9655,  -0.9382,  -2.0349,  -1.4974],
        [ -1.1675,  -1.1846,  -2.5887,  -1.1530,  -1.4299,  -1.1596],
        [ -5.1306,  -1.4877,  -1.9006,  -1.3503,  -2.0445,  -0.9806],
        [ -1.7418,  -1.5651,  -1.7185,  -1.5563,  -1.7501,  -1.2015],
        [ -2.1712,  -1.4568,  -1.3130,  -1.2631,  -1.4042,  -1.1347],
        [ -1.3660,  -1.3377,  -2.3730,  -2.4410,  -1.5652,  -1.0761],
        [ -3.6685,  -2.2581,  -3.0249,  -5.4984,  -2.0231,  -1.9005],
        [ -3.0640,  -2.4661,  -1.6257,  -2.1057,  -3.5123,  -3.0932]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.7749, -10.0404, -13.6259, -11.1371, -21.2672, -16.1467,  -7.5804,
        -11.5902, -12.5006, -19.2489,  -8.5356, -16.6785, -13.2142,  -8.6832,
        -12.8943,  -9.5333,  -8.7430, -10.1590, -18.3735, -15.8670],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.6773e+00, -4.6586e+00, -2.5564e-01, -1.2503e+00, -1.3209e+00,
         -1.5835e+00],
        [-1.4705e+00, -9.7834e-01, -1.2486e+00, -1.0898e+00, -1.3135e+00,
         -1.6633e+00],
        [-3.0379e+00, -1.3305e+00, -1.9325e+00, -3.8170e+00, -2.8543e+00,
         -4.3113e-01],
        [-1.0128e+00, -3.2072e-01, -6.0910e-01, -1.9852e-01, -3.5936e+00,
         -3.1484e+00],
        [-4.6202e+00, -1.7635e+00, -3.1804e+00, -1.1365e+00, -3.4533e+00,
         -5.1267e+00],
        [-2.4875e+00, -3.7747e+00, -3.5634e+00, -1.1345e-01, -1.5183e+00,
         -4.7370e-03],
        [-4.0778e-01, -3.0500e-01, -1.2006e+00, -4.3244e-01, -2.5178e-01,
         -3.1486e-01],
        [-2.3528e+00, -2.9759e-02, -4.6708e+00, -5.9164e-01, -1.7934e+00,
         -7.1349e-01],
        [-3.6995e+00, -1.4973e-03, -1.5480e+00, -3.9436e-02, -2.6402e+00,
         -3.8077e-02],
        [-7.1231e+00, -1.9761e-01, -8.6601e-01, -1.0907e+00, -3.5699e+00,
         -1.5079e+00],
        [-2.4745e+00, -3.0914e-02, -1.0530e+00, -2.6316e-01, -1.6864e+00,
         -5.2338e-03],
        [-4.3907e+00, -3.6797e+00, -2.4642e+00, -2.0405e-01, -4.1604e+00,
         -2.0109e+00],
        [-3.1418e+00, -8.1409e-01, -2.7609e+00, -4.0124e-02, -2.2055e+00,
         -7.8385e-01],
        [-5.1639e-01, -7.6867e-03, -2.8889e+00, -2.9470e-01, -7.0186e-01,
         -4.5076e-02],
        [-4.6974e+00, -9.9067e-01, -1.7401e+00, -1.8506e-01, -2.4145e+00,
         -7.4438e-03],
        [-6.7284e-01, -6.0298e-03, -1.7869e+00, -9.2609e-01, -6.6044e-01,
         -6.6771e-02],
        [-1.1534e+00, -1.2996e+00, -7.4397e-01, -6.2087e-01, -6.5662e-01,
         -4.8625e-01],
        [-1.2538e+00, -5.2031e-01, -2.5090e+00, -1.8243e+00, -1.1721e+00,
         -4.2475e-03],
        [-3.8321e+00, -1.9829e+00, -3.4181e+00, -5.3926e+00, -1.9751e+00,
         -1.8412e+00],
        [-4.0602e+00, -1.6732e+00, -1.1053e+00, -1.4565e+00, -4.0022e+00,
         -3.5472e+00]], device='cuda:0')
tensor([-11.7462,  -7.7641, -13.4033,  -8.8831, -19.2807, -11.4621,  -2.9125,
        -10.1519,  -7.9667, -14.3552,  -5.5132, -16.9100,  -9.7463,  -4.4546,
        -10.0352,  -4.1190,  -4.9608,  -7.2837, -18.4422, -15.8446],
       device='cuda:0')
tensor([-2.0286, -2.2763, -0.2226, -2.2540, -1.9865, -4.6847, -4.6679, -1.4383,
        -4.5339, -4.8937, -3.0224,  0.2315, -3.4679, -4.2286, -2.8590, -5.4143,
        -3.7823, -2.8753,  0.0686, -0.0224], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[ -4.9765,  -1.8476,  -1.1380,  -3.1412,  -1.4467,  -1.1250],
        [ -2.4827,  -1.0058,  -1.7830,  -1.2963,  -1.7541,  -2.4979],
        [ -1.3793,  -1.2714,  -1.1304,  -1.0589,  -1.2307,  -1.2407],
        [ -1.1584,  -1.6088,  -1.3198,  -1.4518,  -1.1476,  -1.3274],
        [ -3.4605,  -1.1049,  -1.2921,  -1.1468,  -1.2207,  -1.6361],
        [ -2.8264,  -1.4242,  -2.4472,  -1.2553,  -2.0554,  -0.9801],
        [ -1.0355,  -1.8500,  -3.6480,  -1.6323,  -1.3080,  -1.1580],
        [ -1.2215,  -2.6559,  -2.8763,  -1.0408,  -1.2933,  -1.3711],
        [ -3.2169,  -2.5765,  -1.5792,  -5.7836,  -1.1188,  -1.4091],
        [-13.1700,  -1.8023,  -2.6254,  -3.5366,  -1.9164,  -1.1384],
        [ -3.6763,  -1.2480,  -2.6893,  -1.3466,  -1.2400,  -1.3342],
        [ -2.5927,  -1.2681,  -1.1354,  -1.1560,  -1.9881,  -1.0915],
        [ -1.2392,  -1.2054,  -1.1266,  -1.1204,  -1.1694,  -1.1032],
        [ -3.4853,  -1.2489,  -1.6264,  -1.0208,  -1.1278,  -1.4185],
        [ -1.0463,  -1.8272,  -4.0214,  -1.7253,  -1.1902,  -1.1016],
        [ -1.2152,  -1.1520,  -1.2098,  -1.2345,  -1.2959,  -1.3963],
        [ -1.2461,  -4.2064,  -4.4011,  -1.7926,  -1.5730,  -1.0388],
        [ -0.9730,  -2.7801,  -1.0972,  -2.1046,  -1.4040,  -1.0695],
        [ -3.1240,  -1.0689,  -2.8236,  -3.2647,  -2.5075,  -1.1545],
        [ -2.4043,  -1.4651,  -1.4317,  -1.3390,  -1.8624,  -2.7165]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.6749, -10.8197,  -7.3114,  -8.0138,  -9.8612, -10.9887, -10.6318,
        -10.4589, -15.6840, -24.1891, -11.5344,  -9.2318,  -6.9641,  -9.9276,
        -10.9120,  -7.5037, -14.2580,  -9.4285, -13.9430, -11.2191],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.9428e+00, -1.0040e+00, -3.7746e-01, -4.0552e+00, -3.8219e-01,
         -1.8863e-01],
        [-1.6178e+00, -6.9748e-03, -1.4616e+00, -6.9396e-01, -1.4583e+00,
         -2.7406e+00],
        [-1.3762e+00, -6.4652e-01, -5.8889e-05, -2.5555e-01, -5.2453e-01,
         -4.3177e-01],
        [-4.0613e-01, -7.5094e-01, -5.8663e-03, -5.3310e-01, -5.2052e-01,
         -5.9523e-01],
        [-2.9670e+00, -2.7393e-01, -6.3793e-01, -1.4681e-01, -1.6620e-01,
         -1.1580e+00],
        [-1.7815e+00, -1.1162e+00, -2.0167e+00, -2.5058e-04, -2.1046e+00,
         -3.7155e-01],
        [-6.0434e-03, -1.9195e+00, -3.4297e+00, -1.1219e+00, -7.0575e-01,
         -3.0639e-01],
        [-7.3483e-01, -2.6937e+00, -3.3500e+00, -3.4254e-01, -4.5757e-01,
         -5.1017e-01],
        [-2.9407e+00, -2.2236e+00, -1.2718e+00, -4.5799e+00, -1.2151e-03,
         -9.5661e-01],
        [-5.2371e+00, -9.1422e-01, -3.5558e+00, -5.1254e+00, -1.6908e+00,
         -7.0190e-04],
        [-2.5951e+00, -7.4150e-01, -2.6788e+00, -7.5774e-01, -4.7340e-02,
         -4.9809e-01],
        [-3.2773e+00, -7.5471e-01, -4.4958e-02, -8.1384e-02, -2.2548e+00,
         -6.2851e-02],
        [-5.7582e-01, -2.9398e-03, -1.3706e-02, -2.0924e-01, -1.8385e-01,
         -5.9012e-03],
        [-2.9850e+00, -8.5115e-01, -1.4447e+00, -1.5825e-02, -3.0950e-01,
         -4.0766e-01],
        [-8.4594e-01, -2.0262e+00, -3.4474e+00, -1.2318e+00, -6.1741e-01,
         -1.7167e-01],
        [-7.3487e-01, -2.2965e-01, -1.2858e-01, -7.8374e-01, -5.3044e-01,
         -6.1346e-01],
        [-6.1495e-01, -3.5797e+00, -4.4775e+00, -1.7733e+00, -1.0810e+00,
         -3.9778e-02],
        [-5.8169e-02, -2.7140e+00, -1.4750e-01, -1.7701e+00, -7.6401e-01,
         -7.2937e-02],
        [-2.0272e+00, -2.1965e-01, -3.5012e+00, -3.9814e+00, -3.1420e+00,
         -1.3313e-01],
        [-7.9377e-01, -4.4107e-05, -5.7468e-01, -6.7551e-02, -1.0581e+00,
         -3.2478e+00]], device='cuda:0')
tensor([ -8.9502,  -7.9792,  -3.2346,  -2.8118,  -5.3498,  -7.3909,  -7.4892,
         -8.0888, -11.9739, -16.5240,  -7.3185,  -6.4760,  -0.9915,  -6.0139,
         -8.3404,  -3.0208, -11.5661,  -5.5267, -13.0046,  -5.7419],
       device='cuda:0')
tensor([-4.7247, -2.8405, -4.0768, -5.2020, -4.5114, -3.5977, -3.1426, -2.3701,
        -3.7101, -7.6651, -4.2159, -2.7558, -5.9727, -3.9138, -2.5716, -4.4829,
        -2.6919, -3.9018, -0.9385, -5.4771], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.1866, -1.3346, -1.5848, -1.2530, -1.5504, -1.1214],
        [-1.9994, -1.3768, -1.6909, -1.2792, -1.6187, -1.4853],
        [-6.0973, -1.4291, -2.0869, -1.2495, -1.4006, -1.1956],
        [-1.7107, -1.0778, -1.1886, -1.6191, -1.5972, -1.6126],
        [-5.7173, -2.4865, -1.9778, -1.1412, -2.2115, -1.0970],
        [-3.5633, -2.8418, -1.8605, -1.2920, -2.1108, -1.6657],
        [-2.3791, -1.2115, -1.2736, -1.1455, -1.5031, -1.2532],
        [-2.8917, -1.8789, -2.7939, -1.3822, -1.9076, -1.0806],
        [-2.3371, -1.2548, -2.1280, -1.2141, -1.5011, -2.6315],
        [-3.8496, -1.4326, -1.5780, -1.2823, -2.3130, -1.2373],
        [-2.2554, -1.1797, -2.7186, -1.1920, -1.8453, -1.0954],
        [-5.3819, -2.6940, -1.8708, -1.2140, -3.4666, -2.1274],
        [-6.7831, -2.5953, -2.4040, -0.9244, -2.7158, -1.2415],
        [-1.4279, -1.1475, -1.7180, -0.8530, -1.2360, -1.4392],
        [-3.9981, -1.4552, -3.0040, -1.8148, -2.0603, -2.1379],
        [-5.6791, -1.4020, -2.9913, -1.1702, -1.9691, -1.2947],
        [-3.4832, -1.0297, -1.2389, -1.9298, -2.6693, -1.3613],
        [-6.2381, -2.0025, -1.8287, -1.7785, -2.2181, -1.7166],
        [-4.2890, -4.5112, -2.3200, -1.9130, -2.6912, -2.1723],
        [-1.2468, -1.2617, -1.6534, -0.9908, -1.5326, -1.3748]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.0306,  -9.4502, -13.4591,  -8.8061, -14.6313, -13.3341,  -8.7661,
        -11.9348, -11.0666, -11.6929, -10.2863, -16.7548, -16.6642,  -7.8216,
        -14.4703, -14.5064, -11.7121, -15.7824, -17.8967,  -8.0600],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-6.8636e-01, -7.2902e-01, -1.2177e+00, -2.7850e-01, -7.7006e-01,
         -6.9504e-02],
        [-2.0671e+00, -3.4170e-01, -1.0381e+00, -7.1704e-04, -1.3429e+00,
         -7.4404e-01],
        [-4.1342e+00, -3.6879e-02, -2.0414e+00, -2.7367e-01, -1.0315e+00,
         -1.3506e-01],
        [-1.4598e+00, -7.3852e-01, -1.3619e-02, -1.0071e+00, -1.4068e+00,
         -1.5241e+00],
        [-4.2443e+00, -2.3893e+00, -2.3836e+00, -1.7978e-01, -2.6160e+00,
         -4.8301e-02],
        [-2.0688e+00, -3.2133e+00, -1.4800e+00, -1.8923e-01, -2.5615e+00,
         -1.1851e+00],
        [-3.6376e+00, -8.4863e-02, -3.6669e-01, -8.1271e-02, -3.0575e-01,
         -2.5892e-02],
        [-3.4312e+00, -1.1394e+00, -3.6428e+00, -3.1696e-01, -2.0299e+00,
         -2.1824e-01],
        [-2.0017e+00, -4.1412e-02, -2.0793e+00, -1.3648e-01, -1.3206e+00,
         -3.5439e+00],
        [-3.5064e+00, -4.2766e-01, -6.8089e-01, -2.5845e-01, -3.3537e+00,
         -2.6800e-01],
        [-2.3813e+00, -3.6952e-02, -3.3755e+00, -7.7819e-02, -1.8301e+00,
         -4.1770e-02],
        [-4.2194e+00, -2.8762e+00, -1.7346e+00, -4.3222e-01, -6.1340e+00,
         -2.5646e+00],
        [-5.2136e+00, -2.3481e+00, -2.1266e+00, -8.8258e-03, -3.9409e+00,
         -2.6260e-02],
        [-1.6199e+00, -5.1615e-01, -1.2613e+00, -7.0740e-03, -6.0973e-01,
         -1.0788e+00],
        [-4.0368e+00, -1.1193e-02, -4.5439e+00, -1.3700e+00, -2.3918e+00,
         -1.7632e+00],
        [-4.0064e+00, -1.9188e-02, -3.7360e+00, -9.7653e-02, -2.8991e+00,
         -2.9282e-01],
        [-3.4586e+00, -3.5018e-01, -6.4609e-01, -1.6665e+00, -3.4783e+00,
         -3.2005e-01],
        [-4.4023e+00, -9.8341e-01, -1.7597e+00, -1.7769e+00, -2.7686e+00,
         -1.9328e+00],
        [-3.6268e+00, -5.0499e+00, -2.6290e+00, -8.5251e-01, -4.2396e+00,
         -2.7928e+00],
        [-9.9685e-01, -5.8885e-01, -1.1504e+00, -1.0514e-01, -8.9489e-01,
         -4.5799e-01]], device='cuda:0')
tensor([ -3.7512,  -5.5346,  -7.6527,  -6.1500, -11.8612, -10.6980,  -4.5020,
        -10.7786,  -9.1234,  -8.4951,  -7.7433, -17.9610, -13.6644,  -5.0929,
        -14.1169, -11.0511,  -9.9197, -13.6238, -19.1907,  -4.1941],
       device='cuda:0')
tensor([-4.2795, -3.9156, -5.8064, -2.6561, -2.7701, -2.6361, -4.2640, -1.1563,
        -1.9431, -3.1979, -2.5430,  1.2062, -2.9998, -2.7287, -0.3535, -3.4553,
        -1.7924, -2.1586,  1.2939, -3.8659], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.2270, -1.3890, -2.3397, -1.4167, -1.8292, -3.0734],
        [-5.2421, -1.3460, -2.2277, -1.4084, -1.8210, -1.1325],
        [-5.4111, -1.2423, -1.3805, -2.9054, -1.3917, -1.3291],
        [-5.9375, -2.3276, -1.6898, -1.9151, -2.3062, -1.7051],
        [-5.0869, -2.6386, -2.9346, -0.9262, -2.8663, -1.4722],
        [-1.5241, -1.1872, -1.7519, -0.8343, -1.1879, -1.0388],
        [-2.7585, -1.5405, -2.1446, -1.3392, -1.5253, -0.9683],
        [-4.2189, -1.2404, -3.6223, -1.1376, -1.5184, -1.3715],
        [-3.2373, -1.1654, -2.2714, -1.2157, -2.0107, -1.6223],
        [-2.1621, -1.2060, -1.1317, -1.2137, -1.6578, -3.3917],
        [-4.1865, -2.6804, -1.7538, -1.1053, -2.1675, -1.3383],
        [-3.3059, -4.0146, -2.6900, -1.3750, -2.2505, -1.4509],
        [-1.1832, -1.5851, -1.4866, -1.0875, -1.7692, -1.4205],
        [-2.6810, -1.6230, -3.9580, -1.4227, -1.7869, -2.3080],
        [-6.0350, -1.2710, -2.4712, -1.2769, -1.7430, -1.2464],
        [-2.8577, -1.4139, -1.1278, -2.7249, -2.2666, -1.1773],
        [-7.0660, -1.7254, -2.3106, -1.9766, -2.2932, -2.2561],
        [-4.5469, -3.0645, -2.3165, -1.4126, -2.1401, -1.5933],
        [-1.5724, -1.1805, -1.5236, -0.9394, -1.6251, -1.1304],
        [-2.3856, -1.5444, -2.8973, -1.4340, -1.4510, -1.2308]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-14.2751, -13.1778, -13.6601, -15.8813, -15.9248,  -7.5242, -10.2764,
        -13.1092, -11.5228, -10.7630, -13.2319, -15.0869,  -8.5320, -13.7796,
        -14.0435, -11.5681, -17.6278, -15.0739,  -7.9714, -10.9431],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.8829e+00, -5.3501e-04, -2.4501e+00, -2.7737e-01, -1.8209e+00,
         -1.4820e+00],
        [-3.8736e+00, -3.2375e-01, -2.4737e+00, -7.3003e-01, -2.4067e+00,
         -1.1226e-01],
        [-4.3329e+00, -2.0051e-02, -9.1979e-01, -2.9788e+00, -9.9202e-01,
         -2.1167e-01],
        [-4.7280e+00, -2.3546e+00, -9.6564e-01, -2.3206e+00, -2.9656e+00,
         -1.2721e+00],
        [-3.6230e+00, -2.4748e+00, -2.6084e+00, -6.4015e-05, -4.5206e+00,
         -5.3843e-01],
        [-1.9855e+00, -6.1383e-01, -1.2520e+00, -2.4110e-03, -3.8950e-01,
         -3.3355e-02],
        [-3.2109e+00, -1.6971e-01, -2.4013e+00, -8.1102e-02, -9.7560e-01,
         -4.3056e-01],
        [-3.5448e+00, -5.2066e-03, -4.5138e+00, -6.8426e-05, -1.3905e+00,
         -2.4431e-01],
        [-3.1724e+00, -4.8974e-01, -1.8455e+00, -7.0785e-02, -2.4551e+00,
         -1.2186e+00],
        [-2.7606e+00, -2.2704e-02, -4.1171e-01, -1.1423e-01, -1.3243e+00,
         -2.4282e+00],
        [-3.0028e+00, -2.7683e+00, -1.3922e+00, -5.7148e-02, -2.8134e+00,
         -2.1937e-01],
        [-2.6831e+00, -4.2756e+00, -3.2159e+00, -3.5373e-01, -2.9831e+00,
         -7.8573e-01],
        [-6.2914e-01, -1.5371e+00, -9.5371e-01, -7.6824e-02, -1.5035e+00,
         -5.9179e-01],
        [-3.1296e+00, -3.4725e-01, -5.8000e+00, -2.8754e-01, -1.7773e+00,
         -1.5122e+00],
        [-4.1983e+00, -5.0782e-01, -2.7673e+00, -3.5210e-01, -2.2066e+00,
         -9.7613e-02],
        [-2.4297e+00, -7.1858e-02, -1.1668e-01, -2.9899e+00, -3.3983e+00,
         -2.9420e-01],
        [-5.0257e+00, -6.7671e-01, -3.3702e+00, -2.1250e+00, -2.9227e+00,
         -2.7026e+00],
        [-4.0825e+00, -3.4683e+00, -3.0256e+00, -1.9873e-01, -2.6025e+00,
         -1.1265e+00],
        [-2.1733e+00, -3.6204e-01, -8.1001e-01, -6.4342e-03, -1.3749e+00,
         -1.8306e-01],
        [-2.7399e+00, -2.2098e-01, -4.0655e+00, -3.3521e-01, -7.0567e-01,
         -8.6799e-01]], device='cuda:0')
tensor([-10.9139,  -9.9201,  -9.4552, -14.6065, -13.7653,  -4.2766,  -7.2692,
         -9.6986,  -9.2521,  -7.0617, -10.2533, -14.2972,  -5.2921, -12.8539,
        -10.1298,  -9.3006, -16.8231, -14.5041,  -4.9098,  -8.9353],
       device='cuda:0')
tensor([-3.3612, -3.2577, -4.2049, -1.2748, -2.1596, -3.2476, -3.0073, -3.4105,
        -2.2707, -3.7013, -2.9787, -0.7897, -3.2400, -0.9257, -3.9137, -2.2675,
        -0.8048, -0.5698, -3.0616, -2.0078], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.3711, -4.5168, -1.9656, -1.3449, -1.4819, -1.6454],
        [-1.2411, -1.4459, -2.7338, -2.9254, -1.4141, -1.2650],
        [-1.1491, -1.3210, -2.3185, -1.3878, -1.3372, -0.8479],
        [-6.6058, -1.2628, -1.7745, -1.1103, -2.1641, -1.0783],
        [-3.4622, -1.1005, -1.1523, -1.6609, -2.8760, -1.4483],
        [-4.7029, -1.7475, -1.4284, -1.3600, -2.6413, -1.2196],
        [-4.3966, -2.6475, -1.8567, -1.5065, -1.9858, -1.4384],
        [-1.3767, -1.3273, -1.6490, -1.1084, -1.5540, -1.3268],
        [-2.7657, -1.4582, -2.7327, -1.2921, -1.4551, -0.8744],
        [-3.0223, -1.2564, -2.3453, -1.2381, -1.3402, -1.2995],
        [-1.5816, -1.6862, -1.2001, -1.7179, -1.2829, -1.8313],
        [-7.3231, -2.5690, -2.3308, -1.2815, -2.2504, -1.2563],
        [-4.8818, -3.7554, -1.8858, -1.7856, -1.9708, -1.3851],
        [-1.3487, -1.2178, -1.9389, -1.5514, -1.5126, -1.2218],
        [-4.6405, -1.4725, -3.2509, -1.2961, -1.9135, -1.6721],
        [-6.5874, -1.3146, -2.6831, -1.1654, -2.2397, -1.5920],
        [-1.8967, -1.3942, -1.2872, -2.0359, -1.5993, -1.1047],
        [-5.4012, -2.5025, -1.9673, -1.5381, -2.3388, -1.9789],
        [-3.1268, -2.3129, -2.2169, -1.6138, -2.5622, -1.6385],
        [-1.7122, -1.1105, -1.4405, -0.8381, -1.3425, -1.4037]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.3256, -11.0253,  -8.3614, -13.9959, -11.7001, -13.0996, -13.8316,
         -8.3421, -10.5782, -10.5018,  -9.3000, -17.0111, -15.6645,  -8.7911,
        -14.2456, -15.5822,  -9.3180, -15.7268, -13.4712,  -7.8475],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.1038e+00, -4.0000e+00, -1.6660e+00, -6.0097e-01, -9.1074e-01,
         -1.2991e+00],
        [-5.0688e-02, -1.1323e+00, -3.2129e+00, -2.7019e+00, -5.9509e-02,
         -1.4615e-01],
        [-8.7284e-01, -2.8695e-02, -2.4815e+00, -2.9678e-01, -9.8377e-02,
         -3.2254e-03],
        [-5.0950e+00, -6.3031e-02, -1.3206e+00, -1.7187e-02, -2.9942e+00,
         -7.1094e-03],
        [-3.3232e+00, -3.4541e-02, -1.1124e-02, -1.1046e+00, -3.7767e+00,
         -8.8981e-01],
        [-3.4111e+00, -9.1508e-01, -8.7319e-01, -7.3041e-01, -3.4623e+00,
         -4.0874e-01],
        [-3.4822e+00, -2.7296e+00, -1.5518e+00, -3.3678e-01, -1.9991e+00,
         -3.0988e-01],
        [-1.5822e+00, -7.3896e-01, -1.1972e+00, -9.4449e-02, -6.9729e-01,
         -2.1787e-01],
        [-3.2530e+00, -8.7153e-02, -3.5799e+00, -1.9833e-02, -7.2384e-01,
         -1.7397e-01],
        [-2.6463e+00, -6.4155e-03, -2.6796e+00, -1.5141e-01, -4.7233e-01,
         -1.8042e-01],
        [-1.3381e+00, -7.0432e-01, -1.1113e-02, -1.1881e+00, -4.6331e-01,
         -1.5763e+00],
        [-4.9801e+00, -2.5663e+00, -3.2191e+00, -6.3963e-01, -2.7684e+00,
         -2.9187e-01],
        [-3.3097e+00, -4.8806e+00, -1.8692e+00, -7.1505e-01, -1.8768e+00,
         -5.9659e-02],
        [-1.3581e+00, -2.0639e-01, -1.9116e+00, -6.4008e-01, -2.9342e-01,
         -1.1607e-01],
        [-4.7775e+00, -2.1411e-01, -4.4492e+00, -8.0592e-02, -1.9570e+00,
         -1.0788e+00],
        [-4.4088e+00, -4.1729e-02, -3.0000e+00, -2.3525e-03, -3.3353e+00,
         -1.4250e+00],
        [-1.7672e+00, -2.6900e-03, -6.0451e-01, -1.7684e+00, -1.5068e+00,
         -7.6332e-02],
        [-4.5552e+00, -2.5876e+00, -2.3474e+00, -1.4466e+00, -3.0896e+00,
         -2.1621e+00],
        [-2.0658e+00, -2.0114e+00, -2.3726e+00, -5.5820e-01, -3.9079e+00,
         -1.3164e+00],
        [-2.3352e+00, -1.5475e-02, -5.8690e-01, -5.3878e-03, -7.8322e-01,
         -8.8854e-01]], device='cuda:0')
tensor([ -9.5806,  -7.3033,  -3.7815,  -9.4971,  -9.1400,  -9.8007, -10.4094,
         -4.5280,  -7.8377,  -6.1365,  -5.2812, -14.4654, -12.7110,  -4.5256,
        -12.5571, -12.2131,  -5.7259, -16.1885, -12.2323,  -4.6148],
       device='cuda:0')
tensor([-3.7450, -3.7220, -4.5800, -4.4988, -2.5601, -3.2989, -3.4222, -3.8141,
        -2.7406, -4.3654, -4.0188, -2.5457, -2.9536, -4.2655, -1.6884, -3.3692,
        -3.5921,  0.4617, -1.2389, -3.2327], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.6070, -1.4598, -2.4385, -1.5680, -1.9110, -1.5258],
        [-6.3549, -1.3736, -2.3213, -1.2842, -1.9489, -1.1601],
        [-5.8322, -1.2910, -1.2847, -2.3908, -1.2907, -1.3042],
        [-4.0175, -2.4547, -2.0162, -1.5678, -3.3734, -2.5723],
        [-1.8934, -2.4084, -2.7952, -0.9228, -1.9425, -1.9299],
        [-1.4305, -1.2012, -1.2132, -0.9994, -2.6244, -1.0952],
        [-1.8371, -1.4901, -1.7523, -1.5284, -1.2999, -1.4018],
        [-3.7127, -1.5417, -2.6238, -1.2209, -1.4128, -1.2491],
        [-7.2348, -1.1898, -2.2289, -1.3388, -1.5402, -1.7017],
        [-1.4488, -1.4486, -1.1974, -1.3430, -1.9827, -4.9876],
        [-4.9336, -3.1362, -1.7176, -1.0518, -1.7525, -1.3915],
        [-2.6314, -3.7037, -1.6230, -2.3510, -1.8973, -1.3807],
        [-1.1713, -1.2507, -1.2950, -1.0057, -1.6762, -1.4438],
        [-2.0841, -1.3234, -2.2864, -1.2853, -1.7588, -1.4723],
        [-2.5110, -1.4932, -2.4811, -2.2795, -1.6792, -1.2038],
        [-1.7676, -3.4062, -3.0512, -4.2939, -1.5679, -1.5818],
        [-5.8750, -2.5742, -2.3814, -1.9707, -1.9331, -1.6182],
        [-2.8622, -2.2780, -2.0489, -4.2551, -1.8356, -1.2856],
        [-1.2553, -2.7449, -4.6949, -1.0247, -1.3361, -1.0367],
        [-2.2313, -1.5815, -1.9056, -1.6109, -1.5889, -1.3474]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.5101, -14.4430, -13.3935, -16.0019, -11.8922,  -8.5639,  -9.3096,
        -11.7609, -15.2342, -12.4081, -13.9831, -13.5870,  -7.8427, -10.2105,
        -11.6478, -15.6687, -16.3525, -14.5654, -12.0926, -10.2656],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.4330e+00, -1.1544e-02, -3.0504e+00, -7.6535e-01, -2.0163e+00,
         -5.7219e-01],
        [-4.4358e+00, -4.8628e-01, -2.6389e+00, -2.3333e-01, -2.7960e+00,
         -5.2498e-01],
        [-4.3428e+00, -6.5840e-02, -6.8372e-01, -2.2945e+00, -3.4909e-01,
         -2.9384e-01],
        [-3.1113e+00, -2.7318e+00, -2.5998e+00, -1.6546e+00, -4.9259e+00,
         -3.6730e+00],
        [-3.3759e-01, -2.0785e+00, -2.6150e+00, -1.4480e-02, -1.9064e+00,
         -1.8631e+00],
        [-2.0202e+00, -4.7688e-01, -1.0848e-01, -1.3247e-03, -2.9963e+00,
         -4.0359e-02],
        [-2.2505e+00, -3.9138e-02, -1.2073e+00, -6.1014e-01, -7.4359e-02,
         -4.0566e-01],
        [-3.0733e+00, -2.7259e-01, -3.5476e+00, -9.8497e-02, -1.0633e+00,
         -1.4159e-01],
        [-5.6135e+00, -5.4586e-01, -1.7034e+00, -5.2837e-01, -1.2770e+00,
         -1.7037e+00],
        [-1.5749e+00, -3.6095e-01, -6.4278e-01, -4.9569e-01, -2.1649e+00,
         -3.7587e+00],
        [-3.8058e+00, -3.6175e+00, -1.3433e+00, -2.1015e-01, -1.4506e+00,
         -2.6698e-01],
        [-1.7187e+00, -3.9441e+00, -9.6022e-01, -1.5212e+00, -1.8817e+00,
         -6.2965e-01],
        [-6.7610e-01, -3.8161e-01, -3.6989e-01, -1.6005e-03, -1.2854e+00,
         -8.8511e-01],
        [-2.4034e+00, -9.0109e-02, -2.6849e+00, -8.6859e-03, -1.6784e+00,
         -6.8743e-01],
        [-2.1812e+00, -7.8951e-02, -3.0864e+00, -2.1259e+00, -2.2239e+00,
         -6.1849e-03],
        [-1.5397e+00, -2.8064e+00, -3.7882e+00, -4.7968e+00, -1.4149e+00,
         -1.4415e+00],
        [-3.8290e+00, -2.9443e+00, -4.1067e+00, -2.2401e+00, -1.8456e+00,
         -9.0148e-01],
        [-1.8584e+00, -1.8509e+00, -2.0737e+00, -3.5750e+00, -1.5330e+00,
         -2.0727e-02],
        [-3.7701e-03, -3.7755e+00, -5.3817e+00, -1.4090e-01, -3.5069e-01,
         -6.6170e-03],
        [-2.2713e+00, -1.8876e+00, -1.3587e+00, -9.5356e-01, -5.7970e-01,
         -7.2136e-01]], device='cuda:0')
tensor([-10.8487, -11.1152,  -8.0298, -18.6964,  -8.8150,  -5.6436,  -4.5871,
         -8.1969, -11.3718,  -8.9979, -10.6943, -10.6556,  -3.5997,  -7.5529,
         -9.7026, -15.7875, -15.8671, -10.9119,  -9.6592,  -7.7723],
       device='cuda:0')
tensor([-1.6614, -3.3278, -5.3636,  2.6945, -3.0772, -2.9204, -4.7225, -3.5640,
        -3.8624, -3.4102, -3.2888, -2.9314, -4.2430, -2.6575, -1.9452,  0.1189,
        -0.4854, -3.6535, -2.4334, -2.4933], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.8945, -1.8713, -1.6909, -1.3867, -1.3692, -1.7176],
        [-2.1656, -2.0205, -1.4567, -2.7062, -1.4465, -1.9191],
        [-3.1708, -1.8992, -2.6646, -3.3631, -2.2100, -1.8755],
        [-3.7770, -1.2706, -2.2973, -1.8289, -1.4066, -2.0820],
        [-2.6436, -1.2540, -1.3230, -1.0714, -2.3534, -1.6166],
        [-1.3021, -1.1997, -1.4334, -1.2123, -1.7073, -1.4671],
        [-4.1651, -2.1555, -2.4843, -1.1494, -2.0730, -1.6089],
        [-4.3795, -3.0122, -3.2518, -1.2008, -2.0173, -2.2897],
        [-1.3577, -1.6528, -1.7626, -1.1623, -1.5259, -1.2631],
        [-2.2329, -1.7235, -2.6565, -1.3334, -1.4812, -0.8982],
        [-3.9511, -1.2649, -2.4276, -1.1881, -1.5618, -1.3781],
        [-9.6775, -1.1879, -2.1406, -1.3458, -2.1077, -1.7614],
        [-2.2058, -1.1808, -1.2175, -1.2381, -2.1537, -3.0201],
        [-4.7778, -3.2770, -1.8640, -1.1240, -2.6249, -1.3598],
        [-2.8140, -3.4466, -2.3315, -2.0568, -2.6538, -1.2739],
        [-1.5423, -1.3307, -1.1693, -1.0466, -2.0301, -1.3198],
        [-2.7830, -1.3944, -3.0844, -1.3894, -1.9510, -1.2916],
        [-7.1904, -1.3238, -2.1508, -1.1317, -2.0187, -1.2837],
        [-2.4951, -1.1203, -1.1073, -1.5088, -2.2844, -1.8647],
        [-5.6685, -2.7896, -2.2833, -1.7068, -3.3535, -1.5490]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.9301, -11.7146, -15.1832, -12.6625, -10.2619,  -8.3220, -13.6363,
        -16.1512,  -8.7244, -10.3257, -11.7714, -18.2209, -11.0161, -15.0274,
        -14.5766,  -8.4387, -11.8939, -15.0991, -10.3806, -17.3507],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.1112e+00, -1.0110e+00, -1.3340e+00, -4.1567e-01, -6.0493e-02,
         -1.0470e+00],
        [-1.1033e+00, -2.7569e+00, -1.1401e+00, -2.3478e+00, -3.3464e-01,
         -1.7366e+00],
        [-4.3507e+00, -2.6549e+00, -3.2726e+00, -3.5867e+00, -2.6208e+00,
         -9.7447e-01],
        [-3.3798e+00, -7.0259e-02, -2.5895e+00, -1.7716e+00, -9.0216e-01,
         -2.0642e+00],
        [-3.0722e+00, -1.2002e+00, -2.1688e-02, -4.3772e-02, -2.7597e+00,
         -1.6596e+00],
        [-1.2649e+00, -1.0302e-01, -7.5835e-01, -2.9192e-02, -1.3471e+00,
         -7.7089e-01],
        [-2.9675e+00, -1.9460e+00, -3.7319e+00, -5.1940e-01, -2.2088e+00,
         -1.0766e+00],
        [-4.0184e+00, -3.1563e+00, -3.8870e+00, -2.0979e-01, -2.4189e+00,
         -2.7828e+00],
        [-1.4683e+00, -1.8159e+00, -1.5292e+00, -1.5216e-01, -7.0705e-01,
         -1.4094e-01],
        [-2.5580e+00, -5.5697e-01, -3.7728e+00, -2.0696e-02, -8.3367e-01,
         -4.0640e-02],
        [-3.6219e+00, -6.1075e-02, -2.7259e+00, -3.1709e-01, -1.4994e+00,
         -1.1005e-01],
        [-6.8727e+00, -1.6375e-01, -1.5492e+00, -4.0783e-01, -2.7617e+00,
         -1.5142e+00],
        [-2.7754e+00, -1.4395e-01, -6.1847e-01, -1.2650e-03, -2.5873e+00,
         -2.2553e+00],
        [-3.4017e+00, -4.3026e+00, -1.7758e+00, -9.7847e-02, -4.0330e+00,
         -6.6426e-02],
        [-1.9772e+00, -3.3239e+00, -2.4498e+00, -1.3166e+00, -3.7083e+00,
         -1.6840e-01],
        [-1.7550e+00, -6.8541e-01, -2.7301e-01, -1.2656e-02, -2.1756e+00,
         -2.9385e-01],
        [-3.2361e+00, -2.7207e-02, -4.3353e+00, -3.0537e-01, -2.0970e+00,
         -4.9546e-01],
        [-4.9639e+00, -1.1542e-02, -2.2752e+00, -8.6913e-03, -2.8673e+00,
         -5.1816e-01],
        [-2.5186e+00, -4.0035e-01, -2.1905e-02, -7.4215e-01, -2.9865e+00,
         -1.8560e+00],
        [-3.9071e+00, -3.1863e+00, -3.2906e+00, -1.4980e+00, -4.9219e+00,
         -1.4233e+00]], device='cuda:0')
tensor([ -5.9794,  -9.4193, -17.4602, -10.7775,  -8.7572,  -4.2735, -12.4502,
        -16.4731,  -5.8136,  -7.7828,  -8.3354, -13.2694,  -8.3817, -13.6774,
        -12.9443,  -5.1955, -10.4965, -10.6448,  -8.5255, -18.2272],
       device='cuda:0')
tensor([-4.9507, -2.2953,  2.2770, -1.8850, -1.5048, -4.0486, -1.1861,  0.3219,
        -2.9109, -2.5430, -3.4360, -4.9515, -2.6344, -1.3500, -1.6323, -3.2432,
        -1.3974, -4.4543, -1.8551,  0.8765], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.6156, -1.2905, -2.0427, -1.4644, -1.3819, -1.6501],
        [-4.1549, -1.4750, -2.2321, -1.1785, -1.3684, -1.2011],
        [-1.2332, -1.7369, -1.4100, -1.6115, -1.3288, -0.9178],
        [-5.2053, -1.6536, -2.9902, -4.1357, -2.2344, -1.5762],
        [-4.4040, -1.1566, -1.6723, -1.2072, -2.5520, -1.2782],
        [-5.0526, -3.5864, -1.6023, -1.3899, -2.5098, -2.0013],
        [-2.6962, -2.7617, -1.7239, -2.8355, -2.1094, -1.4881],
        [-1.3017, -1.0877, -1.1538, -1.8042, -1.4323, -1.0803],
        [-2.6417, -1.2467, -4.3299, -1.2547, -1.7782, -0.9180],
        [-2.8605, -1.7459, -2.3455, -1.2706, -1.7704, -1.0734],
        [-4.3680, -1.3063, -1.6023, -1.2215, -2.3186, -2.7136],
        [-1.9466, -1.3046, -1.7550, -1.2731, -2.3144, -1.4717],
        [-3.8094, -3.1598, -2.0946, -1.2018, -1.8722, -2.3344],
        [-3.3919, -2.6723, -2.3017, -2.0842, -2.6592, -2.0407],
        [-1.2221, -1.1526, -1.5728, -0.9296, -1.1592, -1.1992],
        [-2.6705, -1.3073, -2.5784, -1.8201, -2.4659, -2.0691],
        [-2.2328, -1.4397, -1.7747, -2.6450, -1.4021, -1.2180],
        [-1.8078, -1.7311, -1.8220, -2.5628, -1.8335, -1.1695],
        [-6.7964, -2.0949, -2.0471, -3.4624, -2.4340, -1.6487],
        [-2.5432, -2.4178, -2.6728, -1.9102, -1.5198, -1.6827]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.4451, -11.6100,  -8.2384, -17.7954, -12.2703, -16.1424, -13.6148,
         -7.8599, -12.1692, -11.0663, -13.5303, -10.0655, -14.4722, -15.1500,
         -7.2355, -12.9112, -10.7123, -10.9266, -18.4835, -12.7465],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.6295e+00, -1.3184e-01, -2.3382e+00, -5.2649e-01, -3.5680e-01,
         -1.4152e+00],
        [-3.0041e+00, -1.7713e+00, -2.5691e+00, -9.8848e-02, -9.6616e-01,
         -3.5577e-03],
        [-1.4599e+00, -2.2929e+00, -1.0639e+00, -9.1583e-01, -9.7317e-02,
         -7.5018e-04],
        [-5.9036e+00, -6.9532e-01, -3.7426e+00, -3.7433e+00, -2.2808e+00,
         -7.9890e-01],
        [-4.7827e+00, -9.4674e-01, -1.3492e+00, -2.5459e-01, -3.1979e+00,
         -7.3308e-01],
        [-3.4490e+00, -2.9690e+00, -1.0585e+00, -1.8011e-01, -3.7734e+00,
         -1.9745e+00],
        [-1.4784e+00, -2.9967e+00, -1.0255e+00, -1.5711e+00, -2.5330e+00,
         -5.4705e-01],
        [-1.1280e+00, -2.4319e-03, -4.3501e-03, -1.2288e+00, -1.1368e+00,
         -6.0163e-02],
        [-3.1244e+00, -5.1193e-03, -4.7403e+00, -8.8992e-02, -1.7473e+00,
         -1.2805e-01],
        [-2.4610e+00, -7.3407e-01, -2.7820e+00, -2.7455e-01, -2.2212e+00,
         -5.9232e-02],
        [-3.8573e+00, -1.9147e-01, -7.2995e-01, -4.0607e-01, -3.2036e+00,
         -3.1230e+00],
        [-2.2592e+00, -2.4657e-03, -2.0780e+00, -1.8923e-01, -3.1104e+00,
         -7.0197e-01],
        [-2.8131e+00, -3.4203e+00, -2.6833e+00, -4.1363e-01, -1.5591e+00,
         -2.9668e+00],
        [-2.9670e+00, -2.6405e+00, -2.5692e+00, -1.0728e+00, -4.0287e+00,
         -2.4421e+00],
        [-9.1244e-01, -3.6112e-01, -8.4645e-01, -1.7521e-01, -3.8500e-01,
         -5.5629e-01],
        [-3.2299e+00, -1.6975e-04, -3.0941e+00, -1.3576e+00, -3.4452e+00,
         -1.4706e+00],
        [-2.0463e+00, -2.8173e-03, -1.2465e+00, -2.8202e+00, -1.1757e+00,
         -5.0762e-01],
        [-1.5612e+00, -8.6354e-01, -1.8680e+00, -2.6983e+00, -2.0940e+00,
         -1.6569e-02],
        [-4.9340e+00, -1.6129e+00, -2.7677e+00, -4.1794e+00, -3.2723e+00,
         -1.0411e+00],
        [-1.1543e+00, -1.9263e+00, -2.3723e+00, -1.3022e+00, -3.0869e-01,
         -1.1895e+00]], device='cuda:0')
tensor([ -6.3980,  -8.4130,  -5.8306, -17.1645, -11.2642, -13.4045, -10.1517,
         -3.5605,  -9.8343,  -8.5321, -11.5114,  -8.3413, -13.8563, -15.7203,
         -3.2365, -12.5974,  -7.7992,  -9.1016, -17.8075,  -8.2535],
       device='cuda:0')
tensor([-3.0471, -3.1970, -2.4078, -0.6309, -1.0061, -2.7379, -3.4631, -4.2994,
        -2.3349, -2.5342, -2.0189, -1.7242, -0.6159,  0.5703, -3.9989, -0.3138,
        -2.9131, -1.8250, -0.6760, -4.4930], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.5461, -2.2418, -1.9432, -1.1257, -1.8590, -1.5116],
        [-3.0995, -2.7057, -2.4885, -1.3147, -2.6878, -1.5079],
        [-1.3268, -1.4147, -1.4766, -1.4996, -1.4478, -1.2285],
        [-6.0620, -1.3024, -3.4874, -1.2816, -1.7445, -1.4850],
        [-2.4577, -1.4977, -1.8451, -1.6243, -1.9650, -1.4346],
        [-1.4090, -2.0154, -2.1382, -2.8676, -2.7701, -1.2648],
        [-6.7760, -2.4093, -2.0206, -3.6615, -2.2511, -2.0353],
        [-3.2457, -2.0364, -2.2980, -2.3419, -3.5049, -1.8350],
        [-1.1524, -1.8989, -1.6583, -1.1666, -1.2554, -1.1055],
        [-2.1722, -1.3652, -1.7144, -1.3141, -1.5199, -1.0932],
        [-2.6505, -1.4695, -2.7176, -1.2980, -1.2717, -1.2408],
        [-3.4519, -1.1831, -4.7064, -2.4717, -2.6568, -1.4943],
        [-6.1229, -2.7912, -1.7010, -1.1218, -1.6859, -1.2012],
        [-3.2384, -2.2183, -1.7167, -1.8694, -2.3686, -1.2604],
        [-1.2038, -1.3474, -1.4813, -1.6174, -1.4759, -1.1526],
        [-3.0458, -1.4903, -1.9433, -1.3435, -1.2870, -1.0000],
        [-4.0790, -1.1541, -3.3757, -1.2058, -1.2516, -1.2176],
        [-6.2217, -1.2805, -3.9256, -1.1989, -1.3435, -1.8954],
        [-2.3258, -1.1309, -2.1156, -1.1439, -2.1404, -1.8016],
        [-6.9962, -1.9335, -1.9593, -1.1208, -2.1755, -1.6674]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.2272, -13.8042,  -8.3939, -15.3630, -10.8244, -12.4650, -19.1539,
        -15.2619,  -8.2370,  -9.1790, -10.6480, -15.9641, -14.6240, -12.6719,
         -8.2783, -10.1098, -12.2837, -15.8655, -10.6582, -15.8528],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.4668e+00, -1.8746e+00, -2.2500e+00, -1.8207e-01, -1.7057e+00,
         -4.3342e-01],
        [-1.9833e+00, -2.2384e+00, -2.6457e+00, -5.1289e-01, -3.2594e+00,
         -9.9991e-01],
        [-1.1146e+00, -1.0258e+00, -8.7775e-01, -5.7382e-01, -3.6890e-02,
         -2.3632e-02],
        [-5.6178e+00, -1.0980e-01, -4.6119e+00, -2.6050e-02, -1.6638e+00,
         -7.0069e-01],
        [-2.1267e+00, -2.0804e-02, -1.3349e+00, -1.1349e+00, -2.9404e+00,
         -6.6066e-01],
        [-1.0041e+00, -1.2686e+00, -2.3425e+00, -3.1201e+00, -3.8534e+00,
         -4.1835e-01],
        [-4.4624e+00, -2.4368e+00, -2.6658e+00, -4.1279e+00, -2.8791e+00,
         -2.3646e+00],
        [-2.1390e+00, -1.3980e+00, -2.5868e+00, -1.5684e+00, -4.8349e+00,
         -1.9811e+00],
        [-1.1911e-01, -2.4901e+00, -1.1392e+00, -2.0516e-01, -3.8929e-02,
         -4.8761e-02],
        [-2.3093e+00, -3.2650e-01, -9.8202e-01, -5.5854e-02, -2.2465e-01,
         -5.6652e-01],
        [-2.0408e+00, -3.2938e-02, -3.5666e+00, -2.9469e-01, -2.0856e-01,
         -6.7986e-02],
        [-3.9412e+00, -1.5455e-01, -5.4719e+00, -2.1691e+00, -3.0859e+00,
         -1.2351e+00],
        [-3.8441e+00, -2.8809e+00, -1.2088e+00, -3.6256e-02, -5.8255e-01,
         -2.3992e-02],
        [-1.9553e+00, -1.7575e+00, -1.1130e+00, -1.0207e+00, -2.8391e+00,
         -6.8606e-03],
        [-1.3137e-01, -9.4600e-01, -7.3770e-01, -9.8083e-01, -6.7786e-01,
         -2.1254e-01],
        [-3.4036e+00, -9.5042e-02, -1.8451e+00, -5.9367e-02, -4.9271e-02,
         -5.7388e-02],
        [-3.5070e+00, -2.3814e-01, -4.7431e+00, -4.7062e-01, -7.5665e-01,
         -3.2882e-01],
        [-4.8907e+00, -6.9801e-01, -3.7942e+00, -4.5674e-03, -4.8244e-01,
         -1.7006e+00],
        [-2.4780e+00, -9.4749e-03, -2.2992e+00, -1.2868e-02, -2.4105e+00,
         -1.5597e+00],
        [-5.1360e+00, -1.0013e+00, -2.2440e+00, -6.6231e-02, -2.8642e+00,
         -1.0484e+00]], device='cuda:0')
tensor([ -7.9126, -11.6397,  -3.6525, -12.7301,  -8.2183, -12.0071, -18.9366,
        -14.5081,  -4.0413,  -4.4649,  -6.2115, -16.0577,  -8.5766,  -8.6925,
         -3.6863,  -5.5097, -10.0443, -11.5705,  -8.7698, -12.3602],
       device='cuda:0')
tensor([-3.3146, -2.1645, -4.7414, -2.6330, -2.6060, -0.4579, -0.2173, -0.7538,
        -4.1958, -4.7141, -4.4365,  0.0936, -6.0474, -3.9794, -4.5920, -4.6001,
        -2.2394, -4.2950, -1.8884, -3.4927], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.2400, -1.1531, -1.2732, -0.9207, -1.4284, -1.1400],
        [-3.1746, -1.3800, -2.3111, -1.3161, -1.8761, -1.8332],
        [-4.9485, -1.4876, -3.1201, -1.5150, -1.6859, -1.1361],
        [-3.6746, -1.2933, -1.1484, -2.6577, -2.2075, -1.2075],
        [-3.9252, -1.9585, -1.5623, -1.2434, -2.1239, -1.3934],
        [-3.5376, -2.7371, -2.5538, -1.2525, -2.3892, -1.8758],
        [-1.2253, -1.1999, -2.0418, -1.4992, -1.3895, -1.2274],
        [-3.3481, -1.3229, -3.2288, -1.4728, -1.3414, -0.8780],
        [-2.9865, -1.4312, -1.4349, -1.2496, -1.3353, -2.1660],
        [-1.5680, -2.0776, -1.6244, -1.8171, -1.2967, -2.0614],
        [-1.4288, -1.4812, -2.0411, -4.0389, -1.9000, -1.5981],
        [-2.4041, -1.9274, -1.5484, -2.8040, -2.0562, -1.8389],
        [-2.4053, -1.0608, -2.0044, -1.2368, -4.0959, -1.7604],
        [-9.1217, -2.6172, -1.7508, -1.6123, -1.9535, -3.8181],
        [-3.7680, -2.2441, -1.8251, -2.1956, -2.3144, -2.3514],
        [-1.4731, -1.1534, -1.2853, -1.5743, -1.7090, -1.3976],
        [-3.8065, -1.2421, -2.2667, -1.4457, -2.2685, -1.7210],
        [-1.4503, -1.5432, -2.0632, -1.2771, -1.9731, -1.5883],
        [-4.2780, -1.3496, -1.7239, -3.1674, -1.8307, -1.2704],
        [-3.8445, -1.3794, -2.4192, -1.7285, -2.4040, -2.5111]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -7.1554, -11.8911, -13.8933, -12.1890, -12.2068, -14.3459,  -8.5831,
        -11.5920, -10.6036, -10.4452, -12.4880, -12.5791, -12.5636, -20.8736,
        -14.6986,  -8.5927, -12.7505,  -9.8951, -13.6200, -14.2866],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-8.9438e-01, -2.9257e-02, -2.3842e-01, -6.7359e-02, -6.9543e-01,
         -9.9733e-02],
        [-3.7934e+00, -1.8079e-02, -2.6517e+00, -1.1954e-01, -1.9819e+00,
         -1.0793e+00],
        [-3.7217e+00, -2.3641e-02, -4.2369e+00, -9.1263e-01, -2.1018e+00,
         -1.9819e-01],
        [-3.0559e+00, -5.0200e-02, -3.0989e-01, -2.8982e+00, -3.0525e+00,
         -9.1157e-02],
        [-2.7657e+00, -1.2049e+00, -4.7740e-01, -6.2067e-01, -2.5373e+00,
         -3.4451e-01],
        [-2.6968e+00, -2.7795e+00, -3.3546e+00, -1.6067e-01, -3.1943e+00,
         -2.1155e+00],
        [-2.1458e-01, -2.7871e-01, -1.9738e+00, -6.9461e-01, -1.5283e-01,
         -7.8201e-04],
        [-3.5493e+00, -1.4836e-01, -4.1688e+00, -5.3649e-01, -2.8366e-01,
         -2.2732e-03],
        [-1.9020e+00, -4.0385e-03, -5.6356e-01, -1.0900e-01, -1.8206e-01,
         -1.7106e+00],
        [-6.6768e-01, -2.6911e+00, -1.6476e+00, -1.1761e+00, -4.2011e-01,
         -1.9274e+00],
        [-1.5787e+00, -7.4204e-01, -2.4845e+00, -3.7338e+00, -2.1233e+00,
         -8.5858e-01],
        [-2.9022e+00, -1.4820e+00, -1.0624e+00, -3.0862e+00, -1.8671e+00,
         -1.4878e+00],
        [-3.1009e+00, -6.2546e-01, -2.1142e+00, -2.4314e-01, -4.8749e+00,
         -1.7835e+00],
        [-5.8908e+00, -2.0441e+00, -1.6087e+00, -9.8588e-01, -2.1251e+00,
         -4.0000e+00],
        [-2.9141e+00, -2.0010e+00, -1.4473e+00, -1.0105e+00, -3.3279e+00,
         -3.1957e+00],
        [-1.9318e+00, -7.3321e-02, -2.1204e-02, -1.5924e+00, -1.4898e+00,
         -7.9466e-01],
        [-4.3388e+00, -1.0629e-02, -2.3585e+00, -5.8303e-01, -2.6899e+00,
         -1.0859e+00],
        [-1.2909e+00, -2.0963e-02, -2.0389e+00, -1.4789e-01, -2.8339e+00,
         -1.2297e+00],
        [-3.6161e+00, -8.9836e-03, -9.6182e-01, -3.5086e+00, -2.1746e+00,
         -2.7405e-01],
        [-3.1817e+00, -3.0390e-01, -2.7601e+00, -1.5340e+00, -2.7783e+00,
         -2.4999e+00]], device='cuda:0')
tensor([ -2.0246,  -9.6440, -11.1948,  -9.4579,  -7.9505, -14.3013,  -3.3153,
         -8.6889,  -4.4712,  -8.5299, -11.5209, -11.8877, -12.7421, -16.6546,
        -13.8966,  -5.9033, -11.0667,  -7.5623, -10.5441, -13.0580],
       device='cuda:0')
tensor([-5.1309, -2.2471, -2.6984, -2.7311, -4.2562, -0.0446, -5.2677, -2.9031,
        -6.1323, -1.9153, -0.9671, -0.6913,  0.1784, -4.2190, -0.8020, -2.6895,
        -1.6837, -2.3328, -3.0758, -1.2286], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.2721, -1.6438, -2.1129, -2.1851, -1.4278, -1.2986],
        [-1.8777, -1.2605, -2.3508, -1.2435, -1.8371, -1.4551],
        [-4.0958, -1.5189, -1.7876, -1.1697, -1.3421, -1.5382],
        [-7.6522, -1.2712, -1.6545, -1.1581, -1.9894, -1.3552],
        [-4.3884, -2.4282, -2.1318, -1.2356, -2.2251, -1.6507],
        [-2.2867, -3.1577, -2.8759, -1.0212, -3.6040, -1.5685],
        [-2.4569, -1.2018, -1.2576, -1.2160, -1.7142, -1.3085],
        [-2.9947, -1.3379, -2.6268, -1.2957, -1.2723, -1.0068],
        [-2.7664, -1.8055, -2.5545, -1.3168, -1.3321, -1.0585],
        [-2.9234, -1.0174, -4.4920, -1.1872, -1.7694, -1.6341],
        [-2.4153, -1.3716, -1.1001, -1.1843, -1.7784, -3.8784],
        [-4.9239, -2.9408, -2.3802, -1.0989, -1.9958, -1.4227],
        [-2.3646, -2.6685, -1.8644, -1.7486, -2.0012, -1.5615],
        [-1.2269, -1.4541, -1.6002, -1.9364, -1.4300, -1.6466],
        [-2.2652, -1.2578, -3.0736, -1.3147, -1.4870, -2.4042],
        [-3.8877, -1.5755, -2.1471, -1.2511, -1.6613, -1.3526],
        [-3.2984, -1.1728, -2.0991, -1.9953, -1.5716, -1.4615],
        [-4.9311, -2.5877, -1.8801, -2.4408, -2.9015, -1.8227],
        [-3.3922, -3.7105, -2.3650, -0.9994, -1.9492, -1.3327],
        [-1.3175, -1.0861, -1.5815, -1.1034, -1.3524, -0.9642]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.9403, -10.0247, -11.4522, -15.0806, -14.0598, -14.5140,  -9.1549,
        -10.5342, -10.8339, -13.0235, -11.7282, -14.7623, -12.2087,  -9.2942,
        -11.8025, -11.8752, -11.5987, -16.5639, -13.7490,  -7.4052],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.2827e-01, -1.8458e+00, -2.2255e+00, -1.5743e+00, -1.9935e-01,
         -3.1093e-01],
        [-1.9541e+00, -6.5249e-01, -2.4098e+00, -1.6385e-03, -2.0909e+00,
         -8.5571e-01],
        [-3.4459e+00, -1.3535e+00, -1.1210e+00, -1.3842e-01, -9.7663e-01,
         -1.2609e+00],
        [-6.0234e+00, -9.3810e-01, -1.4620e+00, -1.2385e-01, -2.3590e+00,
         -8.3466e-01],
        [-2.9312e+00, -2.5706e+00, -3.1390e+00, -4.9550e-01, -2.8534e+00,
         -1.4052e+00],
        [-1.0428e+00, -3.3714e+00, -3.5146e+00, -3.7755e-03, -6.1177e+00,
         -1.0444e+00],
        [-3.5730e+00, -2.9108e-02, -4.0726e-01, -6.5948e-02, -1.1964e+00,
         -1.7725e-01],
        [-3.4004e+00, -1.6511e-01, -3.4619e+00, -1.3340e-01, -9.6054e-02,
         -2.9923e-02],
        [-2.4802e+00, -8.3145e-01, -3.2125e+00, -2.6907e-01, -3.5881e-01,
         -7.4256e-02],
        [-2.9257e+00, -2.6762e-01, -4.5734e+00, -2.0701e-02, -1.8347e+00,
         -1.5528e+00],
        [-2.8500e+00, -2.0766e-01, -1.0707e-02, -2.0535e-01, -1.7876e+00,
         -2.7939e+00],
        [-4.0042e+00, -3.2106e+00, -3.0308e+00, -1.7610e-01, -2.1427e+00,
         -6.6920e-01],
        [-9.5281e-01, -2.7253e+00, -1.7360e+00, -8.0499e-01, -2.1045e+00,
         -1.1704e+00],
        [-1.0608e-03, -1.0751e+00, -1.3075e+00, -1.2394e+00, -9.2100e-03,
         -1.1554e+00],
        [-2.4225e+00, -2.6256e-01, -3.3696e+00, -9.2185e-02, -6.7062e-01,
         -1.7136e+00],
        [-3.3927e+00, -2.2745e-02, -2.2606e+00, -4.1571e-02, -1.6145e+00,
         -7.2125e-01],
        [-3.1548e+00, -2.7782e-02, -2.2208e+00, -1.6009e+00, -1.4462e+00,
         -4.7464e-01],
        [-4.4868e+00, -2.8040e+00, -1.8523e+00, -3.3451e+00, -4.5103e+00,
         -1.5932e+00],
        [-1.7590e+00, -3.4748e+00, -1.7392e+00,  0.0000e+00, -1.8703e+00,
         -7.5817e-04],
        [-1.1017e-02, -6.5200e-02, -7.6799e-01, -5.7597e-01, -1.1544e+00,
         -1.1742e-01]], device='cuda:0')
tensor([ -6.3841,  -7.9646,  -8.2964, -11.7410, -13.3949, -15.0947,  -5.4490,
         -7.2868,  -7.2262, -11.1750,  -7.8551, -13.2336,  -9.4940,  -4.7877,
         -8.5311,  -8.0534,  -8.9252, -18.5917,  -8.8440,  -2.6920],
       device='cuda:0')
tensor([-3.5562, -2.0602, -3.1559, -3.3396, -0.6649,  0.5807, -3.7059, -3.2474,
        -3.6077, -1.8485, -3.8731, -1.5287, -2.7148, -4.5065, -3.2713, -3.8218,
        -2.6735,  2.0278, -4.9050, -4.7132], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.7305, -1.2787, -1.8233, -1.2294, -1.3861, -1.0971],
        [-4.7084, -1.9602, -2.7000, -1.3088, -1.5078, -1.8403],
        [-4.5861, -1.2686, -2.1023, -2.0286, -1.7436, -1.4267],
        [-1.2658, -1.1310, -1.2083, -1.1658, -2.2840, -2.6752],
        [-5.7020, -2.1724, -1.8516, -1.1622, -1.9526, -1.3361],
        [-4.2313, -4.4863, -2.4709, -1.2822, -2.9697, -2.3465],
        [-1.2207, -1.3059, -3.7534, -1.4091, -1.4318, -1.1918],
        [-1.8486, -1.2933, -2.6081, -1.3375, -1.6966, -1.0517],
        [-4.2427, -1.4662, -3.6411, -1.2745, -1.6618, -1.1104],
        [-2.4679, -1.0889, -1.4158, -1.4296, -2.2692, -1.6457],
        [-3.1064, -1.4232, -2.8375, -1.1116, -1.8429, -1.1611],
        [-3.7628, -3.1060, -1.9818, -1.5127, -2.2166, -1.5912],
        [-1.2614, -1.2044, -1.3404, -1.1825, -1.4672, -1.2826],
        [-4.5024, -1.1480, -2.6081, -1.2775, -1.3302, -1.9661],
        [-4.6461, -1.6104, -2.9143, -1.2440, -1.6808, -2.2965],
        [-2.1800, -1.8580, -1.6747, -2.8142, -1.8382, -1.6721],
        [-3.5438, -2.7719, -1.7167, -1.2206, -2.0962, -1.3524],
        [-3.6120, -3.7106, -2.1565, -1.6486, -3.1645, -1.3469],
        [-1.4506, -1.1848, -1.2255, -0.9607, -1.6443, -1.3684],
        [-1.4241, -1.3538, -2.1449, -1.2253, -2.0108, -1.4602]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.5452, -14.0254, -13.1558,  -9.7301, -14.1770, -17.7868, -10.3126,
         -9.8359, -13.3968, -10.3172, -11.4826, -14.1712,  -7.7386, -12.8323,
        -14.3920, -12.0372, -12.7016, -15.6392,  -7.8344,  -9.6191],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.9607e+00, -5.1837e-03, -1.7416e+00, -4.2049e-02, -4.7474e-01,
         -1.5659e-01],
        [-3.7188e+00, -1.1986e+00, -3.1898e+00, -1.8450e-01, -9.4038e-01,
         -1.5936e+00],
        [-4.1312e+00, -4.9555e-02, -1.7529e+00, -1.7382e+00, -1.7964e+00,
         -6.4391e-01],
        [-9.4310e-01, -2.2924e-04, -5.1338e-01, -1.3816e-01, -2.8560e+00,
         -3.0028e+00],
        [-3.8317e+00, -1.6267e+00, -1.6897e+00, -2.6175e-01, -1.8735e+00,
         -7.5712e-02],
        [-3.4154e+00, -4.3925e+00, -2.6947e+00, -4.2444e-01, -3.9227e+00,
         -2.7312e+00],
        [-9.9083e-02, -5.7595e-01, -4.5907e+00, -5.0123e-01, -2.6606e-02,
         -5.2953e-02],
        [-1.9086e+00, -5.2726e-01, -2.9656e+00, -1.6353e-01, -1.7498e+00,
         -2.2127e-01],
        [-3.3656e+00, -8.6090e-02, -5.3462e+00, -2.5973e-01, -2.0571e+00,
         -1.8240e-01],
        [-2.5266e+00, -1.6689e-06, -5.4088e-01, -5.9780e-01, -2.7167e+00,
         -1.5182e+00],
        [-2.2864e+00, -1.3242e-01, -3.4829e+00, -2.8716e-01, -1.6058e+00,
         -8.2082e-02],
        [-2.4545e+00, -4.5563e+00, -1.7218e+00, -3.1265e-01, -2.7365e+00,
         -8.4997e-01],
        [-8.3021e-01, -2.5638e-03, -4.9435e-01, -1.9580e-01, -3.3750e-02,
         -3.9216e-01],
        [-4.4920e+00, -5.6696e-04, -2.7835e+00, -1.7130e-01, -3.4830e-01,
         -1.8065e+00],
        [-4.1313e+00, -1.2030e-02, -3.8963e+00, -7.3402e-02, -1.7338e+00,
         -1.7044e+00],
        [-1.7913e+00, -1.0613e+00, -1.7318e+00, -2.9838e+00, -2.2766e+00,
         -1.5368e+00],
        [-2.5324e+00, -2.9027e+00, -1.3257e+00, -3.9343e-01, -2.5249e+00,
         -2.2128e-03],
        [-2.1825e+00, -3.5547e+00, -2.1759e+00, -6.6812e-01, -4.3616e+00,
         -3.2045e-01],
        [-1.8009e+00, -3.4893e-04, -7.0101e-02, -1.4579e-04, -1.2734e+00,
         -6.0267e-01],
        [-1.3985e+00, -7.7484e-01, -2.2805e+00, -5.1547e-02, -2.2154e+00,
         -7.8939e-01]], device='cuda:0')
tensor([ -6.3809, -10.8258, -10.1122,  -7.4537,  -9.3591, -17.5809,  -5.8465,
         -7.5361, -11.2971,  -7.9001,  -7.8767, -12.6317,  -1.9488,  -9.6022,
        -11.5513, -11.3817,  -9.6814, -13.2632,  -3.7476,  -7.5103],
       device='cuda:0')
tensor([-4.1643, -3.1996, -3.0436, -2.2764, -4.8178, -0.2059, -4.4661, -2.2997,
        -2.0997, -2.4170, -3.6059, -1.5395, -5.7897, -3.2301, -2.8407, -0.6555,
        -3.0202, -2.3759, -4.0868, -2.1088], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[ -3.6165,  -1.3129,  -2.3781,  -1.3435,  -1.9806,  -0.9150],
        [ -6.3055,  -1.3534,  -3.7840,  -1.1943,  -1.3809,  -1.1313],
        [ -3.9470,  -1.4524,  -1.6581,  -1.0469,  -1.4430,  -2.2894],
        [ -3.8451,  -1.2252,  -1.8421,  -1.1144,  -2.7575,  -2.4093],
        [ -4.5212,  -2.2228,  -1.7013,  -1.1441,  -1.8702,  -1.4205],
        [ -4.3289,  -4.4053,  -1.5739,  -1.2393,  -1.9831,  -1.6254],
        [ -1.1769,  -1.1800,  -1.2732,  -1.1810,  -1.5974,  -1.3548],
        [ -4.0436,  -1.3351,  -2.9660,  -1.2949,  -1.9651,  -1.3246],
        [ -5.4705,  -1.4029,  -2.7068,  -1.1866,  -2.1746,  -1.2059],
        [ -1.6150,  -1.4349,  -1.9289,  -2.6025,  -1.4366,  -1.3178],
        [ -5.5743,  -2.0381,  -2.1756,  -1.3464,  -2.1281,  -1.7570],
        [ -3.5422,  -3.7687,  -2.5864,  -2.1522,  -2.0267,  -1.4083],
        [ -1.1993,  -1.2095,  -1.7710,  -0.9036,  -2.0792,  -1.1777],
        [ -2.2113,  -1.2702,  -2.1635,  -1.2726,  -1.2843,  -1.6860],
        [-10.4475,  -1.5326,  -1.4276,  -1.3041,  -1.5284,  -1.4945],
        [ -1.0592,  -1.4733,  -2.3715,  -2.7796,  -1.1671,  -1.2591],
        [ -3.4662,  -1.8150,  -2.6177,  -5.0530,  -1.9569,  -2.4610],
        [ -1.4299,  -1.3489,  -1.7574,  -1.0908,  -2.8864,  -1.6805],
        [ -5.0197,  -2.5711,  -1.6289,  -1.5856,  -2.5410,  -3.5800],
        [ -4.9836,  -4.7039,  -1.7311,  -1.8902,  -2.4113,  -2.2740]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.5465, -15.1494, -11.8368, -13.1937, -12.8801, -15.1560,  -7.7633,
        -12.9293, -14.1473, -10.3358, -15.0195, -15.4844,  -8.3403,  -9.8879,
        -17.7347, -10.1099, -17.3699, -10.1938, -16.9262, -17.9940],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.8962e+00, -1.7327e-02, -2.2586e+00, -1.4924e-01, -2.4541e+00,
         -7.6610e-02],
        [-4.2974e+00, -2.6614e-01, -5.1980e+00, -2.1701e-02, -6.6537e-01,
         -1.6014e-01],
        [-3.7210e+00, -1.1681e+00, -1.0327e+00, -8.1644e-02, -1.0271e+00,
         -2.4980e+00],
        [-4.2835e+00, -1.4100e-01, -2.0325e+00, -9.1261e-02, -3.2654e+00,
         -1.8009e+00],
        [-3.4820e+00, -1.5605e+00, -1.1168e+00, -6.0347e-02, -1.5475e+00,
         -1.4906e-01],
        [-3.4920e+00, -4.2745e+00, -7.6160e-01, -3.5223e-01, -2.2818e+00,
         -1.2269e+00],
        [-1.8090e-01, -4.9645e-02, -4.2878e-01, -2.0368e-01, -8.7597e-01,
         -5.9818e-01],
        [-4.3288e+00, -8.3331e-03, -3.8700e+00, -2.8913e-02, -2.0881e+00,
         -4.2642e-01],
        [-3.7870e+00, -9.3559e-03, -3.2134e+00, -2.3842e-06, -3.3155e+00,
         -1.7410e-02],
        [-1.1666e+00, -2.3976e-01, -2.1945e+00, -2.7130e+00, -8.9437e-01,
         -8.2323e-02],
        [-4.1068e+00, -1.3966e+00, -3.0670e+00, -8.0386e-01, -2.6560e+00,
         -1.6034e+00],
        [-2.0563e+00, -3.6647e+00, -3.0355e+00, -1.2933e+00, -2.1582e+00,
         -4.4886e-01],
        [-4.8014e-01, -1.0822e-01, -1.2520e+00, -9.2161e-02, -2.4033e+00,
         -4.3699e-01],
        [-2.2303e+00, -3.9643e-01, -2.4619e+00, -1.1619e-01, -2.0252e-01,
         -1.3057e+00],
        [-4.7546e+00, -6.8657e-01, -6.2751e-01, -6.2876e-03, -8.2717e-03,
         -1.0472e+00],
        [-3.4329e-01, -7.5101e-01, -2.8339e+00, -2.8180e+00, -7.0649e-02,
         -4.4529e-01],
        [-3.9974e+00, -9.4221e-01, -2.9380e+00, -3.8028e+00, -2.0815e+00,
         -2.0480e+00],
        [-1.0830e+00, -3.5630e-01, -1.5467e+00, -5.1048e-02, -3.8954e+00,
         -1.8937e+00],
        [-3.5988e+00, -1.7361e+00, -1.1530e+00, -1.0274e+00, -3.4111e+00,
         -3.4924e+00],
        [-5.0250e+00, -5.1926e+00, -1.1448e+00, -6.7816e-01, -3.1104e+00,
         -2.9896e+00]], device='cuda:0')
tensor([ -8.8521, -10.6088,  -9.5286, -11.6146,  -7.9163, -12.3891,  -2.3372,
        -10.7506, -10.3427,  -7.2905, -13.6337, -12.6569,  -4.7728,  -6.7130,
         -7.1305,  -7.2621, -15.8099,  -8.8261, -14.4188, -18.1404],
       device='cuda:0')
tensor([ -2.6944,  -4.5406,  -2.3082,  -1.5791,  -4.9638,  -2.7668,  -5.4261,
         -2.1787,  -3.8046,  -3.0453,  -1.3858,  -2.8275,  -3.5675,  -3.1750,
        -10.6043,  -2.8478,  -1.5600,  -1.3677,  -2.5075,   0.1464],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([[-0.9642, -1.7876, -1.8924, -1.2697, -1.3008, -1.2503],
        [-5.6741, -3.8815, -1.7562, -2.5699, -1.6894, -1.0957],
        [-4.7884, -2.2805, -1.5274, -1.2184, -2.2900, -2.3830],
        [-1.5005, -1.1567, -1.3309, -1.0128, -1.2820, -1.5026],
        [-3.4540, -1.2269, -2.2684, -1.2462, -1.7937, -1.1521],
        [-1.6198, -1.3787, -1.3680, -3.6517, -1.2407, -1.4753],
        [-3.9379, -3.7495, -1.5936, -5.3267, -1.5278, -1.9983],
        [-4.9463, -3.6185, -1.7924, -2.0163, -1.6053, -1.4012],
        [-1.3837, -1.0999, -2.6743, -1.0232, -1.3940, -1.2154],
        [-4.1472, -1.6916, -1.0954, -1.1199, -1.1337, -0.9795],
        [-5.3852, -1.9625, -2.8538, -1.6306, -1.3197, -1.8999],
        [-0.9551, -1.3264, -3.8322, -1.8765, -1.2734, -1.1648],
        [-3.0353, -2.0243, -2.6097, -1.3060, -1.2725, -1.2535],
        [-1.3170, -2.9021, -2.0846, -1.9387, -1.6287, -2.2163],
        [-7.2988, -1.5389, -2.6228, -2.8655, -1.9846, -1.4005],
        [-3.0806, -3.4525, -2.5753, -1.2682, -2.3879, -1.1354],
        [-1.1810, -2.6778, -2.4534, -1.5768, -1.2165, -1.1193],
        [-2.8695, -2.3574, -1.2322, -1.1907, -1.2676, -1.2173],
        [-0.8205, -1.4760, -1.2222, -1.2928, -1.5672, -1.5592],
        [-1.1606, -0.9671, -1.0963, -0.9976, -1.0818, -0.9984]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.4651, -16.6669, -14.4877,  -7.7856, -11.1412, -10.7342, -18.1338,
        -15.3800,  -8.7906, -10.1672, -15.0518, -10.4284, -11.5013, -12.0873,
        -17.7111, -13.8999, -10.2249, -10.1348,  -7.9378,  -6.3019],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-5.8681e-02, -1.2785e+00, -1.8209e+00, -4.9359e-01, -5.3050e-01,
         -6.1474e-01],
        [-3.6576e+00, -3.6464e+00, -1.9048e+00, -2.8241e+00, -1.1203e+00,
         -4.2692e-01],
        [-2.8476e+00, -1.4932e+00, -1.0734e+00, -1.9233e-01, -2.4514e+00,
         -2.5980e+00],
        [-1.7190e+00, -1.8217e-01, -5.5092e-01, -2.4855e-01, -1.1200e-01,
         -1.1458e+00],
        [-3.6318e+00, -3.5075e-02, -2.7137e+00, -1.8940e-01, -1.9977e+00,
         -1.6128e-01],
        [-1.2586e+00, -1.9189e-01, -8.4803e-01, -3.8631e+00, -2.6021e-01,
         -1.0191e+00],
        [-2.3301e+00, -4.6755e+00, -1.2420e+00, -5.6962e+00, -6.4372e-01,
         -1.1885e+00],
        [-3.4310e+00, -3.3640e+00, -1.4784e+00, -2.0489e+00, -1.0645e+00,
         -8.1868e-01],
        [-2.5301e-01, -1.8368e-01, -2.4294e+00, -8.2691e-02, -3.4472e-01,
         -1.1847e-01],
        [-4.0278e+00, -1.9131e+00, -1.4162e-02, -2.0218e-01, -4.7851e-01,
         -1.5604e-04],
        [-3.5813e+00, -2.4714e+00, -3.6389e+00, -1.3018e+00, -1.1721e-01,
         -1.6574e+00],
        [-4.8244e-02, -5.5008e-01, -3.9199e+00, -1.6293e+00, -7.7198e-01,
         -4.9661e-01],
        [-2.9951e+00, -1.5734e+00, -2.3359e+00, -7.7128e-01, -4.3346e-01,
         -4.3182e-01],
        [-8.2649e-01, -2.3432e+00, -1.9432e+00, -1.3033e+00, -1.1069e+00,
         -2.6390e+00],
        [-5.0226e+00, -2.9203e-01, -3.3832e+00, -3.4078e+00, -2.0419e+00,
         -1.0604e+00],
        [-1.7779e+00, -3.4605e+00, -3.0539e+00, -1.3527e-02, -2.2956e+00,
         -2.7323e-01],
        [-5.0907e-02, -3.3168e+00, -2.7007e+00, -7.3271e-01, -1.0505e-02,
         -7.6041e-02],
        [-2.6932e+00, -3.7549e+00, -5.2419e-02, -1.7770e-02, -9.2355e-01,
         -3.0512e-01],
        [-2.6104e-01, -3.1402e-01, -1.0632e-01, -2.5735e-01, -5.2493e-01,
         -8.5239e-01],
        [-6.7448e-02, -6.5744e-02, -5.1153e-04, -2.1581e-02, -5.7724e-03,
         -2.9888e-02]], device='cuda:0')
tensor([ -4.7968, -13.5801, -10.6559,  -3.9584,  -8.7290,  -7.4411, -15.7760,
        -12.2054,  -3.4120,  -6.6360, -12.7679,  -7.4161,  -8.5410, -10.1620,
        -15.2079, -10.8747,  -6.8876,  -7.7470,  -2.3161,  -0.1909],
       device='cuda:0')
tensor([-3.6683, -3.0868, -3.8318, -3.8272, -2.4122, -3.2932, -2.3578, -3.1745,
        -5.3786, -3.5313, -2.2839, -3.0123, -2.9604, -1.9253, -2.5032, -3.0252,
        -3.3373, -2.3878, -5.6217, -6.1109], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.4800, -1.2591, -1.6822, -1.4053, -1.5430, -1.3517],
        [-2.8357, -1.3237, -2.5630, -1.3744, -1.5890, -1.2810],
        [-4.3359, -1.4062, -2.1738, -1.4632, -1.4514, -1.1494],
        [-2.2161, -1.1468, -2.2917, -2.6384, -1.4663, -1.1774],
        [-5.6495, -2.2523, -2.4631, -1.8072, -2.1192, -1.8727],
        [-3.6589, -1.6962, -2.1320, -1.2138, -1.5103, -2.0740],
        [-1.8520, -1.2875, -1.8246, -0.9765, -1.7265, -1.1996],
        [-1.0092, -1.5686, -2.4503, -1.4509, -1.4210, -1.3663],
        [-4.6968, -1.2576, -2.6399, -1.1555, -2.4257, -1.3004],
        [-4.2721, -1.9331, -1.4405, -2.1424, -1.8715, -1.3575],
        [-5.9495, -2.6971, -1.9461, -1.7193, -1.8434, -1.9709],
        [-3.7391, -3.2801, -2.4649, -1.7446, -2.2706, -1.5106],
        [-1.1805, -1.4943, -3.1861, -1.0044, -1.4315, -1.2655],
        [-1.4196, -1.5954, -3.8179, -1.3403, -1.5530, -1.7389],
        [-8.3167, -1.2195, -3.3306, -1.1393, -1.2891, -1.3436],
        [-6.1396, -1.3381, -2.8241, -1.1984, -1.6184, -2.4169],
        [-1.8170, -1.2067, -1.5335, -1.1232, -2.6966, -1.5354],
        [-5.6169, -3.0790, -1.8150, -1.0985, -2.3314, -1.4411],
        [-2.9042, -4.5588, -1.7907, -1.1841, -1.6483, -1.3471],
        [-1.3400, -1.1557, -1.3841, -1.0974, -1.4365, -1.5429]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.7212, -10.9667, -11.9799, -10.9367, -16.1640, -12.2852,  -8.8667,
         -9.2664, -13.4760, -13.0171, -16.1262, -15.0099,  -9.5622, -11.4650,
        -16.6388, -15.5354,  -9.9124, -15.3818, -13.4331,  -7.9566],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.8125e+00, -2.3299e-01, -1.4163e+00, -4.1779e-01, -4.2053e-01,
         -4.4787e-01],
        [-3.2103e+00, -9.2076e-03, -3.3069e+00, -2.4744e-01, -1.1641e+00,
         -4.9101e-01],
        [-3.3999e+00, -3.3507e-02, -2.2569e+00, -8.0373e-01, -1.1584e+00,
         -6.2469e-02],
        [-1.9391e+00, -9.8097e-04, -2.7126e+00, -2.7454e+00, -1.1909e+00,
         -1.3341e-01],
        [-3.5210e+00, -2.0262e+00, -4.5312e+00, -1.8445e+00, -2.5119e+00,
         -2.1466e+00],
        [-2.8648e+00, -8.9130e-01, -2.1843e+00, -1.4716e-01, -4.3416e-01,
         -2.4472e+00],
        [-2.7256e+00, -6.8061e-01, -1.6141e+00, -1.3566e-02, -1.5292e+00,
         -2.4021e-01],
        [-5.3782e-01, -7.1715e-02, -3.3034e+00, -2.6780e-01, -3.9911e-01,
         -2.7985e-01],
        [-3.6478e+00, -8.7105e-02, -2.8880e+00, -3.6870e-02, -3.8390e+00,
         -5.6192e-02],
        [-3.2889e+00, -1.2612e+00, -1.2750e+00, -2.2070e+00, -2.3480e+00,
         -6.6266e-01],
        [-4.1659e+00, -2.9195e+00, -2.3646e+00, -1.6405e+00, -1.6670e+00,
         -2.1427e+00],
        [-2.4902e+00, -3.2914e+00, -2.8622e+00, -7.3337e-01, -2.7793e+00,
         -9.0433e-01],
        [-3.7512e-01, -1.4032e+00, -3.7531e+00, -4.0764e-02, -7.2606e-01,
         -5.3587e-01],
        [-1.4096e+00, -3.3197e-01, -5.3908e+00, -9.3151e-02, -1.1868e+00,
         -9.2208e-01],
        [-5.2010e+00, -1.4494e-01, -4.3086e+00, -2.7064e-02, -6.1581e-01,
         -5.5088e-01],
        [-4.8729e+00, -8.1383e-02, -2.8861e+00, -9.1949e-02, -1.5788e+00,
         -2.2826e+00],
        [-1.9055e+00, -3.5763e-06, -1.2838e+00, -6.3609e-02, -3.3798e+00,
         -1.0795e+00],
        [-4.1209e+00, -3.6336e+00, -1.5186e+00, -3.9406e-02, -3.3538e+00,
         -2.9663e-02],
        [-2.0931e+00, -4.6060e+00, -1.3149e+00, -1.9052e-01, -1.2706e+00,
         -5.1385e-01],
        [-1.2807e+00, -1.1854e-01, -7.3663e-01, -1.8595e-01, -4.2600e-01,
         -1.0389e+00]], device='cuda:0')
tensor([ -4.7480,  -8.4288,  -7.7150,  -8.7224, -16.5814,  -8.9689,  -6.8033,
         -4.8597, -10.5549, -11.0428, -14.9001, -13.0608,  -6.8341,  -9.3343,
        -10.8483, -11.7937,  -7.7122, -12.6960,  -9.9890,  -3.7867],
       device='cuda:0')
tensor([-3.9732, -2.5379, -4.2649, -2.2143,  0.4174, -3.3162, -2.0634, -4.4067,
        -2.9211, -1.9743, -1.2261, -1.9490, -2.7281, -2.1307, -5.7905, -3.7417,
        -2.2002, -2.6858, -3.4442, -4.1698], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.1441, -1.2577, -2.0641, -0.9725, -1.2762, -1.3006],
        [-2.7291, -1.3391, -3.1233, -1.8190, -1.6546, -1.0153],
        [-5.0317, -1.2505, -2.8052, -1.1171, -2.5726, -1.2399],
        [-3.7690, -1.2994, -1.2364, -1.2535, -1.6561, -1.4848],
        [-6.1966, -2.3633, -2.1033, -1.2232, -2.5705, -1.3861],
        [-4.9768, -2.2987, -2.2850, -1.0264, -1.8435, -1.4154],
        [-1.4931, -1.1651, -1.6450, -2.8710, -1.4343, -1.2466],
        [-2.8084, -1.3989, -3.2704, -1.5220, -1.2958, -1.1383],
        [-5.2815, -1.2745, -1.4019, -1.2488, -1.3076, -1.6338],
        [-3.1475, -1.5686, -2.8025, -1.6268, -1.0806, -1.4369],
        [-2.9565, -2.1539, -1.4892, -3.6653, -1.4211, -1.0986],
        [-5.4592, -1.2265, -2.5233, -3.6044, -1.6392, -3.2157],
        [-1.9061, -1.0768, -1.7209, -1.0795, -2.3478, -1.8250],
        [-6.1821, -2.2655, -2.2008, -1.6302, -1.8910, -1.3273],
        [-4.2538, -4.1162, -2.0940, -2.2135, -2.1289, -1.6737],
        [-1.2348, -1.0875, -1.1548, -1.2220, -1.5480, -1.1951],
        [-3.9750, -1.2860, -2.4835, -1.3502, -2.0121, -2.4799],
        [-2.4470, -1.4526, -2.2765, -1.4329, -1.9291, -2.2343],
        [-3.0456, -1.4317, -2.1801, -2.9711, -1.8325, -1.1933],
        [-2.9952, -1.9301, -2.2016, -2.4141, -2.6332, -2.3631]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.0152, -11.6804, -14.0170, -10.6993, -15.8431, -13.8458,  -9.8552,
        -11.4338, -12.1481, -11.6628, -12.7846, -17.6683,  -9.9561, -15.4968,
        -16.4801,  -7.4422, -13.5867, -11.7724, -12.6543, -14.5372],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.2284e-01, -6.0530e-01, -2.0554e+00, -1.3557e-02, -8.1321e-03,
         -4.8213e-01],
        [-3.0399e+00, -5.7105e-03, -3.8231e+00, -1.3546e+00, -1.4850e+00,
         -1.6848e-01],
        [-3.9207e+00, -5.8639e-01, -3.7507e+00, -1.9300e-01, -3.8268e+00,
         -7.7927e-02],
        [-3.3566e+00, -3.6446e-01, -5.7513e-01, -3.6656e-01, -1.6915e+00,
         -9.5995e-01],
        [-3.9630e+00, -2.2727e+00, -3.0455e+00, -3.4829e-01, -3.7879e+00,
         -8.4986e-01],
        [-4.2323e+00, -2.0467e+00, -2.4448e+00, -2.7041e-02, -1.5534e+00,
         -4.4716e-02],
        [-2.0501e+00, -3.6279e-01, -1.0980e+00, -2.7569e+00, -4.9362e-01,
         -1.6926e-02],
        [-2.5942e+00, -4.4130e-03, -4.8345e+00, -6.9626e-01, -3.2075e-01,
         -2.4189e-02],
        [-3.9876e+00, -1.5061e-02, -2.7481e-01, -1.7238e-01, -4.5047e-02,
         -8.8604e-01],
        [-2.3243e+00, -1.4300e+00, -3.1023e+00, -8.5726e-01, -2.1331e-01,
         -1.3098e+00],
        [-4.1207e+00, -3.3098e+00, -1.1402e+00, -2.9964e+00, -7.2725e-01,
         -3.9655e-02],
        [-4.9798e+00, -1.3441e-01, -2.8921e+00, -4.0911e+00, -1.3235e+00,
         -3.5523e+00],
        [-2.2026e+00, -5.3376e-01, -1.4512e+00, -1.2571e-02, -2.6408e+00,
         -1.7725e+00],
        [-4.4194e+00, -1.6345e+00, -2.8027e+00, -1.0813e+00, -1.9060e+00,
         -9.6002e-01],
        [-3.3442e+00, -4.8728e+00, -2.1267e+00, -1.1188e+00, -2.5956e+00,
         -1.2009e+00],
        [-1.0762e+00, -1.3840e-01, -5.7834e-02, -5.6193e-01, -1.2870e+00,
         -2.1940e-01],
        [-4.4902e+00, -6.0081e-05, -2.6185e+00, -3.3161e-01, -2.2166e+00,
         -1.7725e+00],
        [-2.2770e+00, -5.7134e-01, -2.6296e+00, -6.4494e-01, -2.6389e+00,
         -2.2282e+00],
        [-2.5986e+00, -1.9610e-04, -2.9571e+00, -3.2043e+00, -2.4193e+00,
         -2.1770e-01],
        [-2.4203e+00, -1.1325e+00, -3.2072e+00, -3.0830e+00, -3.7573e+00,
         -2.9128e+00]], device='cuda:0')
tensor([ -3.4873,  -9.8769, -12.3556,  -7.3142, -14.2673, -10.3489,  -6.7783,
         -8.4743,  -5.3809,  -9.2370, -12.3340, -16.9731,  -8.6135, -12.8039,
        -15.2591,  -3.3407, -11.4295, -10.9900, -11.3972, -16.5131],
       device='cuda:0')
tensor([-4.5279, -1.8035, -1.6614, -3.3851, -1.5758, -3.4969, -3.0769, -2.9595,
        -6.7672, -2.4259, -0.4506, -0.6952, -1.3426, -2.6929, -1.2210, -4.1015,
        -2.1572, -0.7824, -1.2571,  1.9759], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.5746, -3.8144, -1.7519, -1.2527, -2.2015, -1.6096],
        [-1.4683, -1.0558, -0.9895, -0.9149, -2.1750, -1.3149],
        [-2.7281, -1.2637, -2.5600, -1.5670, -2.9823, -4.7096],
        [-2.7002, -1.4377, -3.0250, -1.6658, -2.5192, -2.1898],
        [-1.6169, -1.4869, -1.8403, -3.7637, -1.7359, -1.6655],
        [-3.6865, -2.5148, -2.3587, -4.3250, -1.9864, -1.5277],
        [-4.4606, -3.0767, -2.5734, -1.0449, -1.5600, -1.6225],
        [-1.5941, -1.0851, -1.6999, -0.9067, -1.4529, -1.0154],
        [-2.7005, -1.4725, -2.2230, -1.7209, -1.4748, -1.0179],
        [-0.7895, -1.4826, -1.2562, -1.3631, -1.3879, -1.3573],
        [-1.5119, -1.2360, -1.1248, -1.1551, -1.1974, -1.5488],
        [-2.7606, -1.3790, -2.8355, -1.6473, -2.0082, -3.2638],
        [-1.0948, -1.4635, -1.4488, -2.8760, -1.1980, -3.8118],
        [-3.8109, -1.5609, -3.6850, -3.5252, -2.8697, -2.2981],
        [-5.4762, -1.0437, -1.3448, -1.1835, -2.2571, -1.5286],
        [-5.0485, -1.9019, -2.0816, -2.3008, -2.4289, -3.7694],
        [-9.3533, -3.6022, -2.6512, -1.4714, -2.4720, -1.9214],
        [-1.2933, -1.1585, -2.1229, -1.3012, -1.5794, -1.2632],
        [-2.1949, -1.2393, -1.4626, -1.3283, -1.6189, -2.6436],
        [-4.9002, -1.6725, -1.8819, -1.4918, -1.8284, -1.1743]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.2047,  -7.9185, -15.8106, -13.5377, -12.1092, -16.3990, -14.3381,
         -7.7542, -10.6096,  -7.6366,  -7.7740, -13.8944, -11.8930, -17.7498,
        -12.8339, -17.5312, -21.4715,  -8.7183, -10.4876, -12.9492],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.6384e+00, -4.2511e+00, -1.1847e+00, -1.5497e-01, -2.9588e+00,
         -1.1865e+00],
        [-1.7119e+00, -4.4798e-03, -1.1171e-02, -1.9056e-02, -2.4176e+00,
         -4.7809e-01],
        [-3.2426e+00, -1.0412e-01, -2.9592e+00, -8.6765e-01, -3.5634e+00,
         -3.4012e+00],
        [-2.5562e+00, -2.0935e-01, -3.6549e+00, -1.3119e+00, -4.3325e+00,
         -2.0006e+00],
        [-1.1123e+00, -2.9712e-01, -1.9507e+00, -3.9137e+00, -2.0562e+00,
         -9.9538e-01],
        [-3.2824e+00, -2.8753e+00, -3.3987e+00, -5.4185e+00, -1.8866e+00,
         -5.7827e-01],
        [-2.7952e+00, -2.6736e+00, -1.9484e+00, -4.2762e-02, -4.5344e-01,
         -1.0607e+00],
        [-2.1332e+00, -6.9913e-02, -1.1247e+00, -1.7301e-01, -1.2666e+00,
         -1.5666e-01],
        [-2.8514e+00, -1.9403e-02, -2.7477e+00, -1.1497e+00, -7.7221e-01,
         -1.1310e-01],
        [-2.8133e-04, -1.7672e-01, -2.4918e-02, -4.2433e-01, -1.1570e-01,
         -5.7902e-02],
        [-1.3294e+00, -1.1731e-01, -1.0999e-01, -4.0915e-02, -2.8703e-01,
         -1.0143e+00],
        [-2.1029e+00, -8.2599e-01, -3.2584e+00, -1.0541e+00, -2.0310e+00,
         -3.4432e+00],
        [-9.3001e-01, -6.6947e-01, -8.5284e-01, -3.1078e+00, -6.3663e-01,
         -2.4563e+00],
        [-4.2047e+00, -8.2588e-01, -4.4213e+00, -3.7359e+00, -3.0227e+00,
         -2.1715e+00],
        [-5.3247e+00, -9.5590e-02, -3.7639e-01, -2.3251e-01, -2.5814e+00,
         -1.5847e+00],
        [-3.5094e+00, -9.2111e-01, -2.3620e+00, -2.4703e+00, -3.3916e+00,
         -3.8496e+00],
        [-5.5231e+00, -3.5968e+00, -2.7879e+00, -3.6922e-01, -3.3106e+00,
         -1.9233e+00],
        [-1.0066e+00, -9.5013e-02, -2.1201e+00, -1.1797e+00, -1.1526e+00,
         -1.2358e-01],
        [-2.6766e+00, -2.9741e-01, -6.3489e-01, -2.3353e-01, -1.1726e+00,
         -1.8114e+00],
        [-3.9111e+00, -2.9181e-01, -1.4082e+00, -6.7632e-01, -2.1899e+00,
         -5.2102e-01]], device='cuda:0')
tensor([-11.3744,  -4.6424, -14.1382, -14.0654, -10.3255, -17.4398,  -8.9740,
         -4.9241,  -7.6535,  -0.7998,  -2.8989, -12.7156,  -8.6530, -18.3820,
        -10.1952, -16.5040, -17.5110,  -5.6775,  -6.8265,  -8.9983],
       device='cuda:0')
tensor([-1.8303, -3.2762, -1.6725,  0.5277, -1.7837,  1.0408, -5.3641, -2.8300,
        -2.9561, -6.8367, -4.8751, -1.1788, -3.2400,  0.6321, -2.6387, -1.0272,
        -3.9606, -3.0408, -3.6611, -3.9509], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[ -2.8284,  -4.1478,  -1.8070,  -2.9707,  -1.2926,  -1.3137],
        [ -1.5928,  -2.1100,  -1.8663,  -3.1927,  -1.6661,  -1.1002],
        [ -2.4908,  -1.5547,  -3.5458,  -2.9747,  -1.8300,  -1.1286],
        [ -2.9368,  -0.9832,  -1.3364,  -1.0435,  -1.6794,  -2.0689],
        [ -4.2067,  -2.5045,  -2.3794,  -1.6098,  -2.1180,  -1.8392],
        [ -3.3783,  -2.2265,  -2.1089,  -2.2419,  -2.1707,  -1.5748],
        [ -1.7973,  -1.2509,  -1.7342,  -1.0957,  -1.3756,  -1.2734],
        [-12.8999,  -1.1128,  -5.5524,  -1.4343,  -2.6973,  -2.7822],
        [ -2.1061,  -2.3158,  -1.7284,  -3.2416,  -1.6204,  -0.8851],
        [ -5.3469,  -1.7565,  -1.6246,  -1.4057,  -3.2381,  -5.6546],
        [ -3.4619,  -2.6777,  -2.7297,  -1.2421,  -1.4183,  -1.1872],
        [ -1.3680,  -1.9270,  -2.3169,  -3.4290,  -1.2532,  -1.1016],
        [ -2.8676,  -1.7359,  -1.9159,  -1.3192,  -1.5174,  -1.0617],
        [ -2.7099,  -1.4322,  -1.8563,  -1.5808,  -1.5084,  -1.2219],
        [ -3.3610,  -0.9343,  -1.3668,  -1.0465,  -1.8547,  -1.5651],
        [ -1.5832,  -1.6453,  -1.5644,  -1.0952,  -2.1290,  -4.4681],
        [ -4.2225,  -2.0723,  -1.9739,  -1.0417,  -1.9715,  -2.1537],
        [ -3.1742,  -3.6149,  -2.1101,  -2.2054,  -2.7638,  -1.2345],
        [ -1.2977,  -3.1728,  -1.5771,  -5.4618,  -1.3391,  -1.0630],
        [ -2.9235,  -1.2475,  -2.6284,  -1.3269,  -1.7219,  -4.4203]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-14.3603, -11.5282, -13.5246, -10.0482, -14.6577, -13.7011,  -8.5273,
        -26.4789, -11.8976, -19.0263, -12.7170, -11.3956, -10.4178, -10.3094,
        -10.1285, -12.4853, -13.4356, -15.1029, -13.9114, -14.2686],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.1972e+00, -6.2734e+00, -2.1043e+00, -2.4658e+00, -7.9081e-02,
         -5.3703e-01],
        [-2.2823e+00, -3.4729e+00, -2.0289e+00, -3.2075e+00, -1.6956e+00,
         -6.6930e-01],
        [-2.3788e+00, -7.6300e-01, -4.2759e+00, -3.6834e+00, -1.8024e+00,
         -1.9380e-01],
        [-3.4314e+00, -5.2521e-01, -3.3503e-01, -3.0060e-02, -1.5366e+00,
         -2.6350e+00],
        [-2.4666e+00, -2.0149e+00, -3.7666e+00, -9.2307e-01, -2.5048e+00,
         -1.9918e+00],
        [-2.4998e+00, -1.9254e+00, -2.3035e+00, -1.0093e+00, -2.7464e+00,
         -7.4521e-01],
        [-2.8139e+00, -9.8805e-02, -1.6475e+00, -6.2292e-01, -9.0794e-02,
         -1.7777e-01],
        [-8.2758e+00, -4.7998e-01, -5.8571e+00, -6.7822e-01, -3.7746e+00,
         -2.9731e+00],
        [-9.6889e-01, -2.3128e+00, -1.4323e+00, -3.6992e+00, -1.2628e+00,
         -1.8356e-01],
        [-3.8676e+00, -8.7398e-01, -1.8204e+00, -9.4127e-01, -4.8752e+00,
         -5.8414e+00],
        [-2.2796e+00, -2.5556e+00, -2.4578e+00, -7.3485e-01, -3.5151e-01,
         -4.6932e-03],
        [-2.0844e-02, -2.6084e+00, -2.6360e+00, -3.1415e+00, -1.9206e-01,
         -2.1182e-02],
        [-2.7725e+00, -2.3610e+00, -1.6316e+00, -1.8424e-01, -1.5896e+00,
         -4.6030e-02],
        [-2.1326e+00, -1.7441e-02, -1.6483e+00, -9.6782e-01, -1.2188e+00,
         -2.9317e-01],
        [-3.4862e+00, -1.1664e-01, -4.8487e-01, -1.6149e-02, -1.9506e+00,
         -1.5342e+00],
        [-1.6631e+00, -7.6956e-01, -1.3677e+00, -2.0591e-01, -2.4132e+00,
         -4.1584e+00],
        [-2.8203e+00, -1.4863e+00, -2.3277e+00, -6.8167e-02, -2.0955e+00,
         -2.6225e+00],
        [-1.8060e+00, -3.0145e+00, -1.9574e+00, -1.8066e+00, -3.0472e+00,
         -3.7221e-01],
        [-1.1014e-01, -4.6280e+00, -1.1914e+00, -4.3740e+00, -9.6736e-02,
         -8.8201e-03],
        [-3.1506e+00, -3.4777e-01, -2.6849e+00, -6.6343e-02, -1.7234e+00,
         -3.1730e+00]], device='cuda:0')
tensor([-13.6569, -13.3565, -13.0973,  -8.4932, -13.6678, -11.2296,  -5.4516,
        -22.0387,  -9.8595, -18.2199,  -8.3840,  -8.6201,  -8.5850,  -6.2782,
         -7.5886, -10.5779, -11.4204, -12.0038, -10.4091, -11.1461],
       device='cuda:0')
tensor([-0.7034,  1.8283, -0.4273, -1.5550, -0.9898, -2.4715, -3.0756, -4.4402,
        -2.0381, -0.8063, -4.3329, -2.7755, -1.8327, -4.0312, -2.5398, -1.9075,
        -2.0152, -3.0991, -3.5023, -3.1225], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[ -3.0747,  -1.2520,  -1.2824,  -2.4239,  -1.5157,  -1.1403],
        [ -9.3060,  -2.5746,  -1.9006,  -1.9618,  -2.3845,  -2.3221],
        [ -7.3464,  -2.5784,  -3.5333,  -0.9364,  -1.9434,  -1.4949],
        [ -1.1023,  -1.6404,  -3.2491,  -1.6441,  -1.5015,  -1.2056],
        [ -2.4387,  -1.5853,  -2.7811,  -1.4034,  -1.4172,  -1.0700],
        [ -6.4868,  -2.6442,  -3.2129,  -2.0203,  -1.3222,  -1.5681],
        [ -1.1493,  -1.8765,  -3.9907,  -2.3509,  -1.1146,  -1.6551],
        [ -4.1974,  -1.3653,  -2.7005,  -1.5043,  -1.8913,  -1.4338],
        [ -3.3746,  -1.1094,  -1.4653,  -1.2104,  -2.2767,  -2.5721],
        [-10.9796,  -3.8479,  -2.2854,  -1.3346,  -2.2251,  -1.9679],
        [ -3.3153,  -2.2244,  -2.1340,  -2.9625,  -2.2635,  -1.4469],
        [ -1.3694,  -1.1324,  -1.3082,  -1.3545,  -1.9657,  -1.2343],
        [ -3.4887,  -1.6984,  -2.9000,  -1.3227,  -1.5200,  -1.2471],
        [ -3.7586,  -1.2227,  -2.1915,  -1.1541,  -2.4525,  -1.2325],
        [ -1.0071,  -1.5389,  -1.3564,  -1.6866,  -2.0874,  -2.3568],
        [ -3.4992,  -2.4789,  -1.7567,  -1.5067,  -2.1983,  -2.8913],
        [ -2.6240,  -2.4872,  -2.6695,  -0.9689,  -2.1027,  -1.6779],
        [ -1.1855,  -1.2191,  -1.3003,  -1.4928,  -1.4411,  -1.3969],
        [ -2.8881,  -1.4199,  -1.9535,  -1.5455,  -1.4382,  -0.9436],
        [ -3.5605,  -1.5349,  -1.2762,  -1.3507,  -1.4430,  -2.0485]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.6891, -20.4497, -17.8328, -10.3430, -10.6958, -17.2545, -12.1372,
        -13.0926, -12.0085, -22.6405, -14.3466,  -8.3647, -12.1769, -12.0118,
        -10.0333, -14.3311, -12.5301,  -8.0357, -10.1890, -11.2138],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.5453e+00, -1.0293e-01, -8.5296e-01, -2.4923e+00, -1.3423e+00,
         -5.1217e-03],
        [-6.4885e+00, -2.9570e+00, -2.0157e+00, -2.3452e+00, -3.3091e+00,
         -3.1829e+00],
        [-5.3488e+00, -2.0972e+00, -3.1450e+00, -1.1068e-01, -1.9554e+00,
         -7.0762e-01],
        [-2.7575e-01, -1.9454e+00, -3.8164e+00, -1.0563e+00, -7.5292e-01,
         -6.8172e-03],
        [-2.5868e+00, -1.4741e-02, -4.2222e+00, -2.7595e-01, -7.7401e-01,
         -6.2487e-01],
        [-4.6357e+00, -4.0515e+00, -4.4233e+00, -2.3387e+00, -1.2156e-02,
         -3.8199e-01],
        [-5.7673e-01, -2.5353e+00, -4.5933e+00, -2.2353e+00, -1.0866e-01,
         -5.4164e-01],
        [-4.8059e+00, -3.4237e-01, -2.9686e+00, -1.1508e+00, -1.6488e+00,
         -7.2777e-01],
        [-4.0619e+00, -7.0941e-01, -7.3594e-01, -2.7652e-01, -2.5680e+00,
         -3.3974e+00],
        [-7.4792e+00, -4.0380e+00, -3.2151e+00, -3.9519e-01, -2.9006e+00,
         -2.1718e+00],
        [-2.4128e+00, -2.0007e+00, -2.3896e+00, -1.6723e+00, -3.1132e+00,
         -1.7571e-01],
        [-1.5074e+00, -2.0854e-01, -2.5014e-01, -1.0194e+00, -2.0428e+00,
         -4.1316e-01],
        [-4.0546e+00, -6.9360e-01, -3.4691e+00, -1.9064e-01, -9.4574e-01,
         -5.0810e-01],
        [-3.2823e+00, -3.0885e-01, -2.4220e+00, -7.6843e-02, -3.5378e+00,
         -5.9754e-01],
        [-7.6010e-01, -7.2908e-03, -6.4732e-01, -1.1306e+00, -3.0605e+00,
         -1.9815e+00],
        [-2.2416e+00, -2.5115e+00, -1.8705e+00, -1.1255e+00, -2.7201e+00,
         -3.5286e+00],
        [-1.5111e+00, -2.4004e+00, -3.2497e+00, -1.9960e-03, -2.6118e+00,
         -1.2696e+00],
        [-1.0356e+00, -4.1307e-03, -4.7501e-01, -7.5022e-01, -1.4303e-01,
         -4.0505e-01],
        [-3.2870e+00, -2.1369e-01, -1.9045e+00, -7.1745e-01, -6.9271e-01,
         -3.5722e-01],
        [-2.3373e+00, -1.7940e-01, -6.1400e-02, -2.3459e-01, -1.6184e-01,
         -1.5901e+00]], device='cuda:0')
tensor([ -7.3410, -20.2985, -13.3647,  -7.8537,  -8.4987, -15.8434, -10.5908,
        -11.6442, -11.7492, -20.1999, -11.7642,  -5.4414,  -9.8618, -10.2253,
         -7.5873, -13.9977, -11.0445,  -2.8130,  -7.1726,  -4.5646],
       device='cuda:0')
tensor([-3.3481, -0.1511, -4.4681, -2.4894, -2.1972, -1.4111, -1.5463, -1.4484,
        -0.2593, -2.4406, -2.5825, -2.9233, -2.3151, -1.7865, -2.4460, -0.3334,
        -1.4856, -5.2227, -3.0164, -6.6492], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.7382, -3.7988, -2.7536, -1.2152, -1.9776, -1.5624],
        [-2.5617, -1.2900, -1.2004, -1.9872, -1.4634, -1.2653],
        [-2.2866, -1.5319, -2.5025, -1.8184, -1.5635, -0.9255],
        [-2.8388, -1.4628, -1.7088, -1.3103, -1.9919, -1.4413],
        [-2.1309, -1.6243, -1.3800, -0.9853, -1.2929, -1.4735],
        [-3.8487, -1.4044, -3.0593, -2.5887, -1.8042, -1.5562],
        [-1.0724, -1.1344, -3.2561, -2.1497, -1.2938, -1.5584],
        [-1.6195, -2.0096, -2.9859, -3.3441, -2.2919, -4.8596],
        [-2.2748, -3.1264, -1.6137, -1.9011, -4.2390, -1.5749],
        [-7.8925, -3.2053, -1.8184, -1.4611, -2.2151, -1.7510],
        [-2.6349, -3.8157, -2.3300, -1.7099, -2.9330, -2.7525],
        [-1.2382, -1.1972, -1.5222, -1.0361, -1.8116, -1.2220],
        [-1.8600, -1.6630, -2.3412, -1.4393, -1.9598, -2.1578],
        [-4.9132, -1.5384, -1.7965, -1.3228, -2.3101, -1.2751],
        [-4.0930, -1.2885, -1.2294, -1.3638, -2.2084, -1.7375],
        [-1.7981, -1.4720, -1.5503, -1.1114, -2.0120, -1.2054],
        [-3.6483, -1.9077, -3.2015, -1.2648, -2.2680, -1.6911],
        [-4.6776, -3.6070, -5.4901, -1.0191, -2.0286, -1.5608],
        [-1.3542, -1.3801, -3.5516, -1.0213, -2.2713, -1.2023],
        [-1.0319, -1.4262, -1.7248, -1.2483, -1.9822, -1.0954]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-14.0457,  -9.7680, -10.6284, -10.7540,  -8.8869, -14.2614, -10.4647,
        -17.1106, -14.7298, -18.3434, -16.1761,  -8.0273, -11.4211, -13.1562,
        -11.9206,  -9.1491, -13.9815, -18.3831, -10.7807,  -8.5088],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.9282e+00, -3.9575e+00, -3.3274e+00, -1.4369e-01, -2.3523e+00,
         -1.0751e+00],
        [-3.7987e+00, -6.0954e-01, -9.0255e-02, -1.6223e+00, -4.1531e-01,
         -4.2155e-03],
        [-2.3761e+00, -1.3794e+00, -3.2755e+00, -1.2391e+00, -1.5117e+00,
         -9.5402e-02],
        [-1.7916e+00, -1.1921e-07, -1.3942e+00, -5.1426e-02, -1.3844e+00,
         -3.6120e-01],
        [-9.8017e-01, -1.6951e+00, -6.7515e-01, -4.6319e-02, -3.4209e-03,
         -1.3228e+00],
        [-3.2928e+00, -8.0021e-01, -3.6667e+00, -1.9499e+00, -1.7277e+00,
         -1.1037e+00],
        [-4.4382e-02, -3.7136e-01, -3.5703e+00, -2.0239e+00, -3.4784e-01,
         -1.3385e+00],
        [-1.8811e+00, -1.5066e+00, -3.7477e+00, -4.0316e+00, -2.3768e+00,
         -3.7939e+00],
        [-2.8465e+00, -2.2526e+00, -1.0732e+00, -1.2273e+00, -4.6924e+00,
         -1.5071e+00],
        [-6.1490e+00, -2.4822e+00, -1.3476e+00, -5.6130e-01, -2.6817e+00,
         -1.5168e+00],
        [-1.8241e+00, -4.4694e+00, -3.3275e+00, -3.7678e-01, -3.9866e+00,
         -3.8225e+00],
        [-9.8534e-01, -2.4953e-01, -7.4936e-01, -4.7295e-01, -1.6924e+00,
         -1.4117e-01],
        [-2.1333e+00, -1.9002e+00, -2.4344e+00, -5.5899e-01, -1.9714e+00,
         -1.5064e+00],
        [-3.8763e+00, -1.3854e-01, -1.1205e+00, -4.9567e-02, -3.8809e+00,
         -6.4618e-01],
        [-3.5672e+00, -7.2384e-01, -6.4458e-02, -5.7808e-01, -3.0509e+00,
         -1.7513e+00],
        [-2.1238e+00, -4.8288e-02, -1.2573e+00, -8.5322e-02, -2.3929e+00,
         -1.8908e-01],
        [-2.9892e+00, -1.4086e+00, -4.0007e+00, -1.0199e+00, -2.8195e+00,
         -1.1965e+00],
        [-3.2332e+00, -3.2387e+00, -4.4553e+00, -2.0719e-04, -2.0390e+00,
         -9.3421e-01],
        [-1.5105e+00, -1.8440e-01, -4.3085e+00, -2.9734e-01, -2.7388e+00,
         -5.4343e-01],
        [-4.2402e-01, -1.1488e+00, -1.5442e+00, -9.3709e-02, -1.9794e+00,
         -2.5577e-01]], device='cuda:0')
tensor([-12.7843,  -6.5404,  -9.8772,  -4.9828,  -4.7230, -12.5411,  -7.6963,
        -17.3376, -13.5990, -14.7385, -17.8069,  -4.2907, -10.5047,  -9.7120,
         -9.7358,  -6.0966, -13.4345, -13.9006,  -9.5830,  -5.4459],
       device='cuda:0')
tensor([-1.2614, -3.2276, -0.7512, -5.7712, -4.1639, -1.7204, -2.7684,  0.2271,
        -1.1309, -3.6049,  1.6308, -3.7366, -0.9164, -3.4442, -2.1849, -3.0525,
        -0.5470, -4.4825, -1.1978, -3.0629], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.1003, -1.0004, -1.1942, -1.6574, -1.9783, -1.4680],
        [-1.8846, -1.9972, -1.4971, -1.3030, -1.7947, -1.3023],
        [-4.2505, -2.5790, -1.9697, -1.5407, -2.4173, -1.4652],
        [-1.4688, -1.3071, -1.4174, -1.1839, -1.4810, -1.3922],
        [-6.1058, -1.1510, -2.4565, -1.2773, -1.6615, -0.9763],
        [-3.0804, -1.5775, -3.0145, -1.2166, -1.5043, -1.2096],
        [-2.1151, -1.1962, -1.6863, -1.7801, -1.5730, -4.6138],
        [-2.2611, -1.3417, -1.2638, -1.3227, -2.3135, -2.4564],
        [-5.5786, -2.9242, -1.9997, -1.1243, -1.8643, -1.3183],
        [-3.4643, -2.1934, -2.1655, -1.1921, -2.4202, -1.7917],
        [-1.3678, -1.3548, -1.4010, -1.1122, -1.4197, -1.1442],
        [-1.2087, -1.3994, -3.0547, -1.4679, -1.6121, -1.0054],
        [-1.4610, -1.3204, -2.1373, -1.1095, -1.7452, -1.2714],
        [-5.3771, -1.1304, -1.1536, -1.7824, -1.6362, -1.2351],
        [-5.1313, -2.1591, -2.0960, -1.2345, -2.3743, -1.8613],
        [-3.1561, -2.9109, -2.2631, -1.8355, -1.7760, -1.4586],
        [-1.0384, -1.2284, -2.1035, -0.9619, -1.8347, -1.1885],
        [-3.0102, -2.1299, -3.2375, -1.3886, -1.5400, -1.2850],
        [-1.4814, -1.2915, -1.9867, -1.1549, -1.4472, -1.2604],
        [-3.5425, -1.2948, -1.0763, -1.9646, -2.0564, -1.7007]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.3986,  -9.7788, -14.2224,  -8.2504, -13.6283, -11.6029, -12.9644,
        -10.9593, -14.8094, -13.2271,  -7.7996,  -9.7483,  -9.0447, -12.3148,
        -14.8565, -13.4002,  -8.3555, -12.5912,  -8.6220, -11.6354],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.3120e+00, -1.4188e-01, -1.2466e-01, -1.0619e+00, -2.1643e+00,
         -9.5278e-01],
        [-5.3654e-01, -1.1989e+00, -1.2919e+00, -3.6179e-01, -1.5460e+00,
         -2.4847e-01],
        [-2.9685e+00, -2.7981e+00, -2.0832e+00, -4.4741e-01, -3.3449e+00,
         -1.9101e-01],
        [-1.8737e+00, -5.9415e-01, -6.6613e-01, -1.1922e-01, -1.1226e-01,
         -4.3422e-01],
        [-5.8680e+00, -1.9627e-02, -2.6474e+00, -4.6697e-02, -1.4081e+00,
         -1.4041e-01],
        [-2.5608e+00, -4.0983e-01, -3.6456e+00, -1.2291e-02, -1.1525e+00,
         -1.2646e-01],
        [-2.1646e+00, -4.4918e-04, -1.0747e+00, -1.2638e+00, -1.3216e+00,
         -5.3439e+00],
        [-2.9003e+00, -6.9173e-02, -7.4483e-01, -4.0518e-01, -2.8985e+00,
         -1.6703e+00],
        [-4.0950e+00, -3.4479e+00, -2.4858e+00, -1.9768e-01, -1.6863e+00,
         -3.9220e-04],
        [-2.6129e+00, -1.7385e+00, -2.2096e+00, -2.1568e-01, -3.1056e+00,
         -1.7283e+00],
        [-1.4886e+00, -8.2675e-01, -6.8150e-01, -1.0882e-01, -2.4103e-01,
         -1.1702e-02],
        [-1.0562e+00, -1.8268e-01, -4.3784e+00, -4.0274e-01, -1.2824e+00,
         -7.0600e-02],
        [-1.1360e+00, -4.3785e-01, -2.2503e+00, -4.8757e-05, -2.3657e+00,
         -3.7350e-01],
        [-4.8347e+00, -2.5487e-02, -4.5984e-01, -1.3189e+00, -1.5552e+00,
         -2.8958e-02],
        [-3.7804e+00, -1.7312e+00, -2.8154e+00, -4.6776e-01, -3.3255e+00,
         -1.9243e+00],
        [-2.2057e+00, -3.0829e+00, -2.5896e+00, -7.0454e-01, -1.2112e+00,
         -5.3635e-01],
        [-2.7537e-01, -8.1090e-01, -1.8411e+00, -1.4172e-01, -1.7043e+00,
         -2.2125e-01],
        [-3.4637e+00, -1.5849e+00, -4.5761e+00, -2.1702e-01, -1.0121e+00,
         -3.7596e-01],
        [-1.1994e+00, -1.1986e-02, -1.8491e+00, -5.8403e-02, -1.2977e+00,
         -6.7317e-04],
        [-3.0446e+00, -4.1254e-02, -4.4868e-03, -1.7930e+00, -2.7518e+00,
         -1.3762e+00]], device='cuda:0')
tensor([ -7.7576,  -5.1836, -11.8331,  -3.7997, -10.1302,  -7.9074, -11.1691,
         -8.6882, -11.9131, -11.6105,  -3.3584,  -7.3730,  -6.5634,  -8.2231,
        -14.0444, -10.3303,  -4.9947, -11.2297,  -4.4173,  -9.0114],
       device='cuda:0')
tensor([-2.6410, -4.5952, -2.3893, -4.4507, -3.4981, -3.6955, -1.7953, -2.2711,
        -2.8962, -1.6166, -4.4412, -2.3753, -2.4812, -4.0916, -0.8121, -3.0699,
        -3.3608, -1.3614, -4.2047, -2.6240], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.7362, -1.2277, -3.2799, -1.2257, -1.8725, -1.4324],
        [-2.4425, -1.4000, -1.5046, -1.1476, -1.9274, -2.9119],
        [-3.6006, -3.8429, -2.1458, -1.1024, -2.7382, -1.3552],
        [-3.5806, -4.7669, -2.2281, -1.6030, -2.3700, -1.7768],
        [-1.2382, -1.2054, -1.1969, -0.9765, -1.5931, -1.5720],
        [-3.2456, -1.4118, -2.2276, -1.2738, -1.4080, -2.0037],
        [-1.8014, -1.6713, -1.7057, -1.5857, -1.7546, -0.9783],
        [-2.0719, -1.5875, -2.7136, -5.1315, -3.4365, -2.6385],
        [-2.5067, -2.7582, -1.1967, -1.1113, -2.0062, -1.2708],
        [-2.5951, -2.5133, -2.0082, -0.9280, -1.2977, -1.2596],
        [-4.6542, -1.3380, -1.5221, -1.1446, -2.5248, -1.3468],
        [-1.7451, -1.4080, -1.8566, -1.3739, -1.3212, -1.1831],
        [-0.9085, -1.1372, -1.3262, -1.7423, -1.4005, -1.4089],
        [-0.8993, -1.5887, -1.4324, -1.1904, -1.5511, -2.3039],
        [-3.6837, -6.0784, -3.1051, -1.4451, -2.0304, -2.2100],
        [-2.5213, -1.8824, -1.9492, -3.5212, -1.3490, -1.6646],
        [-2.1437, -4.5891, -2.4929, -4.1808, -1.6282, -2.0553],
        [-1.6455, -1.2292, -1.9321, -1.4025, -2.1912, -1.3667],
        [-2.1206, -1.2378, -1.7535, -1.2090, -1.4277, -1.3206],
        [-1.6062, -1.2826, -1.4364, -1.2593, -2.0434, -1.5690]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.7744, -11.3340, -14.7851, -16.3253,  -7.7821, -11.5706,  -9.4971,
        -17.5795, -10.8499, -10.6018, -12.5305,  -8.8879,  -7.9237,  -8.9659,
        -18.5527, -12.8877, -17.0899,  -9.7671,  -9.0692,  -9.1967],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.4584e+00, -2.3505e-01, -3.1884e+00, -7.2807e-03, -2.1677e+00,
         -5.3384e-01],
        [-2.4178e+00, -3.9410e-01, -1.2005e+00, -9.7877e-02, -2.0244e+00,
         -3.2998e+00],
        [-2.6030e+00, -4.6195e+00, -2.9276e+00, -2.4243e-01, -4.0077e+00,
         -1.1879e-01],
        [-2.4795e+00, -4.5796e+00, -2.2846e+00, -8.2173e-01, -2.8974e+00,
         -1.7004e+00],
        [-9.5680e-01, -1.5752e-01, -4.1638e-02, -3.5356e-03, -9.7151e-01,
         -1.1304e+00],
        [-2.6673e+00, -1.5020e+00, -2.1290e+00, -5.8412e-02, -8.0144e-01,
         -2.2977e+00],
        [-1.9395e+00, -7.1894e-02, -2.9108e-01, -8.0970e-01, -2.1452e+00,
         -5.3920e-03],
        [-1.6003e+00, -8.7063e-01, -2.9954e+00, -4.8674e+00, -3.8417e+00,
         -2.4543e+00],
        [-9.3515e-01, -2.9940e+00, -6.5967e-02, -3.0969e-01, -1.8711e+00,
         -2.1744e-02],
        [-1.5291e+00, -2.0513e+00, -1.8363e+00, -2.7107e-02, -2.5449e-01,
         -2.9421e-01],
        [-5.5370e+00, -1.5237e-01, -6.9917e-01, -1.8390e-03, -2.5140e+00,
         -6.8987e-01],
        [-1.6538e+00, -6.4967e-02, -1.9105e+00, -3.2676e-01, -1.0534e+00,
         -5.8137e-01],
        [-6.3945e-02, -1.2684e-04, -4.1852e-01, -1.2778e+00, -9.0519e-02,
         -5.4930e-02],
        [-2.1431e-01, -4.3543e-01, -1.1331e+00, -2.0051e-01, -9.2066e-01,
         -1.6790e+00],
        [-2.7211e+00, -6.1885e+00, -3.8424e+00, -9.0316e-01, -2.2859e+00,
         -2.0929e+00],
        [-1.4380e+00, -1.1015e+00, -1.7268e+00, -3.1602e+00, -4.7289e-01,
         -1.1062e+00],
        [-1.3756e+00, -5.5557e+00, -2.4427e+00, -3.3853e+00, -1.3488e+00,
         -1.8368e+00],
        [-1.4806e+00, -9.8275e-01, -2.0543e+00, -3.7831e-01, -2.6727e+00,
         -5.3336e-01],
        [-1.7619e+00, -9.5141e-04, -1.4006e+00, -1.4465e-01, -9.0272e-01,
         -6.8321e-02],
        [-1.3537e+00, -2.3375e-01, -4.8775e-01, -2.8903e-01, -2.1516e+00,
         -1.3472e+00]], device='cuda:0')
tensor([ -8.5907,  -9.4345, -14.5190, -14.7632,  -3.2614,  -9.4559,  -5.2628,
        -16.6295,  -6.1976,  -5.9926,  -9.5942,  -5.5908,  -1.9058,  -4.5830,
        -18.0339,  -9.0055, -15.9449,  -8.1019,  -4.2791,  -5.8630],
       device='cuda:0')
tensor([-3.1837, -1.8995, -0.2661, -1.5621, -4.5207, -2.1147, -4.2343, -0.9500,
        -4.6523, -4.6093, -2.9363, -3.2972, -6.0179, -4.3829, -0.5187, -3.8821,
        -1.1450, -1.6652, -4.7901, -3.3337], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.8329, -3.2563, -2.5559, -1.5885, -2.5657, -1.5024],
        [-1.2935, -2.9588, -3.6136, -5.9953, -1.9126, -1.3386],
        [-1.6849, -3.2802, -1.2499, -1.3043, -1.6920, -1.5801],
        [-3.0530, -1.7220, -1.6749, -1.5491, -1.6411, -1.1365],
        [-3.0002, -2.0416, -1.3443, -0.8879, -3.8003, -0.9812],
        [-5.2428, -1.1572, -1.2584, -1.2365, -2.8403, -1.5226],
        [-4.0398, -2.0785, -1.6394, -1.0940, -2.1468, -1.6775],
        [-3.3998, -5.0519, -2.0227, -1.3048, -1.8352, -1.8754],
        [-1.3805, -1.4846, -1.2189, -2.4238, -1.3416, -1.2260],
        [-3.7445, -1.2331, -3.8844, -1.1895, -1.2559, -1.8331],
        [-1.8844, -1.4321, -1.6433, -1.6970, -2.7057, -1.0585],
        [-0.8857, -2.0785, -2.3565, -2.0858, -1.4639, -3.2674],
        [-3.9593, -1.7600, -2.4194, -2.5017, -3.3379, -2.5183],
        [-3.3444, -5.4286, -2.0633, -1.0102, -2.2773, -1.4335],
        [-1.2326, -1.1571, -1.5156, -0.8194, -1.1882, -1.0287],
        [-3.1692, -1.4003, -1.9912, -1.6951, -1.8330, -0.8141],
        [-2.1904, -1.2536, -1.3386, -1.3492, -1.4139, -1.4954],
        [-3.7286, -2.0288, -1.5702, -1.3333, -1.2781, -1.7901],
        [-1.6118, -1.6471, -2.4320, -3.3471, -1.7881, -1.4596],
        [-2.3519, -2.2339, -2.1123, -2.1737, -1.7753, -2.0907]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-16.3018, -17.1124, -10.7914, -10.7766, -12.0555, -13.2578, -12.6760,
        -15.4898,  -9.0754, -13.1405, -10.4209, -12.1376, -16.4967, -15.5574,
         -6.9416, -10.9029,  -9.0411, -11.7290, -12.2857, -12.7377],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.9622e+00, -2.7055e+00, -2.9729e+00, -7.8532e-01, -2.8964e+00,
         -9.9570e-01],
        [-1.3047e-02, -4.3715e+00, -4.4395e+00, -4.9591e+00, -2.0720e+00,
         -4.2365e-01],
        [-1.6020e+00, -5.7862e+00, -2.6685e-03, -1.1514e-01, -1.7125e+00,
         -1.1779e+00],
        [-2.6739e+00, -1.5160e-02, -7.9810e-02, -7.8098e-01, -3.9485e-01,
         -8.4152e-02],
        [-4.4289e+00, -2.0544e+00, -8.9802e-01, -2.9830e-02, -3.6381e+00,
         -8.1857e-02],
        [-4.5881e+00, -1.3733e-03, -7.1136e-01, -5.0023e-03, -3.7642e+00,
         -1.1727e+00],
        [-2.9025e+00, -1.5476e+00, -1.1233e+00, -1.4034e-02, -2.6871e+00,
         -1.3804e+00],
        [-2.3727e+00, -4.4751e+00, -1.7192e+00, -8.0137e-01, -1.7718e+00,
         -1.7258e+00],
        [-1.5795e-01, -5.1058e-01, -6.3410e-02, -1.8034e+00, -3.6739e-02,
         -3.8371e-01],
        [-3.4816e+00, -6.3302e-02, -4.0666e+00, -4.7791e-02, -6.1835e-01,
         -1.8461e+00],
        [-1.8594e+00, -4.5185e-01, -7.1501e-01, -1.4520e+00, -4.4342e+00,
         -3.3497e-01],
        [-1.7435e-01, -1.5389e+00, -3.2674e+00, -1.7451e+00, -8.8680e-01,
         -3.3352e+00],
        [-3.6465e+00, -8.8974e-01, -3.3330e+00, -3.3134e+00, -4.3229e+00,
         -3.0198e+00],
        [-1.8579e+00, -5.0550e+00, -1.6881e+00, -5.9090e-02, -2.7884e+00,
         -3.7813e-01],
        [-4.8670e-01, -3.4353e-01, -5.3438e-01, -2.7076e-02, -6.8148e-01,
         -1.2999e-02],
        [-3.5850e+00, -2.1550e-02, -1.8711e+00, -9.9831e-01, -2.0469e+00,
         -5.5978e-02],
        [-1.4185e+00, -6.5646e-03, -4.8440e-01, -4.1281e-01, -2.2496e-01,
         -3.0619e-01],
        [-3.0529e+00, -2.4425e+00, -1.3889e+00, -3.4220e-01, -5.8015e-01,
         -1.5301e+00],
        [-2.0347e+00, -1.6427e+00, -2.9271e+00, -3.0516e+00, -1.8612e+00,
         -6.0940e-01],
        [-2.8110e+00, -2.1432e+00, -2.3672e+00, -2.3220e+00, -1.3363e+00,
         -1.7743e+00]], device='cuda:0')
tensor([-13.3180, -16.2788, -10.3964,  -4.0288, -11.1310, -10.2427,  -9.6550,
        -12.8659,  -2.9558, -10.1238,  -9.2475, -10.9478, -18.5254, -11.8266,
         -2.0862,  -8.5788,  -2.8534,  -9.3366, -12.1268, -12.7540],
       device='cuda:0')
tensor([-2.9838, -0.8336, -0.3950, -6.7477, -0.9245, -3.0151, -3.0211, -2.6239,
        -6.1196, -3.0167, -1.1734, -1.1898,  2.0287, -3.7308, -4.8554, -2.3242,
        -6.1877, -2.3924, -0.1589,  0.0163], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.1526, -1.4262, -1.9837, -0.9342, -1.3034, -1.1646],
        [-3.2348, -1.6001, -3.2739, -1.4155, -1.4866, -1.0189],
        [-4.4591, -1.1971, -1.9677, -1.1798, -2.4292, -1.6102],
        [-1.6440, -1.3127, -1.6033, -1.5512, -1.8999, -1.8516],
        [-7.1549, -2.4202, -1.8142, -1.2509, -2.1491, -1.4239],
        [-4.4470, -2.7578, -2.2903, -1.3390, -2.0768, -1.4844],
        [-1.2883, -1.2268, -2.1829, -1.9356, -1.3802, -1.2743],
        [-2.5744, -1.3978, -2.2874, -1.4694, -1.3229, -1.1321],
        [-2.6885, -1.4923, -2.3976, -1.2083, -1.3424, -1.0727],
        [-3.9088, -1.0636, -3.7026, -1.2408, -1.3525, -2.0248],
        [-1.9961, -1.3681, -1.1166, -1.1740, -2.5895, -2.8451],
        [-4.0405, -2.1071, -1.9075, -1.0289, -2.0514, -1.6615],
        [-4.0367, -2.9400, -2.3552, -1.8589, -2.2552, -1.4676],
        [-1.3190, -1.3876, -1.9596, -1.7329, -1.4707, -1.6095],
        [-5.2951, -1.3253, -3.1566, -1.2754, -1.4249, -1.8260],
        [-4.9538, -1.2983, -4.3327, -1.2377, -1.5448, -1.4602],
        [-2.1269, -1.3297, -1.2321, -1.4382, -1.7007, -1.7601],
        [-1.2528, -1.2795, -1.4092, -1.1412, -1.7214, -1.1311],
        [-4.9536, -3.1194, -2.0070, -1.0216, -2.1473, -1.5847],
        [-6.0704, -1.7742, -2.1572, -0.9774, -2.2118, -1.2601]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -7.9647, -12.0299, -12.8431,  -9.8627, -16.2131, -14.3953,  -9.2880,
        -10.1840, -10.2020, -13.2932, -11.0894, -12.7970, -14.9136,  -9.4793,
        -14.3033, -14.8275,  -9.5877,  -7.9352, -14.8336, -14.4512],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.9621e-01, -1.2297e+00, -1.7875e+00, -5.0115e-02, -2.2683e-01,
         -3.4823e-01],
        [-3.0739e+00, -3.3783e-01, -3.9619e+00, -3.2000e-01, -6.2494e-01,
         -2.2325e-01],
        [-3.9857e+00, -9.3407e-02, -1.9120e+00, -1.3951e-01, -3.3748e+00,
         -1.5163e+00],
        [-1.6420e+00, -3.3977e-02, -1.5716e+00, -8.3906e-01, -2.2202e+00,
         -1.2323e+00],
        [-4.8613e+00, -2.4450e+00, -1.5503e+00, -4.6793e-01, -2.6339e+00,
         -6.6435e-01],
        [-3.9189e+00, -2.7821e+00, -2.5018e+00, -2.6529e-01, -2.5628e+00,
         -8.3300e-01],
        [-1.1736e+00, -3.4106e-01, -2.3020e+00, -1.3593e+00, -7.6299e-02,
         -1.5675e-01],
        [-2.5992e+00, -1.3094e-01, -2.7884e+00, -4.5871e-01, -2.4981e-01,
         -1.9384e-01],
        [-2.3120e+00, -8.7305e-03, -2.9246e+00, -8.2562e-03, -3.8205e-01,
         -2.5084e-02],
        [-4.0657e+00, -4.2397e-01, -3.8162e+00, -2.3363e-01, -7.6974e-01,
         -2.3281e+00],
        [-2.2464e+00, -2.5564e-01, -8.4343e-02, -6.3704e-02, -3.2517e+00,
         -2.8657e+00],
        [-2.8361e+00, -1.7382e+00, -2.1926e+00, -1.6487e-01, -2.1797e+00,
         -1.1995e+00],
        [-2.7683e+00, -2.6236e+00, -2.4261e+00, -1.2097e+00, -2.5406e+00,
         -8.3912e-01],
        [-4.5465e-02, -9.2492e-01, -1.9512e+00, -9.4137e-01, -4.2121e-02,
         -1.1827e+00],
        [-5.4231e+00, -9.9815e-03, -3.8877e+00, -4.1747e-03, -6.6164e-01,
         -1.0658e+00],
        [-3.9883e+00, -4.6633e-01, -6.1314e+00, -3.5162e-01, -1.5921e+00,
         -1.1875e+00],
        [-1.9836e+00, -5.0705e-01, -6.6024e-03, -6.0290e-01, -1.8226e+00,
         -1.4326e+00],
        [-1.2463e+00, -2.3092e-03, -7.7956e-01, -7.6873e-02, -1.3567e+00,
         -6.4707e-04],
        [-4.0886e+00, -3.5958e+00, -1.9871e+00, -7.5514e-02, -2.6217e+00,
         -8.1152e-01],
        [-4.8300e+00, -1.1573e+00, -1.8815e+00, -2.4697e-03, -2.7353e+00,
         -4.1341e-02]], device='cuda:0')
tensor([ -3.9386,  -8.5418, -11.0216,  -7.5391, -12.6229, -12.8639,  -5.4089,
         -6.4209,  -5.6607, -11.6373,  -8.7675, -10.3110, -12.4074,  -5.0878,
        -11.0525, -13.7172,  -6.3554,  -3.4624, -13.1802, -10.6480],
       device='cuda:0')
tensor([-4.0261, -3.4880, -1.8215, -2.3236, -3.5903, -1.5313, -3.8791, -3.7632,
        -4.5413, -1.6560, -2.3219, -2.4860, -2.5062, -4.3915, -3.2508, -1.1103,
        -3.2323, -4.4727, -1.6535, -3.8032], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.1684, -1.2291, -2.2066, -1.2231, -1.6004, -1.3495],
        [-3.3354, -1.3358, -2.5793, -1.4504, -1.2543, -0.9436],
        [-3.0538, -1.4058, -2.5826, -1.1913, -1.4128, -1.2156],
        [-2.4733, -1.1159, -3.2374, -1.2430, -1.5535, -1.4702],
        [-3.2949, -1.5108, -1.0696, -1.0773, -2.9131, -2.1618],
        [-3.9604, -2.6966, -1.6322, -1.0659, -1.9856, -1.5760],
        [-3.5244, -7.0651, -2.7015, -1.7516, -4.2856, -1.4064],
        [-1.5242, -1.2019, -3.5754, -1.8675, -1.5711, -1.2421],
        [-4.1211, -1.3390, -2.6832, -1.2332, -2.1375, -1.0767],
        [-1.4114, -1.6648, -3.5585, -1.4965, -1.4731, -1.2409],
        [-1.0184, -1.5478, -1.2008, -2.8112, -1.6506, -1.2925],
        [-6.3512, -1.9348, -1.5653, -1.1436, -2.3685, -1.4908],
        [-2.9697, -3.1196, -1.9541, -1.9157, -1.9273, -1.3845],
        [-1.2564, -1.3630, -2.3792, -2.3150, -1.4682, -1.2680],
        [-3.3641, -1.5542, -1.8379, -1.3287, -1.3986, -1.0861],
        [-3.8924, -1.1581, -2.6846, -1.1356, -1.4245, -1.2718],
        [-2.9237, -1.4727, -1.4213, -1.3312, -1.5983, -2.2192],
        [-5.0917, -2.1221, -1.8094, -1.2539, -1.8717, -1.3752],
        [-4.8006, -2.6758, -1.9361, -1.6370, -2.0157, -1.3765],
        [-1.4131, -1.1913, -1.2971, -1.2162, -1.4854, -1.1938]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.7772, -10.8989, -10.8618, -11.0932, -12.0275, -12.9167, -20.7346,
        -10.9822, -12.5906, -10.8452,  -9.5212, -14.8542, -13.2710, -10.0498,
        -10.5696, -11.5670, -10.9664, -13.5240, -14.4418,  -7.7968],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-6.6198e-01, -1.7742e-01, -2.4007e+00, -2.1610e-01, -7.6177e-01,
         -3.1991e-01],
        [-3.5520e+00, -1.2069e-01, -3.3682e+00, -4.7286e-01, -6.9721e-02,
         -1.3578e-03],
        [-2.7188e+00, -1.1885e-04, -3.3586e+00, -1.2155e-03, -1.0778e+00,
         -1.8867e-01],
        [-2.3013e+00, -5.1963e-02, -2.9939e+00, -1.8846e-01, -1.3399e+00,
         -1.0090e+00],
        [-3.9637e+00, -4.9792e-01, -3.4345e-01, -2.7728e-04, -3.7259e+00,
         -1.7804e+00],
        [-2.7207e+00, -3.1032e+00, -9.3319e-01, -1.3197e-01, -2.1041e+00,
         -7.8220e-01],
        [-2.2837e+00, -6.2421e+00, -3.1334e+00, -1.0784e+00, -5.2742e+00,
         -8.1920e-01],
        [-1.9171e+00, -5.5030e-02, -4.3370e+00, -1.2504e+00, -6.5692e-01,
         -4.1980e-02],
        [-4.1735e+00, -1.1325e-05, -3.5157e+00, -6.4185e-03, -2.3520e+00,
         -1.4956e-01],
        [-1.1585e+00, -3.4623e-01, -4.3093e+00, -8.4182e-01, -9.4471e-01,
         -5.5906e-01],
        [-4.9980e-01, -5.1525e-01, -1.2942e-01, -3.1345e+00, -1.5769e+00,
         -6.1886e-01],
        [-4.4705e+00, -1.2156e+00, -7.9082e-01, -3.2346e-01, -3.1963e+00,
         -7.0398e-01],
        [-1.5886e+00, -3.5248e+00, -2.0386e+00, -9.7752e-01, -1.7959e+00,
         -8.7882e-01],
        [-9.1450e-02, -7.6106e-01, -2.7515e+00, -1.6008e+00, -5.8644e-02,
         -3.2587e-01],
        [-3.3079e+00, -3.1848e-01, -1.3848e+00, -1.2064e-01, -3.7436e-02,
         -3.9966e-01],
        [-3.5063e+00, -1.8104e-01, -3.7075e+00, -3.3208e-02, -1.2569e+00,
         -5.4592e-01],
        [-2.9945e+00, -8.8571e-02, -8.9122e-01, -3.7379e-01, -1.4559e+00,
         -1.6487e+00],
        [-3.3184e+00, -1.7053e+00, -1.9096e+00, -5.8758e-01, -1.6455e+00,
         -2.0489e-01],
        [-3.6754e+00, -2.6733e+00, -1.6694e+00, -6.7358e-01, -2.2493e+00,
         -3.3796e-01],
        [-1.6202e+00, -1.0620e-02, -3.4578e-01, -1.9530e-01, -3.4537e-01,
         -2.3091e-04]], device='cuda:0')
tensor([ -4.5379,  -7.5848,  -7.3452,  -7.8844, -10.3116,  -9.7753, -18.8310,
         -8.2584, -10.1972,  -8.1597,  -6.4747, -10.7007, -10.8043,  -5.5894,
         -5.5690,  -9.2309,  -7.4527,  -9.3713, -11.2789,  -2.5175],
       device='cuda:0')
tensor([-4.2393, -3.3141, -3.5166, -3.2088, -1.7159, -3.1413, -1.9036, -2.7238,
        -2.3934, -2.6855, -3.0465, -4.1535, -2.4667, -4.4604, -5.0007, -2.3362,
        -3.5137, -4.1527, -3.1629, -5.2793], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.0040, -3.0682, -2.6984, -0.9742, -1.7070, -1.3595],
        [-1.3599, -1.1064, -3.3570, -1.0544, -1.3709, -1.1507],
        [-2.4580, -1.4339, -2.8214, -1.3399, -1.5972, -1.8675],
        [-3.3189, -1.2257, -2.5037, -1.2113, -1.4129, -1.3753],
        [-3.6666, -1.2274, -3.3726, -1.1028, -1.4611, -3.0654],
        [-2.0966, -1.2307, -1.1320, -1.1233, -2.0430, -3.0496],
        [-5.2535, -2.1908, -1.8616, -1.1263, -2.2038, -1.4759],
        [-2.4534, -2.9064, -1.8811, -1.1156, -2.0097, -1.5698],
        [-1.3347, -1.2105, -2.6372, -1.4686, -1.4296, -1.3249],
        [-3.9677, -1.2990, -2.8277, -1.2741, -1.6085, -1.3541],
        [-8.3118, -1.3931, -2.3095, -1.1878, -1.6507, -1.1514],
        [-1.2730, -1.4284, -1.6855, -2.1714, -1.8981, -1.3225],
        [-5.3680, -1.8456, -1.7602, -1.1389, -2.2282, -1.3060],
        [-2.7031, -4.1090, -2.6039, -1.1397, -1.7965, -1.7571],
        [-1.2061, -1.2138, -1.4430, -1.0533, -1.4938, -1.2915],
        [-2.6464, -1.3787, -2.1800, -1.2628, -1.5931, -1.3439],
        [-3.8694, -1.3968, -3.4603, -1.1845, -1.5696, -1.4325],
        [-6.7154, -1.2436, -1.1622, -1.6011, -2.0912, -1.2839],
        [-4.5474, -3.1829, -2.8727, -1.2322, -2.0602, -1.3096],
        [-3.9667, -2.2325, -2.1591, -1.6456, -1.6122, -1.4360]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.8112,  -9.3992, -11.5179, -11.0477, -13.8958, -10.6752, -14.1118,
        -11.9361,  -9.4055, -12.3311, -16.0043,  -9.7790, -13.6468, -14.1093,
         -7.7013, -10.4048, -12.9131, -14.0973, -15.2051, -13.0522],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.9191e+00, -3.0608e+00, -3.5109e+00, -5.9247e-05, -1.2482e+00,
         -2.4042e-01],
        [-1.4218e+00, -1.8104e-01, -3.5416e+00, -2.5279e-01, -6.0036e-01,
         -9.6551e-02],
        [-2.5514e+00, -5.0108e-02, -3.9972e+00, -2.0931e-01, -1.2710e+00,
         -1.7994e+00],
        [-3.0710e+00, -3.9382e-03, -2.9311e+00, -1.5644e-01, -3.9567e-01,
         -1.9298e-02],
        [-3.7371e+00, -2.7318e-01, -3.5730e+00, -1.4353e-03, -1.0416e+00,
         -3.2196e+00],
        [-2.8101e+00, -1.1075e-02, -1.7304e-01, -3.2563e-02, -2.1743e+00,
         -1.8166e+00],
        [-4.1660e+00, -1.5526e+00, -1.8367e+00, -3.8160e-02, -2.8380e+00,
         -6.8626e-01],
        [-1.1032e+00, -2.6696e+00, -1.4895e+00, -1.8505e-01, -2.2728e+00,
         -1.0373e+00],
        [-1.3461e+00, -1.5333e-01, -2.9786e+00, -6.7725e-01, -1.2803e-02,
         -3.7629e-01],
        [-3.8190e+00, -1.7492e-02, -3.7789e+00, -4.5166e-02, -1.3028e+00,
         -6.6725e-01],
        [-5.0985e+00, -1.0020e-01, -2.4782e+00, -3.7809e-02, -1.8792e+00,
         -2.6034e-03],
        [-7.5070e-01, -2.3328e-01, -1.7175e+00, -1.9650e+00, -2.3688e+00,
         -5.6152e-01],
        [-3.7529e+00, -8.7430e-01, -1.6300e+00, -3.2643e-01, -2.8988e+00,
         -1.7920e-03],
        [-1.2905e+00, -3.9587e+00, -3.1595e+00, -1.2997e-01, -1.5700e+00,
         -1.6644e+00],
        [-4.9377e-01, -2.4678e-01, -8.3888e-01, -4.8108e-02, -3.9757e-01,
         -3.9739e-01],
        [-3.0374e+00, -3.5786e-02, -2.4609e+00, -3.3748e-03, -1.1740e+00,
         -5.5264e-01],
        [-3.0282e+00, -2.0504e-04, -4.5588e+00, -8.2138e-02, -1.6443e+00,
         -1.0474e+00],
        [-5.3130e+00, -4.2188e-02, -4.8466e-02, -9.3414e-01, -2.7479e+00,
         -1.8661e-01],
        [-3.5140e+00, -3.2615e+00, -4.1500e+00, -2.0674e-01, -2.3168e+00,
         -3.4039e-01],
        [-2.9104e+00, -1.8893e+00, -2.4241e+00, -4.9286e-01, -5.2496e-03,
         -3.2874e-02]], device='cuda:0')
tensor([-10.9794,  -6.0941,  -9.8785,  -6.5775, -11.8459,  -7.0176, -11.1177,
         -8.7573,  -5.5444,  -9.6306,  -9.5965,  -7.5968,  -9.4842, -11.7730,
         -2.4225,  -7.2642, -10.3610,  -9.2723, -13.7894,  -7.7548],
       device='cuda:0')
tensor([-2.8318, -3.3051, -1.6395, -4.4703, -2.0498, -3.6576, -2.9941, -3.1788,
        -3.8611, -2.7005, -6.4077, -2.1822, -4.1626, -2.3363, -5.2788, -3.1406,
        -2.5521, -4.8251, -1.4157, -5.2973], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.1311, -3.1332, -2.5549, -1.6938, -2.1871, -1.8407],
        [-1.1473, -1.3375, -3.0722, -2.1785, -1.3827, -1.4350],
        [-1.9589, -1.4461, -2.5141, -1.4713, -1.7341, -1.1258],
        [-1.2742, -1.2225, -3.3528, -1.4249, -1.4549, -1.3905],
        [-3.1779, -1.2730, -1.2860, -1.3976, -2.0390, -1.8353],
        [-2.9231, -2.1677, -3.1267, -1.0415, -1.5944, -2.0824],
        [-2.8328, -3.5418, -2.0639, -1.7871, -2.0157, -1.5184],
        [-1.5837, -2.5222, -2.0671, -4.4148, -1.2723, -1.2751],
        [-5.0981, -1.4081, -3.9444, -1.3674, -2.4357, -1.4389],
        [-3.9002, -1.2281, -3.3380, -1.2932, -1.8857, -1.2810],
        [-1.8117, -1.7922, -1.9857, -2.8788, -1.6727, -1.1169],
        [-5.9697, -2.6130, -1.9224, -1.2584, -2.1934, -1.7025],
        [-3.6963, -4.0990, -2.5883, -1.6036, -2.6577, -1.8076],
        [-1.2136, -1.1481, -2.0205, -1.0022, -1.5354, -1.7359],
        [-3.8602, -1.2942, -2.3354, -1.2737, -1.4134, -0.9354],
        [-3.8812, -1.4577, -2.2336, -1.4058, -1.5781, -1.2362],
        [-3.1012, -1.2180, -2.1439, -1.1905, -1.2903, -1.7064],
        [-1.5192, -1.5915, -2.2679, -1.7037, -1.9348, -3.1159],
        [-4.5574, -3.7451, -2.3039, -1.1016, -1.8128, -1.7360],
        [-3.3833, -3.9594, -3.6779, -1.1067, -2.6372, -2.8912]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-15.5408, -10.5532, -10.2501, -10.1198, -11.0087, -12.9359, -13.7598,
        -13.1351, -15.6926, -12.9262, -11.2580, -15.6593, -16.4524,  -8.6557,
        -11.1123, -11.7925, -10.6502, -12.1330, -15.2568, -17.6556],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.6329e+00, -3.3778e+00, -3.0180e+00, -7.4679e-01, -2.6599e+00,
         -1.7776e+00],
        [-4.9808e-01, -8.3502e-01, -3.6827e+00, -1.5975e+00, -3.4255e-01,
         -9.0059e-01],
        [-1.8830e+00, -4.9591e-02, -3.3682e+00, -4.5757e-01, -1.8667e+00,
         -5.1106e-01],
        [-8.9127e-01, -4.1623e-02, -4.3222e+00, -5.4633e-01, -8.1826e-01,
         -4.6360e-01],
        [-3.7263e+00, -1.1019e-01, -2.4435e-01, -4.9079e-01, -2.2650e+00,
         -1.4941e+00],
        [-2.0676e+00, -1.9803e+00, -4.2155e+00, -1.4147e-01, -9.9237e-01,
         -2.1200e+00],
        [-1.2797e+00, -5.1548e+00, -2.5057e+00, -7.7404e-01, -2.1290e+00,
         -6.4886e-02],
        [-3.9715e-01, -3.0941e+00, -1.9241e+00, -3.5570e+00, -3.1095e-01,
         -5.3404e-01],
        [-5.4042e+00, -2.5630e-02, -4.8915e+00, -3.8345e-03, -3.1976e+00,
         -9.6998e-01],
        [-3.1862e+00, -5.5200e-01, -4.1913e+00, -4.8240e-01, -2.4658e+00,
         -4.3909e-01],
        [-1.7347e+00, -7.2110e-01, -2.5948e+00, -2.7863e+00, -1.7099e+00,
         -9.4069e-03],
        [-4.4365e+00, -2.5156e+00, -2.0843e+00, -4.3753e-01, -2.7759e+00,
         -1.1169e+00],
        [-2.7046e+00, -4.4634e+00, -3.1346e+00, -4.6815e-01, -3.9339e+00,
         -1.8419e+00],
        [-4.7317e-01, -1.4539e-02, -1.9228e+00, -4.0505e-02, -9.5985e-01,
         -1.5536e+00],
        [-4.0779e+00, -2.8677e-03, -2.7510e+00, -4.3514e-02, -6.6160e-01,
         -4.2764e-02],
        [-3.1053e+00, -6.5223e-02, -2.4031e+00, -5.6867e-01, -1.6520e+00,
         -6.9084e-02],
        [-2.6084e+00, -3.8134e-03, -1.8138e+00, -3.4704e-02, -4.7320e-01,
         -1.7514e+00],
        [-1.8839e+00, -5.3458e-01, -2.6523e+00, -1.0827e+00, -1.8886e+00,
         -2.0465e+00],
        [-3.3853e+00, -4.5934e+00, -3.4476e+00, -1.3721e-02, -1.4164e+00,
         -1.3421e+00],
        [-2.0900e+00, -3.3298e+00, -4.2399e+00, -2.5214e-01, -2.8670e+00,
         -3.1559e+00]], device='cuda:0')
tensor([-15.2130,  -7.8564,  -8.1362,  -7.0832,  -8.3308, -11.5172, -11.9081,
         -9.8174, -14.4927, -11.3167,  -9.5561, -13.3668, -16.5465,  -4.9645,
         -7.5797,  -7.8633,  -6.6853, -10.0886, -14.1986, -15.9347],
       device='cuda:0')
tensor([-0.3277, -2.6967, -2.1139, -3.0366, -2.6779, -1.4187, -1.8517, -3.3178,
        -1.1998, -1.6095, -1.7019, -2.2924,  0.0941, -3.6912, -3.5326, -3.9292,
        -3.9649, -2.0444, -1.0582, -1.7209], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.4397, -1.9938, -1.8257, -1.6190, -2.6450, -1.4132],
        [-1.1969, -1.6780, -2.1324, -1.1729, -1.3495, -1.2060],
        [-1.7445, -1.3740, -4.1664, -1.3346, -1.4055, -0.9953],
        [-4.1764, -1.2593, -3.3882, -1.1161, -1.2707, -1.2504],
        [-1.8263, -1.2297, -1.9327, -1.3970, -2.3220, -1.3995],
        [-0.9244, -1.2333, -1.8256, -1.1529, -1.9597, -1.7112],
        [-2.8562, -3.0806, -2.1668, -1.0368, -2.0381, -1.5010],
        [-3.8073, -3.0818, -2.1064, -1.7478, -2.5815, -2.2294],
        [-1.1909, -1.3292, -1.2250, -1.0875, -1.6489, -1.1508],
        [-2.7405, -1.7874, -2.8062, -1.2695, -1.4803, -0.7773],
        [-2.9757, -1.1272, -4.7397, -1.1785, -1.9974, -1.3675],
        [-3.0763, -1.2585, -1.3697, -1.3582, -2.0913, -3.5112],
        [-5.9966, -1.9762, -3.1899, -1.2260, -1.7904, -1.1255],
        [-4.4255, -2.7877, -2.5883, -1.1840, -2.0934, -1.6146],
        [-1.7328, -1.2368, -1.2100, -1.2410, -1.6990, -1.2940],
        [-3.2611, -1.3963, -2.7823, -1.2429, -2.3116, -1.5172],
        [-7.9551, -1.3705, -2.5382, -1.1993, -1.7120, -1.2361],
        [-4.5379, -1.4274, -1.6063, -1.9629, -1.5701, -1.3842],
        [-2.9490, -2.6284, -2.3688, -1.4224, -2.7052, -1.9457],
        [-3.7328, -3.4674, -2.8635, -1.1836, -2.3624, -1.5479]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.9364,  -8.7356, -11.0201, -12.4610, -10.1071,  -8.8071, -12.6795,
        -15.5542,  -7.6323, -10.8610, -13.3860, -12.6651, -15.3046, -14.6934,
         -8.4137, -12.5114, -16.0112, -12.4889, -14.0194, -15.1576],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.2819e+00, -1.3723e+00, -1.4568e+00, -5.1968e-01, -3.8013e+00,
         -4.4571e-01],
        [-4.8722e-01, -1.8049e+00, -2.2217e+00, -2.0642e-01, -1.3194e-01,
         -1.6538e-01],
        [-1.9263e+00, -8.5644e-03, -5.4819e+00, -1.4456e-01, -5.3833e-01,
         -1.1711e-01],
        [-3.3345e+00, -1.2585e-01, -4.5372e+00, -3.2981e-02, -4.4671e-01,
         -4.5884e-01],
        [-1.7278e+00, -6.0029e-02, -1.4627e+00, -5.2500e-01, -3.0083e+00,
         -6.6975e-01],
        [-1.0664e-01, -4.0498e-02, -1.7998e+00, -9.7430e-02, -2.2214e+00,
         -1.5494e+00],
        [-1.5569e+00, -3.8086e+00, -3.0395e+00, -1.4743e-01, -2.1950e+00,
         -1.6251e-01],
        [-3.3950e+00, -2.9646e+00, -2.1112e+00, -8.6472e-01, -3.4793e+00,
         -2.6550e+00],
        [-7.9206e-01, -4.8401e-01, -1.4234e-04, -1.5162e-03, -1.0254e+00,
         -4.4891e-02],
        [-3.2685e+00, -8.1937e-01, -3.6747e+00, -1.4023e-01, -1.1756e+00,
         -5.6350e-03],
        [-2.7552e+00, -1.6767e-02, -5.8078e+00, -9.1998e-02, -2.6955e+00,
         -2.9403e-01],
        [-2.7913e+00, -1.6868e-01, -3.3726e-01, -3.9154e-01, -2.7833e+00,
         -3.6357e+00],
        [-4.3728e+00, -1.4438e+00, -4.3416e+00, -3.3152e-01, -1.4861e+00,
         -6.3731e-02],
        [-2.9880e+00, -3.2167e+00, -3.8006e+00, -1.0432e-01, -2.5248e+00,
         -5.4253e-01],
        [-2.5669e+00, -2.9102e-01, -2.2506e-01, -2.5002e-01, -1.0466e+00,
         -7.8422e-02],
        [-3.6922e+00, -1.3040e-01, -3.5381e+00, -6.6527e-02, -2.7002e+00,
         -8.2435e-01],
        [-4.9777e+00, -5.6162e-01, -2.8877e+00, -8.3591e-03, -2.0456e+00,
         -4.5145e-01],
        [-3.6963e+00, -2.9634e-01, -1.5865e+00, -1.6604e+00, -1.4546e+00,
         -6.4455e-01],
        [-1.7740e+00, -2.9300e+00, -3.9337e+00, -8.7353e-01, -4.1821e+00,
         -2.2516e+00],
        [-3.2171e+00, -3.8262e+00, -3.5510e+00, -9.6583e-02, -3.5115e+00,
         -9.3119e-01]], device='cuda:0')
tensor([ -9.8777,  -5.0176,  -8.2167,  -8.9361,  -7.4536,  -5.8151, -10.9100,
        -15.4699,  -2.3480,  -9.0840, -11.6612, -10.1078, -12.0395, -13.1770,
         -4.4580, -10.9517, -10.9325,  -9.3387, -15.9449, -15.1335],
       device='cuda:0')
tensor([-3.0587, -3.7180, -2.8034, -3.5249, -2.6535, -2.9920, -1.7695, -0.0844,
        -5.2843, -1.7770, -1.7248, -2.5573, -3.2651, -1.5164, -3.9557, -1.5597,
        -5.0788, -3.1502,  1.9255, -0.0241], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-0.9903, -1.5774, -2.2357, -1.2324, -1.3638, -1.7962],
        [-4.7909, -2.7951, -1.6310, -0.9377, -2.0240, -1.4952],
        [-3.3630, -3.9291, -1.8306, -2.0330, -1.6030, -1.6923],
        [-2.5620, -1.5210, -1.2048, -1.1436, -1.1851, -1.1750],
        [-4.8392, -1.6307, -2.9502, -1.3036, -1.4752, -1.0760],
        [-2.0629, -1.1579, -2.6408, -1.0902, -1.7506, -1.4883],
        [-2.9350, -1.3248, -1.1447, -1.3390, -2.2374, -1.1781],
        [-4.8325, -2.6153, -1.8341, -1.1886, -1.7276, -1.6917],
        [-3.0548, -3.4929, -2.7846, -1.9919, -2.0356, -1.3870],
        [-2.8529, -1.1873, -1.4665, -1.1967, -2.1216, -1.3794],
        [-2.6734, -1.2023, -2.9058, -1.1793, -3.3652, -2.2910],
        [-3.8955, -1.5650, -1.8721, -1.6704, -2.1161, -0.9772],
        [-4.4738, -1.5284, -2.0916, -2.1066, -1.4839, -1.5648],
        [-3.7765, -1.9809, -2.2816, -2.9212, -2.3397, -3.1153],
        [-4.5390, -5.3702, -2.3823, -1.0959, -1.6526, -1.4268],
        [-1.1886, -1.1311, -2.4156, -1.4570, -1.2290, -1.6271],
        [-1.6163, -1.4836, -3.2208, -1.5717, -1.2475, -1.0442],
        [-2.9403, -1.2632, -1.5651, -1.1976, -1.3378, -1.2228],
        [-3.6792, -1.6046, -1.2646, -1.3680, -1.3092, -1.7945],
        [-1.5445, -1.3880, -1.4605, -2.4642, -1.3896, -1.8686]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.1959, -13.6740, -14.4509,  -8.7916, -13.2748, -10.1908, -10.1592,
        -13.8898, -14.7467, -10.2043, -13.6170, -12.0963, -13.2492, -16.4151,
        -16.4669,  -9.0483, -10.1841,  -9.5268, -11.0201, -10.1155],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.8215e-03, -4.8180e-01, -2.8286e+00, -5.3719e-01, -4.1765e-01,
         -1.4812e+00],
        [-3.2774e+00, -3.0640e+00, -9.8104e-01, -2.4798e-03, -2.2987e+00,
         -2.4971e-02],
        [-2.7235e+00, -3.7755e+00, -1.4535e+00, -1.3762e+00, -1.2146e+00,
         -1.4882e+00],
        [-3.0093e+00, -1.4892e+00, -3.9965e-01, -2.6859e-02, -4.6737e-02,
         -2.6623e-01],
        [-5.0760e+00, -2.9038e-01, -4.2795e+00, -2.3853e-01, -1.4401e+00,
         -2.9913e-01],
        [-1.6991e+00, -5.7237e-01, -3.1994e+00, -9.2119e-03, -2.0010e+00,
         -1.0631e+00],
        [-2.6797e+00, -2.6307e-01, -3.3495e-01, -6.0358e-01, -2.9284e+00,
         -2.4991e-01],
        [-3.6782e+00, -2.7201e+00, -1.8937e+00, -1.7647e-01, -1.1112e+00,
         -1.5463e+00],
        [-2.1981e+00, -3.8513e+00, -3.3415e+00, -1.0064e+00, -2.4775e+00,
         -2.2415e-01],
        [-4.2499e+00, -3.7888e-03, -1.0352e+00, -6.5753e-01, -2.4435e+00,
         -6.5617e-01],
        [-3.1840e+00, -2.0731e-01, -3.4416e+00, -1.5762e-01, -4.4473e+00,
         -1.9922e+00],
        [-3.6223e+00, -4.9681e-02, -1.3961e+00, -1.0733e+00, -3.1030e+00,
         -6.4898e-02],
        [-3.3164e+00, -3.8072e-01, -2.6116e+00, -2.0822e+00, -1.0472e+00,
         -8.3978e-01],
        [-3.0378e+00, -1.4156e+00, -3.3212e+00, -3.6833e+00, -2.8612e+00,
         -4.2546e+00],
        [-2.6432e+00, -4.8155e+00, -1.9645e+00, -1.0191e-02, -8.6634e-01,
         -3.3240e-01],
        [-3.4753e-01, -2.1998e-01, -2.6167e+00, -8.7626e-01, -5.8156e-01,
         -1.4686e+00],
        [-1.4576e+00, -8.7860e-03, -4.7985e+00, -8.6582e-01, -2.8854e-02,
         -3.0947e-01],
        [-1.9624e+00, -3.3019e-02, -1.1338e+00, -9.5958e-02, -3.2130e-03,
         -5.7792e-02],
        [-2.0670e+00, -1.8583e+00, -7.5183e-01, -5.1204e-01, -1.2311e-01,
         -1.9605e+00],
        [-1.7521e+00, -6.2594e-01, -1.0484e+00, -1.9356e+00, -7.7128e-01,
         -1.4657e+00]], device='cuda:0')
tensor([ -5.7482,  -9.6486, -12.0315,  -5.2380, -11.6236,  -8.5441,  -7.0597,
        -11.1260, -13.0990,  -9.0461, -13.4300,  -9.3093, -10.2779, -18.5737,
        -10.6321,  -6.1107,  -7.4690,  -3.2862,  -7.2728,  -7.5990],
       device='cuda:0')
tensor([-3.4477, -4.0254, -2.4195, -3.5536, -1.6512, -1.6467, -3.0995, -2.7638,
        -1.6477, -1.1582, -0.1870, -2.7871, -2.9713,  2.1586, -5.8347, -2.9377,
        -2.7151, -6.2406, -3.7473, -2.5165], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.3626, -1.2675, -1.4281, -1.9720, -1.4468, -1.2809],
        [-3.1326, -1.2158, -1.5981, -1.3841, -1.4975, -1.6895],
        [-3.4952, -1.4734, -1.9919, -1.4011, -1.8700, -1.2888],
        [-1.6841, -1.3668, -1.8366, -2.3790, -1.6594, -1.1770],
        [-2.9905, -2.4681, -1.9999, -1.1146, -1.6342, -1.4728],
        [-4.0296, -2.9401, -1.8904, -1.2246, -1.9343, -1.2869],
        [-1.5273, -1.4430, -1.2645, -0.9278, -1.3007, -1.0290],
        [-1.8722, -1.5826, -2.3914, -1.2702, -1.5243, -1.7111],
        [-3.4427, -1.0951, -4.7995, -1.1534, -1.3243, -1.6857],
        [-1.6196, -1.4932, -1.4882, -1.7074, -1.9896, -1.6386],
        [-4.0882, -2.0212, -2.2577, -1.0830, -1.8469, -1.0890],
        [-2.0351, -2.8154, -2.2720, -1.6415, -2.2260, -1.3824],
        [-1.3501, -1.3021, -2.2750, -1.7326, -1.3954, -1.2109],
        [-2.6569, -1.3819, -3.0314, -1.2228, -1.8625, -0.9176],
        [-2.3166, -1.0473, -5.4694, -1.0500, -1.7825, -1.2808],
        [-1.3026, -1.3991, -1.2285, -1.4713, -2.0698, -3.1513],
        [-3.5371, -2.9486, -2.0498, -1.1044, -1.9016, -1.5784],
        [-3.4599, -3.1807, -2.2037, -1.7925, -2.0708, -1.4613],
        [-1.6240, -1.0827, -1.7621, -0.9159, -1.6944, -1.0356],
        [-2.7048, -1.5289, -2.2660, -1.2577, -2.8307, -3.0767]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.7578, -10.5176, -11.5204, -10.1028, -11.6801, -13.3059,  -7.4924,
        -10.3518, -13.5008,  -9.9366, -12.3859, -12.3724,  -9.2660, -11.0730,
        -12.9466, -10.6227, -13.1199, -14.1689,  -8.1146, -13.6648],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-6.2937e-02, -2.8537e-01, -7.0026e-01, -1.2771e+00, -9.8325e-02,
         -4.1299e-01],
        [-3.9581e+00, -5.5977e-01, -8.6468e-01, -4.5519e-01, -9.8123e-01,
         -9.9025e-01],
        [-4.4043e+00, -1.2226e-02, -1.8277e+00, -6.4374e-01, -2.3717e+00,
         -5.4540e-01],
        [-1.4671e+00, -3.8203e-01, -1.7216e+00, -2.2016e+00, -1.5743e+00,
         -4.3505e-01],
        [-2.6631e+00, -2.9510e+00, -2.5186e+00, -4.6620e-01, -1.0089e+00,
         -6.5029e-01],
        [-4.4774e+00, -3.1497e+00, -1.7393e+00, -8.0678e-02, -2.1018e+00,
         -5.6921e-01],
        [-1.6572e+00, -1.1168e+00, -2.6484e-01, -1.4023e-03, -3.7744e-01,
         -2.8526e-02],
        [-2.7892e+00, -3.1641e-01, -3.1653e+00, -1.6689e-05, -1.0344e+00,
         -5.4339e-01],
        [-3.6765e+00, -2.6566e-02, -6.2944e+00, -2.3138e-01, -9.1054e-01,
         -1.7842e+00],
        [-1.6072e+00, -4.3479e-01, -6.1180e-01, -1.3562e+00, -2.5111e+00,
         -1.2901e+00],
        [-3.8149e+00, -1.6394e+00, -2.6075e+00, -2.3301e-01, -1.6864e+00,
         -1.0674e-03],
        [-8.1615e-01, -4.0133e+00, -3.4841e+00, -3.8444e-01, -3.0183e+00,
         -2.0473e-01],
        [-5.5862e-01, -5.9072e-01, -2.6437e+00, -9.9628e-01, -1.0161e-02,
         -1.1903e-01],
        [-3.6406e+00, -2.0129e-01, -3.9473e+00, -1.1014e-02, -1.9080e+00,
         -2.6138e-01],
        [-2.5983e+00, -4.5812e-04, -6.7406e+00, -4.5784e-02, -2.0873e+00,
         -3.1240e-01],
        [-1.0461e+00, -1.8388e-01, -4.1687e-04, -7.2748e-01, -2.4253e+00,
         -2.8591e+00],
        [-3.4367e+00, -3.2809e+00, -2.4622e+00, -2.1232e-01, -2.0066e+00,
         -1.1228e+00],
        [-4.3078e+00, -3.9982e+00, -2.9202e+00, -5.1737e-01, -2.6005e+00,
         -8.1600e-01],
        [-1.9905e+00, -5.1326e-02, -1.3943e+00, -5.0603e-02, -1.7875e+00,
         -2.5906e-02],
        [-4.2350e+00, -3.0798e-01, -2.7826e+00, -3.2204e-03, -3.4582e+00,
         -1.8065e+00]], device='cuda:0')
tensor([ -2.8370,  -7.8092,  -9.8050,  -7.7817, -10.2582, -12.1181,  -3.4462,
         -7.8487, -12.9236,  -7.8112,  -9.9823, -11.9210,  -4.9185,  -9.9697,
        -11.7847,  -7.2424, -12.5216, -15.1600,  -5.3002, -12.5936],
       device='cuda:0')
tensor([-5.9209, -2.7084, -1.7154, -2.3212, -1.4219, -1.1879, -4.0462, -2.5031,
        -0.5771, -2.1254, -2.4036, -0.4514, -4.3475, -1.1033, -1.1618, -3.3803,
        -0.5983,  0.9910, -2.8144, -1.0712], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.9090, -1.3366, -3.4487, -1.2856, -1.2137, -2.0891],
        [-3.3997, -1.3503, -1.4503, -1.2120, -1.2732, -1.1319],
        [-2.7779, -2.7242, -1.5057, -1.7680, -1.4077, -1.1522],
        [-1.5273, -2.0640, -2.2314, -1.9461, -1.5410, -1.3843],
        [-3.0058, -1.6739, -3.1095, -3.8656, -1.8811, -3.9590],
        [-2.4415, -0.9846, -2.3708, -1.1914, -3.8179, -2.9389],
        [-5.8276, -2.6472, -1.9116, -1.3938, -1.7549, -0.9570],
        [-3.0969, -3.2842, -2.1939, -1.6400, -2.1273, -1.9603],
        [-1.3451, -2.4149, -2.4084, -1.1976, -1.3518, -1.1749],
        [-1.8659, -1.3080, -2.5578, -1.4723, -1.2521, -0.7167],
        [-1.7867, -1.3362, -1.8059, -1.1507, -1.9709, -1.2062],
        [-1.8572, -1.0344, -1.4932, -1.4011, -1.5833, -1.6504],
        [-2.0961, -1.3824, -1.3443, -1.2985, -2.7330, -1.8316],
        [-4.8253, -3.3130, -2.2782, -1.0711, -1.8836, -1.3460],
        [-2.6246, -2.1781, -2.1517, -1.1267, -2.0468, -1.3636],
        [-1.3551, -1.5362, -1.2548, -1.0594, -1.4103, -1.4623],
        [-3.0601, -1.2310, -1.9622, -1.3513, -1.7411, -1.5077],
        [-3.1908, -1.2863, -1.7821, -1.1946, -1.7648, -2.0350],
        [-3.5907, -1.0768, -2.2013, -2.3306, -2.4720, -1.0936],
        [-2.4746, -2.1988, -2.3716, -1.1993, -1.7160, -1.8435]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.2827,  -9.8174, -11.3356, -10.6942, -17.4950, -13.7452, -14.4921,
        -14.3025,  -9.8926,  -9.1729,  -9.2566,  -9.0196, -10.6859, -14.7172,
        -11.4914,  -8.0780, -10.8533, -11.2536, -12.7651, -11.8038],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.2691e+00, -6.0287e-02, -5.0324e+00, -1.6047e-01, -1.3908e-01,
         -2.1372e+00],
        [-3.2377e+00, -1.7177e-01, -6.0322e-01, -1.5270e-01, -8.7321e-04,
         -3.7718e-02],
        [-2.0510e+00, -3.7407e+00, -1.3724e+00, -1.2784e+00, -4.3783e-01,
         -7.0242e-01],
        [-1.7060e+00, -2.7028e+00, -2.3645e+00, -1.2666e+00, -1.3119e+00,
         -4.4000e-01],
        [-3.5865e+00, -1.0494e+00, -3.9833e+00, -4.3306e+00, -1.6423e+00,
         -3.9514e+00],
        [-3.6079e+00, -3.4754e-01, -2.5868e+00, -2.5355e-01, -4.1882e+00,
         -3.0723e+00],
        [-5.7365e+00, -2.4509e+00, -2.1196e+00, -3.2536e-01, -1.5903e+00,
         -4.1915e-01],
        [-2.9785e+00, -4.1111e+00, -2.6007e+00, -4.6471e-01, -2.8401e+00,
         -2.1764e+00],
        [-8.9231e-02, -2.9609e+00, -2.7369e+00, -2.1522e-01, -5.9605e-07,
         -1.2006e-01],
        [-2.5453e+00, -4.6018e-03, -2.9658e+00, -5.9531e-01, -3.9703e-02,
         -2.0366e-01],
        [-1.8352e+00, -1.6693e-01, -1.6344e+00, -2.3933e-02, -2.3942e+00,
         -6.6923e-01],
        [-2.0535e+00, -8.4905e-02, -6.4146e-01, -6.1812e-01, -1.3393e+00,
         -1.3586e+00],
        [-2.5221e+00, -4.1082e-01, -8.3883e-01, -5.6252e-01, -3.3540e+00,
         -1.3940e+00],
        [-4.9367e+00, -4.3260e+00, -3.2421e+00, -2.0777e-01, -2.0129e+00,
         -1.7028e-01],
        [-2.1468e+00, -1.7656e+00, -2.2989e+00, -1.7382e-01, -2.3223e+00,
         -8.0408e-01],
        [-8.8551e-01, -1.3075e+00, -2.1649e-01, -3.3343e-02, -2.2884e-01,
         -9.3672e-01],
        [-4.2824e+00, -2.5074e-01, -1.9517e+00, -3.2548e-01, -1.7203e+00,
         -6.7769e-01],
        [-3.8088e+00, -5.7559e-01, -1.2580e+00, -2.8903e-01, -2.2661e+00,
         -2.0169e+00],
        [-4.2458e+00, -5.1585e-02, -2.8332e+00, -2.2856e+00, -3.1106e+00,
         -3.4761e-03],
        [-2.2810e+00, -2.1159e+00, -3.9432e+00, -6.3294e-01, -1.2024e+00,
         -1.7901e+00]], device='cuda:0')
tensor([ -9.7986,  -4.2040,  -9.5827,  -9.7918, -18.5434, -14.0563, -12.6419,
        -15.1716,  -6.1223,  -6.3544,  -6.7238,  -6.0959,  -9.0822, -14.8959,
         -9.5115,  -3.6084,  -9.2083, -10.2144, -12.5302, -11.9655],
       device='cuda:0')
tensor([-1.4841, -5.6133, -1.7529, -0.9024,  1.0485,  0.3111, -1.8502,  0.8691,
        -3.7703, -2.8186, -2.5328, -2.9237, -1.6036,  0.1787, -1.9800, -4.4697,
        -1.6450, -1.0392, -0.2348,  0.1617], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.6686, -1.1295, -1.9783, -0.9053, -1.3673, -0.9469],
        [-1.4300, -1.6382, -2.3497, -1.1590, -2.1460, -0.9257],
        [-1.9568, -1.0542, -1.6212, -1.1929, -1.3090, -1.3026],
        [-2.4797, -2.6965, -1.3738, -1.4740, -1.2004, -0.9870],
        [-1.2861, -1.5861, -2.0236, -2.7575, -1.4530, -1.0523],
        [-5.6115, -1.4984, -1.5064, -2.7277, -3.9000, -1.3592],
        [-2.1920, -1.0031, -1.4568, -2.7769, -2.0650, -1.7956],
        [-3.8496, -2.3770, -2.4254, -1.5747, -2.1477, -3.7387],
        [-2.4394, -1.7508, -2.2778, -1.7954, -2.2998, -1.9032],
        [-1.2777, -1.8675, -2.0729, -0.9826, -1.2390, -1.1939],
        [-2.1214, -1.1807, -1.8915, -1.2202, -1.4150, -0.8258],
        [-1.5906, -1.2361, -1.4098, -2.1224, -2.1799, -1.2250],
        [-1.0953, -2.1281, -1.9553, -2.3977, -1.5407, -1.2230],
        [-4.9560, -2.3853, -2.2590, -1.6097, -1.7718, -1.2209],
        [-2.5156, -2.0526, -2.2648, -1.2681, -2.4323, -1.4808],
        [-1.1943, -3.1332, -1.8946, -1.5806, -1.3042, -1.1743],
        [-2.3083, -1.2920, -3.5235, -1.3150, -1.2738, -0.8265],
        [-0.9901, -1.2692, -1.5351, -1.8136, -1.1913, -0.9964],
        [-1.2234, -1.4183, -2.1403, -1.3550, -1.4455, -1.1919],
        [-1.3712, -1.7746, -2.2859, -1.6640, -1.2612, -1.0923]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -7.9958,  -9.6485,  -8.4366, -10.2115, -10.1586, -16.6032, -11.2894,
        -16.1131, -12.4663,  -8.6335,  -8.6546,  -9.7639, -10.3401, -14.2028,
        -12.0143, -10.2812, -10.5390,  -7.7956,  -8.7743,  -9.4493],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.0332e+00, -2.9022e-01, -1.8583e+00, -6.0101e-02, -1.0694e+00,
         -1.1312e-03],
        [-1.1980e+00, -1.7932e+00, -2.8458e+00, -2.1064e-04, -2.3299e+00,
         -7.9136e-02],
        [-1.3532e+00, -3.6069e-02, -1.2101e+00, -1.6197e-01, -1.1295e-02,
         -4.7767e-03],
        [-1.8720e+00, -3.1209e+00, -1.0582e+00, -8.8296e-01, -2.4177e-01,
         -9.5288e-02],
        [-1.0058e+00, -1.3936e+00, -2.0184e+00, -2.4431e+00, -9.2021e-01,
         -3.9400e-01],
        [-5.5780e+00, -7.5638e-01, -1.0600e+00, -2.7611e+00, -4.3777e+00,
         -8.0070e-01],
        [-2.5909e+00, -7.2326e-02, -8.6548e-01, -2.2719e+00, -1.9802e+00,
         -1.8583e+00],
        [-3.4652e+00, -1.8454e+00, -3.3945e+00, -1.1101e+00, -2.7269e+00,
         -3.0081e+00],
        [-1.7392e+00, -1.1800e+00, -2.1257e+00, -9.9638e-01, -3.1199e+00,
         -1.6962e+00],
        [-1.7238e-01, -2.0626e+00, -2.2296e+00, -1.1254e-01, -9.1172e-03,
         -5.9682e-01],
        [-2.5028e+00, -4.8682e-01, -1.5299e+00, -9.7047e-02, -8.6887e-01,
         -2.3168e-02],
        [-1.4167e+00, -3.9326e-02, -2.8527e-01, -1.9231e+00, -2.8370e+00,
         -6.0350e-01],
        [-6.3906e-02, -1.4771e+00, -2.0886e+00, -2.1408e+00, -1.2092e+00,
         -5.3290e-01],
        [-4.6145e+00, -2.7048e+00, -3.2048e+00, -1.5690e+00, -1.5161e+00,
         -1.3432e-02],
        [-1.8633e+00, -1.6872e+00, -2.6873e+00, -2.1317e-01, -3.1050e+00,
         -1.2119e+00],
        [-8.9858e-02, -4.0753e+00, -1.8446e+00, -7.7966e-01, -1.3910e-01,
         -5.0057e-01],
        [-2.7665e+00, -2.8465e-01, -4.3293e+00, -3.3315e-01, -2.8762e-01,
         -5.5857e-02],
        [-1.4557e-02, -4.0611e-02, -6.8745e-01, -1.6035e+00, -1.8796e-01,
         -3.9629e-02],
        [-1.2631e-01, -1.2779e+00, -2.1844e+00, -8.0851e-01, -7.1900e-01,
         -4.5269e-01],
        [-1.0975e+00, -1.6996e+00, -3.1186e+00, -8.2487e-01, -1.1677e-01,
         -8.5962e-04]], device='cuda:0')
tensor([ -5.3124,  -8.2462,  -2.7774,  -7.2712,  -8.1751, -15.3340,  -9.6391,
        -15.5503, -10.8573,  -5.1831,  -5.5087,  -7.1049,  -7.5125, -13.6226,
        -10.7680,  -7.4291,  -8.0571,  -2.5737,  -5.5688,  -6.8582],
       device='cuda:0')
tensor([-2.6834, -1.4023, -5.6592, -2.9404, -1.9835, -1.2692, -1.6503, -0.5628,
        -1.6090, -3.4504, -3.1460, -2.6590, -2.8276, -0.5802, -1.2463, -2.8521,
        -2.4819, -5.2219, -3.2055, -2.5910], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.9536, -1.1292, -3.4297, -1.1267, -1.5034, -1.4498],
        [-2.7513, -1.2898, -3.3421, -1.0664, -2.1871, -1.6443],
        [-3.6257, -1.2358, -1.2564, -1.1455, -2.3519, -1.7495],
        [-3.5811, -2.3347, -2.0119, -1.1463, -1.8654, -1.2766],
        [-3.1991, -3.2043, -2.7506, -1.1080, -2.1297, -1.3829],
        [-1.4427, -1.2949, -1.6185, -1.5366, -1.5335, -1.6336],
        [-1.8558, -1.5541, -2.4380, -1.2348, -1.4192, -1.6036],
        [-4.3467, -1.1238, -4.0631, -1.1090, -2.0883, -1.9411],
        [-2.2225, -1.1907, -1.2246, -2.1576, -1.4859, -1.6368],
        [-2.9694, -1.9541, -2.2033, -1.2163, -2.1173, -1.3853],
        [-2.8855, -3.0102, -2.4422, -1.9137, -2.1351, -1.4151],
        [-1.3611, -1.1858, -1.3080, -0.9856, -1.4524, -1.2457],
        [-2.3083, -1.3406, -3.0297, -1.2648, -2.1513, -2.7400],
        [-3.4856, -1.3424, -2.4162, -1.2342, -1.9066, -1.3539],
        [-2.8956, -1.2957, -2.2366, -3.4299, -1.9055, -1.8384],
        [-3.4964, -2.2394, -1.7486, -1.9157, -2.8074, -2.3080],
        [-4.8516, -2.3227, -3.4935, -0.9575, -1.7989, -2.0858],
        [-1.5137, -1.3948, -2.1841, -1.3723, -1.9687, -0.9560],
        [-3.7859, -1.3230, -2.7063, -1.2302, -1.4084, -1.8948],
        [-3.7120, -1.1416, -3.4185, -1.2241, -1.9641, -1.7102]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.5924, -12.2809, -11.3648, -12.2160, -13.7745,  -9.0598, -10.1056,
        -14.6720,  -9.9181, -11.8458, -13.8018,  -7.5386, -12.8346, -11.7389,
        -13.6017, -14.5156, -15.5101,  -9.3896, -12.3487, -13.1705],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.1958e+00, -2.5524e-03, -4.2748e+00, -1.1747e-02, -1.0711e+00,
         -1.1346e+00],
        [-4.0431e+00, -1.2190e+00, -4.0836e+00, -4.1325e-02, -2.3072e+00,
         -1.0931e+00],
        [-4.5028e+00, -3.4090e-01, -6.0050e-01, -1.9289e-02, -2.5697e+00,
         -1.0982e+00],
        [-3.7934e+00, -2.3429e+00, -2.4161e+00, -5.5743e-03, -1.9917e+00,
         -5.2178e-04],
        [-3.3781e+00, -3.3704e+00, -3.4450e+00, -1.3874e-01, -2.9391e+00,
         -6.9608e-01],
        [-1.5346e+00, -5.6886e-01, -1.3314e+00, -8.1446e-01, -5.7426e-01,
         -1.1725e+00],
        [-2.2782e+00, -3.6484e-01, -3.3235e+00, -1.7631e-04, -6.6199e-01,
         -7.9444e-01],
        [-4.4741e+00, -1.0271e-01, -5.5121e+00, -2.6075e-02, -2.7909e+00,
         -2.1505e+00],
        [-2.5500e+00, -1.6124e-02, -3.6775e-01, -2.1766e+00, -1.1623e+00,
         -1.2418e+00],
        [-2.7777e+00, -1.3302e+00, -3.1764e+00, -3.9071e-01, -2.7807e+00,
         -9.2237e-01],
        [-2.7102e+00, -3.5377e+00, -3.3588e+00, -7.4738e-01, -2.8761e+00,
         -6.9397e-01],
        [-1.0043e+00, -6.1582e-02, -2.2404e-01, -3.4906e-02, -6.6746e-01,
         -3.4718e-01],
        [-3.2512e+00, -1.5711e-01, -3.7866e+00, -1.8279e-01, -2.4369e+00,
         -2.1765e+00],
        [-4.1354e+00, -8.2518e-01, -2.9054e+00, -3.0253e-01, -2.4066e+00,
         -9.4086e-01],
        [-3.0132e+00, -1.0746e-03, -2.6847e+00, -3.5358e+00, -2.4755e+00,
         -1.5706e+00],
        [-4.2690e+00, -2.2869e+00, -1.4340e+00, -2.6004e+00, -4.2082e+00,
         -2.8206e+00],
        [-4.3453e+00, -1.9334e+00, -3.3141e+00, -2.5473e-02, -1.4487e+00,
         -2.1227e+00],
        [-1.4840e+00, -6.6985e-02, -2.5973e+00, -7.8420e-01, -2.1235e+00,
         -6.3410e-01],
        [-3.9616e+00, -1.9306e-01, -3.4965e+00, -1.0602e-01, -8.8887e-01,
         -1.8366e+00],
        [-2.7974e+00, -3.2829e-01, -4.3925e+00, -3.6042e-01, -1.3652e+00,
         -1.0712e+00]], device='cuda:0')
tensor([ -9.6906, -12.7873,  -9.1314, -10.5502, -13.9674,  -5.9961,  -7.4232,
        -15.0563,  -7.5145, -11.3781, -13.9241,  -2.3394, -11.9911, -11.5159,
        -13.2808, -17.6191, -13.1896,  -7.6901, -10.4826, -10.3150],
       device='cuda:0')
tensor([-1.9018,  0.5064, -2.2334, -1.6659,  0.1929, -3.0637, -2.6824,  0.3843,
        -2.4036, -0.4676,  0.1223, -5.1991, -0.8436, -0.2230, -0.3209,  3.1035,
        -2.3205, -1.6995, -1.8660, -2.8555], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.2068, -1.1935, -1.6875, -1.0762, -2.0950, -1.3590],
        [-2.1621, -1.2102, -1.9320, -1.1299, -1.8649, -2.0127],
        [-4.1571, -1.9112, -1.9015, -1.0858, -1.9234, -1.4109],
        [-2.3882, -3.7278, -1.8054, -1.2505, -3.8396, -1.4934],
        [-1.3215, -1.2114, -1.4468, -1.0647, -1.9395, -1.3929],
        [-2.1896, -1.2978, -2.5947, -1.3947, -2.2143, -2.0713],
        [-3.7491, -1.2732, -3.0190, -1.1628, -1.6836, -1.5538],
        [-2.7963, -1.1426, -1.1219, -2.9409, -1.5867, -1.3613],
        [-5.2242, -2.2878, -1.5931, -1.4846, -2.4906, -3.5119],
        [-2.5525, -3.6986, -2.1663, -1.2857, -2.6075, -1.3477],
        [-2.5179, -1.1813, -1.2359, -0.9696, -1.6697, -1.1486],
        [-3.0441, -1.5212, -4.2435, -1.8785, -2.4283, -0.8384],
        [-3.6286, -1.1637, -1.6507, -1.1248, -2.2598, -2.2930],
        [-4.3133, -1.3946, -1.3472, -2.0592, -1.8300, -0.9466],
        [-2.9261, -1.4718, -3.1052, -1.6567, -2.2538, -2.4257],
        [-3.0204, -2.7329, -1.9899, -1.2451, -2.0649, -1.9944],
        [-2.3877, -1.2061, -1.3023, -0.9333, -2.0830, -1.3793],
        [-2.7266, -1.3579, -2.8300, -1.2790, -1.2519, -1.5667],
        [-3.7367, -1.3640, -2.9722, -1.2044, -2.1945, -3.5960],
        [-3.6485, -1.2258, -2.1350, -1.1477, -2.0057, -1.6273]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.6180, -10.3118, -12.3899, -14.5049,  -8.3767, -11.7625, -12.4414,
        -10.9497, -16.5922, -13.6582,  -8.7230, -13.9540, -12.1205, -11.8908,
        -13.8392, -13.0477,  -9.2917, -11.0120, -15.0677, -11.7899],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.4035e+00, -9.4825e-02, -1.1068e+00, -1.7248e-02, -2.2967e+00,
         -4.7744e-01],
        [-2.4828e+00, -6.8721e-03, -1.9675e+00, -1.5064e-01, -1.9148e+00,
         -2.0488e+00],
        [-4.6873e+00, -1.1550e+00, -1.9468e+00, -3.9302e-02, -2.1215e+00,
         -2.8431e-01],
        [-1.7040e+00, -3.5981e+00, -1.3949e+00, -3.7987e-01, -5.0259e+00,
         -1.0630e+00],
        [-5.8787e-02, -1.2252e-01, -8.4296e-01, -3.8945e-02, -2.1493e+00,
         -7.8958e-01],
        [-3.0600e+00, -4.0531e-06, -3.3523e+00, -4.0519e-01, -2.4114e+00,
         -1.1637e+00],
        [-4.2530e+00, -2.5449e-03, -4.0103e+00, -8.1935e-02, -1.9596e+00,
         -1.2995e+00],
        [-3.2544e+00, -1.2802e-02, -1.3019e-01, -3.0601e+00, -1.4404e+00,
         -7.1521e-01],
        [-5.0470e+00, -2.3307e+00, -8.4223e-01, -1.2139e+00, -3.5828e+00,
         -4.4937e+00],
        [-2.1599e+00, -4.5759e+00, -2.6650e+00, -9.3941e-02, -4.1995e+00,
         -4.1448e-01],
        [-4.2447e+00, -1.6051e-01, -9.5795e-03, -1.1058e-01, -1.5328e+00,
         -1.8361e-02],
        [-4.4865e+00, -4.2853e-01, -5.6790e+00, -1.4639e+00, -2.7415e+00,
         -2.4278e-02],
        [-3.8435e+00, -1.0923e-01, -1.0318e+00, -6.9861e-03, -3.1412e+00,
         -2.7745e+00],
        [-4.9076e+00, -1.4114e-02, -1.0356e-01, -1.9508e+00, -2.2595e+00,
         -3.0526e-01],
        [-2.8389e+00, -6.0147e-01, -3.9045e+00, -1.4937e+00, -2.6941e+00,
         -2.3521e+00],
        [-2.9979e+00, -3.2939e+00, -2.0044e+00, -5.7583e-02, -2.6106e+00,
         -2.4384e+00],
        [-3.8313e+00, -6.8477e-01, -3.0698e-01, -3.8795e-02, -2.4787e+00,
         -7.2206e-01],
        [-4.0655e+00, -6.0151e-02, -3.7752e+00, -1.4944e-01, -3.1262e-01,
         -6.3467e-01],
        [-4.3498e+00, -3.6574e-02, -4.2730e+00, -7.6002e-02, -3.0544e+00,
         -3.4697e+00],
        [-4.4502e+00, -9.8892e-01, -1.6274e+00, -7.6973e-04, -2.4646e+00,
         -1.1986e+00]], device='cuda:0')
tensor([ -8.3964,  -8.5713, -10.2343, -13.1658,  -4.0021, -10.3925, -11.6068,
         -8.6130, -17.5103, -14.1087,  -6.0765, -14.8238, -10.9072,  -9.5409,
        -13.8848, -13.4027,  -8.0627,  -8.9976, -15.2595, -10.7305],
       device='cuda:0')
tensor([-2.2216, -1.7405, -2.1555, -1.3390, -4.3746, -1.3700, -0.8345, -2.3367,
         0.9181,  0.4505, -2.6464,  0.8698, -1.2133, -2.3500,  0.0456,  0.3550,
        -1.2290, -2.0145,  0.1918, -1.0594], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.1981, -1.3092, -2.3539, -1.2113, -2.0694, -0.9102],
        [-4.5671, -1.2498, -1.1720, -1.3234, -2.1273, -2.0673],
        [-1.1109, -1.4407, -2.3892, -1.4149, -1.4271, -1.1903],
        [-2.9495, -2.2339, -2.5646, -1.0163, -2.0310, -1.3152],
        [-3.1218, -2.4442, -2.5705, -1.2173, -1.4196, -1.3695],
        [-1.3235, -2.1901, -2.4525, -1.0999, -1.2434, -1.0223],
        [-1.4113, -1.4085, -2.9165, -1.3485, -1.6402, -0.8909],
        [-2.9322, -0.9959, -2.5066, -1.9233, -1.2974, -1.2355],
        [-2.0497, -1.1480, -1.4000, -1.5555, -2.0240, -1.5353],
        [-2.3169, -2.2537, -1.7666, -1.1230, -1.9677, -1.5910],
        [-2.3235, -2.9066, -1.6414, -1.6426, -2.1988, -1.3652],
        [-1.4764, -1.8731, -1.4551, -1.6898, -1.5442, -1.0345],
        [-2.1174, -1.3457, -5.0987, -1.3073, -1.9401, -0.9458],
        [-4.1291, -1.2180, -2.0815, -1.1487, -2.4820, -1.7993],
        [-1.8150, -1.3155, -2.5832, -1.9225, -2.0275, -1.9418],
        [-2.1569, -2.8248, -2.3828, -1.5168, -2.0444, -2.8187],
        [-2.9772, -1.8893, -2.5225, -0.9648, -1.7423, -1.2209],
        [-1.3085, -1.4172, -2.4879, -0.9925, -1.1779, -1.0173],
        [-2.5225, -1.5345, -2.8541, -1.4267, -1.4412, -0.9525],
        [-2.1737, -1.1466, -2.7712, -1.1823, -1.6597, -1.5766]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.0522, -12.5068,  -8.9731, -12.1105, -12.1430,  -9.3316,  -9.6159,
        -10.8909,  -9.7125, -11.0189, -12.0781,  -9.0731, -12.7550, -12.8587,
        -11.6055, -13.7444, -11.3171,  -8.4014, -10.7315, -10.5101],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.6212e+00, -6.3192e-03, -2.7468e+00, -1.8296e-01, -2.8881e+00,
         -7.5158e-02],
        [-4.4555e+00, -6.1760e-02, -5.5667e-03, -4.5960e-01, -2.8695e+00,
         -2.3525e+00],
        [-4.5205e-01, -3.0811e-01, -2.9872e+00, -9.8071e-01, -5.4007e-01,
         -2.1447e-01],
        [-2.6250e+00, -2.1255e+00, -3.7804e+00, -1.6619e-02, -2.4296e+00,
         -9.3293e-02],
        [-2.4628e+00, -2.1768e+00, -2.4457e+00, -2.3072e-01, -2.1463e-01,
         -5.3351e-01],
        [-9.7238e-01, -2.6386e+00, -2.7261e+00, -2.7265e-01, -7.4435e-02,
         -6.9034e-02],
        [-1.2948e+00, -2.6932e-03, -4.3485e+00, -2.1086e-01, -1.5543e+00,
         -4.7600e-04],
        [-3.4622e+00, -6.1182e-02, -3.0417e+00, -1.5166e+00, -9.2599e-01,
         -1.8543e-01],
        [-2.4877e+00, -1.0220e-01, -9.4748e-01, -9.3871e-01, -2.2252e+00,
         -1.1614e+00],
        [-1.6993e+00, -2.3090e+00, -5.9836e-01, -2.6146e-01, -2.2789e+00,
         -8.0370e-01],
        [-1.4951e+00, -3.0353e+00, -8.1427e-01, -7.7454e-01, -2.7127e+00,
         -7.2335e-01],
        [-1.5310e-01, -2.1086e+00, -8.4349e-01, -8.6636e-01, -6.5419e-01,
         -1.7027e-03],
        [-2.7115e+00, -6.4768e-02, -5.4221e+00, -9.0325e-03, -2.5761e+00,
         -2.0484e-01],
        [-4.6468e+00, -4.5469e-01, -2.1760e+00, -1.6985e-01, -3.2716e+00,
         -1.7441e+00],
        [-2.3071e+00, -1.0643e-01, -3.5450e+00, -1.5142e+00, -2.2344e+00,
         -1.1843e+00],
        [-1.5049e+00, -3.6695e+00, -3.9727e+00, -1.4320e+00, -2.6079e+00,
         -3.9451e+00],
        [-2.1595e+00, -1.3863e+00, -2.2064e+00, -3.2084e-03, -1.2996e+00,
         -7.8289e-02],
        [-6.5615e-01, -1.1613e+00, -2.7481e+00, -1.3942e-01, -2.6968e-01,
         -1.6022e-01],
        [-3.1566e+00, -2.6029e-01, -4.0887e+00, -5.1011e-01, -1.0830e+00,
         -2.0153e-01],
        [-2.6000e+00, -3.2160e-02, -3.5195e+00, -1.5228e-01, -1.7197e+00,
         -1.4107e+00]], device='cuda:0')
tensor([ -8.5206, -10.2045,  -5.4826, -11.0705,  -8.0642,  -6.7531,  -7.4117,
         -9.1931,  -7.8628,  -7.9508,  -9.5553,  -4.6275, -10.9883, -12.4629,
        -10.8915, -17.1321,  -7.1332,  -5.1348,  -9.3001,  -9.4343],
       device='cuda:0')
tensor([-1.5316, -2.3023, -3.4906, -1.0401, -4.0788, -2.5785, -2.2043, -1.6978,
        -1.8497, -3.0681, -2.5228, -4.4456, -1.7667, -0.3958, -0.7140,  3.3876,
        -4.1839, -3.2665, -1.4314, -1.0758], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.4781, -1.5501, -2.0403, -1.6841, -1.2221, -0.8325],
        [-2.7347, -1.9085, -1.7108, -5.5503, -2.6429, -2.7263],
        [-2.5279, -0.7922, -1.7512, -1.0025, -3.9271, -2.1728],
        [-2.7419, -1.7546, -1.7868, -1.3312, -1.9975, -1.6972],
        [-3.1786, -3.2036, -2.2211, -1.5105, -2.1174, -1.6584],
        [-1.5632, -1.1342, -1.3008, -0.9642, -1.1554, -1.3905],
        [-3.4973, -1.0716, -5.5420, -1.4515, -2.0803, -5.2326],
        [-1.0925, -2.2403, -1.4870, -1.9575, -2.0302, -0.8143],
        [-7.8215, -3.0095, -1.6637, -1.7029, -2.1693, -1.4617],
        [-3.3370, -2.3278, -1.7731, -1.1463, -1.4617, -1.1445],
        [-1.4252, -1.7454, -1.8275, -2.8195, -1.1227, -1.2526],
        [-3.3831, -1.8849, -2.0131, -1.3817, -1.5920, -1.2086],
        [-5.8940, -1.4741, -1.4455, -1.3976, -1.7407, -1.7688],
        [-1.7075, -1.1025, -1.2601, -1.0114, -1.2721, -1.3099],
        [-1.1666, -2.2697, -1.0107, -1.4256, -1.4444, -1.0143],
        [-3.0625, -3.3826, -2.2287, -1.8121, -1.6082, -1.0615],
        [-1.2088, -1.5505, -1.6614, -2.1050, -1.2917, -1.1355],
        [-2.7274, -1.4620, -2.3191, -2.3139, -1.5404, -0.9995],
        [-4.4627, -1.0019, -1.2964, -0.9972, -3.1966, -3.7110],
        [-1.2114, -1.1204, -1.3429, -1.1691, -1.5572, -1.2512]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.8072, -17.2735, -12.1737, -11.3093, -13.8896,  -7.5083, -18.8754,
         -9.6218, -17.8287, -11.1905, -10.1928, -11.4635, -13.7207,  -7.6635,
         -8.3312, -13.1556,  -8.9527, -11.3623, -14.6657,  -7.6522],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.3916e+00, -1.0360e+00, -2.1595e+00, -8.4726e-01, -8.7677e-02,
         -2.9434e-01],
        [-3.7756e+00, -1.4621e+00, -1.4957e+00, -5.0329e+00, -2.6031e+00,
         -2.0054e+00],
        [-3.7219e+00, -3.1089e-02, -1.4535e+00, -2.1526e-02, -5.1242e+00,
         -2.7228e+00],
        [-2.3469e+00, -7.4941e-01, -1.6507e+00, -2.4306e-01, -2.3157e+00,
         -1.5150e+00],
        [-3.6076e+00, -3.8427e+00, -2.9257e+00, -2.3188e-01, -2.6452e+00,
         -1.4259e+00],
        [-1.8742e+00, -2.4751e-02, -6.6306e-01, -1.6423e-01, -1.2109e-01,
         -7.6119e-01],
        [-4.6974e+00, -4.1134e-01, -6.1000e+00, -7.3723e-01, -2.3415e+00,
         -4.2244e+00],
        [-2.0459e-03, -2.0755e+00, -4.6888e-01, -1.7891e+00, -2.6908e+00,
         -2.3186e-04],
        [-6.1262e+00, -3.4567e+00, -1.4990e+00, -1.4414e+00, -2.3191e+00,
         -7.2566e-01],
        [-2.5227e+00, -1.9891e+00, -1.6981e+00, -1.0650e-02, -2.1535e-01,
         -5.8079e-04],
        [-1.0897e-01, -1.9077e+00, -1.5520e+00, -2.3949e+00, -6.7322e-02,
         -7.5784e-01],
        [-4.1704e+00, -2.8939e+00, -1.9059e+00, -5.4186e-01, -1.5554e+00,
         -3.0435e-01],
        [-3.8727e+00, -2.5322e-01, -1.0563e+00, -7.4036e-01, -9.5285e-01,
         -1.3367e+00],
        [-1.0080e+00, -2.2859e-01, -2.1144e-01, -5.8610e-02, -3.2922e-01,
         -4.2568e-01],
        [-7.4039e-02, -2.6816e+00, -4.8953e-03, -8.1508e-01, -9.4164e-01,
         -3.8397e-04],
        [-2.9321e+00, -3.7615e+00, -2.3456e+00, -1.5473e+00, -1.1580e+00,
         -5.9111e-03],
        [-6.9647e-01, -1.4922e+00, -1.6991e+00, -1.8921e+00, -3.5012e-04,
         -2.3249e-01],
        [-3.8057e+00, -2.5563e-01, -3.0303e+00, -2.5573e+00, -8.8402e-01,
         -3.7667e-02],
        [-5.3282e+00, -3.6557e-01, -2.7464e-01, -2.4422e-02, -3.5491e+00,
         -4.5128e+00],
        [-9.0963e-01, -1.7781e-01, -4.6818e-01, -1.0531e-02, -4.6011e-01,
         -7.9870e-01]], device='cuda:0')
tensor([ -5.8164, -16.3748, -13.0751,  -8.8209, -14.6790,  -3.6085, -18.5118,
         -7.0266, -15.5680,  -6.4365,  -6.7887, -11.3718,  -8.2121,  -2.2616,
         -4.5176, -11.7504,  -6.0127, -10.5705, -14.0547,  -2.8250],
       device='cuda:0')
tensor([-2.9908, -0.8986,  0.9015, -2.4885,  0.7894, -3.8998, -0.3636, -2.5953,
        -2.2606, -4.7539, -3.4041, -0.0917, -5.5085, -5.4019, -3.8136, -1.4052,
        -2.9400, -0.7918, -0.6110, -4.8272], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.9244, -0.9498, -1.3460, -1.0277, -2.5879, -1.4146],
        [-1.5015, -1.2662, -1.3381, -1.2847, -1.4292, -0.9645],
        [-3.0118, -2.8839, -2.4253, -1.0740, -2.2055, -1.5264],
        [-1.5779, -1.1341, -1.3651, -0.8743, -1.4968, -1.1547],
        [-2.6392, -1.5147, -2.0886, -1.1625, -1.5736, -1.0148],
        [-1.3301, -1.8276, -1.9366, -1.3534, -1.6104, -3.9913],
        [-3.7341, -1.0835, -2.4669, -1.3434, -1.9642, -1.3613],
        [-1.7720, -1.1664, -1.1633, -1.2181, -1.8793, -1.1625],
        [-3.6329, -1.9856, -2.4042, -1.1957, -2.3236, -1.3266],
        [-2.5708, -2.4541, -1.9516, -1.7892, -1.8559, -1.2826],
        [-1.6962, -1.1410, -1.2446, -1.3188, -1.5457, -1.1606],
        [-2.0057, -1.2227, -2.7902, -1.1631, -1.7154, -0.9570],
        [-2.2185, -1.3156, -2.4849, -1.1964, -1.3797, -1.4206],
        [-1.8354, -1.1264, -2.1961, -1.1510, -1.9113, -2.6698],
        [-2.5753, -1.1084, -1.2897, -1.1756, -2.9107, -2.6586],
        [-2.7662, -2.6424, -1.8928, -1.0841, -2.1102, -1.4984],
        [-3.4734, -3.1127, -2.1237, -1.2759, -1.9093, -1.8938],
        [-1.7451, -1.2609, -1.2335, -1.0710, -1.4297, -1.2330],
        [-1.5969, -1.5561, -1.5151, -1.0920, -2.4263, -3.6057],
        [-2.0737, -1.5090, -2.3135, -1.3358, -2.1453, -0.7252]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.2504,  -7.7843, -13.1269,  -7.6029,  -9.9934, -12.0494, -11.9535,
         -8.3615, -12.8685, -11.9043,  -8.1068,  -9.8541, -10.0156, -10.8900,
        -11.7182, -11.9942, -13.7887,  -7.9732, -11.7920, -10.1025],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.5047e+00, -4.7310e-01, -2.9043e-01, -2.3319e-02, -2.8288e+00,
         -1.1000e+00],
        [-3.7996e-01, -9.7010e-02, -5.3951e-01, -3.0791e-01, -2.5869e-01,
         -4.3651e-02],
        [-2.8962e+00, -3.3123e+00, -3.0449e+00, -4.3484e-03, -2.9218e+00,
         -9.2866e-01],
        [-2.3128e-01, -4.9795e-03, -5.8598e-01, -1.5378e-05, -5.9480e-01,
         -3.6500e-02],
        [-3.4878e+00, -1.2031e+00, -2.0622e+00, -1.8414e-01, -9.3232e-01,
         -8.2494e-02],
        [-1.5027e+00, -9.0861e-01, -1.5313e+00, -2.0441e-01, -9.8968e-01,
         -2.8989e+00],
        [-4.8385e+00, -3.6384e-02, -2.8340e+00, -5.8557e-01, -1.9307e+00,
         -4.9993e-01],
        [-1.5955e+00, -4.9498e-02, -2.1131e-01, -9.6921e-02, -2.1258e+00,
         -3.9095e-01],
        [-3.6151e+00, -1.4309e+00, -3.5962e+00, -2.0337e-01, -3.1759e+00,
         -7.6636e-01],
        [-2.0156e+00, -2.4806e+00, -1.8205e+00, -7.3985e-01, -1.9660e+00,
         -1.0631e-01],
        [-2.2011e+00, -9.8062e-03, -2.5738e-01, -5.7060e-01, -1.2313e+00,
         -1.8746e-01],
        [-2.2885e+00, -1.5637e-02, -3.6596e+00, -1.7187e-02, -1.4085e+00,
         -1.9801e-04],
        [-2.4202e+00, -1.7218e-02, -2.7325e+00, -2.0268e-03, -2.2786e-01,
         -1.0984e+00],
        [-2.2458e+00, -4.5639e-01, -2.2525e+00, -1.3614e-01, -1.9118e+00,
         -2.4956e+00],
        [-3.0248e+00, -9.7513e-05, -7.1898e-01, -4.2016e-02, -3.5196e+00,
         -2.5600e+00],
        [-2.4958e+00, -3.1686e+00, -1.9217e+00, -1.7939e-01, -2.7871e+00,
         -6.0014e-01],
        [-3.3337e+00, -3.0598e+00, -2.1194e+00, -4.4020e-01, -2.2498e+00,
         -1.8698e+00],
        [-2.4222e+00, -2.3631e-02, -1.1003e-04, -1.0324e-02, -9.9589e-02,
         -3.6130e-01],
        [-1.6137e+00, -1.6871e+00, -9.4563e-01, -1.7527e-03, -2.8279e+00,
         -3.1462e+00],
        [-2.8545e+00, -2.8883e-01, -2.5731e+00, -3.8311e-01, -3.2636e+00,
         -4.2484e-02]], device='cuda:0')
tensor([ -7.2204,  -1.6267, -13.1082,  -1.4536,  -7.9521,  -8.0356, -10.7251,
         -4.4700, -12.7878,  -9.1287,  -4.4577,  -7.3896,  -6.4982,  -9.4982,
         -9.8655, -11.1528, -13.0726,  -2.9171, -10.2223,  -9.4056],
       device='cuda:0')
tensor([-2.0300, -6.1575, -0.0186, -6.1494, -2.0413, -4.0138, -1.2284, -3.8915,
        -0.0807, -2.7756, -3.6491, -2.4645, -3.5174, -1.3918, -1.8527, -0.8415,
        -0.7161, -5.0561, -1.5698, -0.6969], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.3578, -1.3214, -2.5929, -1.1914, -1.3207, -1.2855],
        [-2.9256, -1.2732, -2.0064, -1.1034, -1.9183, -4.6420],
        [-2.2158, -1.1436, -1.7720, -1.1823, -1.3422, -2.6988],
        [-3.2403, -2.5841, -1.6800, -1.1322, -1.7830, -1.2805],
        [-3.0259, -2.1012, -1.9824, -1.2593, -1.5289, -2.1749],
        [-1.3071, -1.2936, -2.4224, -1.0949, -1.4791, -1.9708],
        [-2.7146, -1.4274, -3.5870, -1.3250, -1.5964, -0.8290],
        [-3.6466, -1.0455, -4.4943, -1.1106, -1.5139, -1.3859],
        [-2.6818, -1.0703, -1.2929, -1.3590, -1.6461, -1.7088],
        [-3.4308, -1.2993, -3.5224, -1.1522, -1.8528, -1.1225],
        [-2.9404, -2.4707, -2.0763, -1.6076, -1.8167, -1.3759],
        [-1.3991, -1.1388, -1.7421, -0.9002, -1.9386, -1.0437],
        [-3.1575, -1.2452, -2.0425, -1.2010, -1.7067, -2.0146],
        [-5.0641, -1.6991, -1.5529, -1.1606, -1.4807, -1.0442],
        [-3.1847, -1.1346, -3.1639, -1.2597, -1.7769, -2.9659],
        [-2.6368, -1.1694, -1.4776, -1.4345, -2.5968, -3.1365],
        [-3.2824, -3.5967, -1.9734, -1.1248, -2.2265, -1.4438],
        [-3.1451, -2.7094, -2.3529, -1.6132, -1.9415, -1.9699],
        [-1.3298, -1.0582, -1.2911, -0.8539, -1.2401, -1.0326],
        [-3.6338, -1.2981, -2.7704, -1.4662, -2.2566, -2.1653]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.0697, -13.8690, -10.3546, -11.7000, -12.0726,  -9.5681, -11.4793,
        -13.1967,  -9.7590, -12.3799, -12.2876,  -8.1626, -11.3675, -12.0016,
        -13.4857, -12.4516, -13.6476, -13.7319,  -6.8058, -13.5904],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.8131e+00, -2.9707e-01, -3.3082e+00, -1.9648e-03, -3.2131e-01,
         -7.5632e-01],
        [-3.4567e+00, -5.0101e-01, -1.7052e+00, -1.1020e-01, -2.2930e+00,
         -4.1144e+00],
        [-2.5527e+00, -3.1216e-02, -1.7216e+00, -4.4756e-01, -7.3826e-01,
         -2.5974e+00],
        [-3.4703e+00, -2.3934e+00, -6.8273e-01, -5.5298e-03, -1.4930e+00,
         -5.1844e-04],
        [-2.9779e+00, -1.7319e+00, -1.8701e+00, -2.5740e-01, -1.0372e+00,
         -2.4861e+00],
        [-6.3200e-01, -5.0725e-01, -2.8376e+00, -1.3751e-01, -5.4374e-01,
         -1.8799e+00],
        [-4.2918e+00, -2.2601e-02, -5.3956e+00, -9.4054e-02, -1.2060e+00,
         -1.1262e-02],
        [-3.9996e+00, -7.2643e-02, -5.7259e+00, -8.5757e-02, -1.3330e+00,
         -9.3283e-01],
        [-3.1003e+00, -1.5257e-02, -2.1365e-01, -5.1231e-01, -1.6270e+00,
         -1.5105e+00],
        [-3.1054e+00, -6.3946e-03, -4.6753e+00, -3.7848e-01, -1.7672e+00,
         -5.8973e-02],
        [-2.7750e+00, -2.9678e+00, -2.3923e+00, -3.5386e-01, -1.3005e+00,
         -2.9376e-01],
        [-8.6361e-01, -6.0664e-02, -1.5066e+00, -2.3109e-03, -2.3172e+00,
         -2.3805e-02],
        [-4.4965e+00, -4.1771e-02, -2.2033e+00, -2.0334e-01, -1.3074e+00,
         -1.3273e+00],
        [-5.0536e+00, -7.8079e-01, -4.9367e-01, -3.3100e-03, -1.2805e+00,
         -1.6290e-01],
        [-3.2689e+00, -9.1440e-02, -2.8939e+00, -4.4633e-01, -1.8811e+00,
         -3.6554e+00],
        [-3.3646e+00, -7.5817e-05, -1.4167e+00, -7.3904e-01, -3.4248e+00,
         -2.2175e+00],
        [-3.4598e+00, -4.5102e+00, -2.3248e+00, -1.7466e-02, -3.2530e+00,
         -1.1019e+00],
        [-3.4079e+00, -2.8509e+00, -2.9414e+00, -4.9705e-01, -2.3094e+00,
         -2.2888e+00],
        [-5.2794e-01, -3.2113e-01, -3.4885e-01, -2.2936e-02, -8.6730e-01,
         -2.9380e-01],
        [-5.3291e+00, -1.7464e-04, -3.5240e+00, -6.0256e-01, -2.7327e+00,
         -1.1182e+00]], device='cuda:0')
tensor([ -8.4980, -12.1804,  -8.0887,  -8.0455, -10.3606,  -6.5380, -11.0213,
        -12.1497,  -6.9789,  -9.9918, -10.0832,  -4.7742,  -9.5795,  -7.7748,
        -12.2370, -11.1627, -14.6672, -14.2954,  -2.3819, -13.3068],
       device='cuda:0')
tensor([-2.5717, -1.6886, -2.2659, -3.6545, -1.7120, -3.0301, -0.4580, -1.0470,
        -2.7801, -2.3881, -2.2044, -3.3884, -1.7879, -4.2268, -1.2487, -1.2889,
         1.0196,  0.5635, -4.4238, -0.2837], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.0304, -2.6387, -1.7530, -1.6381, -2.0172, -0.8809],
        [-3.5250, -1.4535, -2.1896, -4.3504, -2.1020, -2.8554],
        [-4.5336, -3.1803, -1.8990, -1.1809, -1.5389, -1.1635],
        [-1.3888, -1.4485, -1.6805, -1.1760, -1.2307, -1.2058],
        [-1.8721, -1.4040, -2.0468, -1.4009, -1.1394, -0.9795],
        [-2.4907, -1.3683, -1.4864, -1.1016, -1.1555, -1.3610],
        [-3.5173, -2.3314, -1.4120, -1.6386, -1.2412, -1.7727],
        [-1.3705, -1.6659, -1.8566, -2.6736, -1.9285, -2.6510],
        [-2.1830, -2.1402, -2.4833, -2.1683, -2.4419, -4.0901],
        [-2.8554, -1.0023, -2.3531, -1.8085, -2.5412, -1.6236],
        [-4.4114, -2.3668, -2.6936, -1.5187, -2.1383, -3.1673],
        [-3.1747, -2.1850, -2.3001, -4.3263, -1.9137, -2.4321],
        [-1.3629, -1.5108, -2.1677, -0.9369, -1.3209, -1.0942],
        [-1.5104, -1.2200, -2.4898, -1.3532, -1.9599, -3.8342],
        [-2.4483, -1.1681, -1.7023, -1.0977, -3.4340, -2.4013],
        [-1.6913, -1.1749, -2.5370, -1.8134, -2.4415, -1.2885],
        [-1.2048, -1.0914, -1.4490, -1.2380, -2.0760, -1.1742],
        [-3.5714, -3.2458, -2.0983, -1.7358, -2.6943, -1.7553],
        [-3.3224, -2.6218, -2.7796, -0.9625, -1.6795, -1.2018],
        [-1.5089, -1.1889, -1.7460, -0.9563, -1.2894, -0.9657]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.9583, -16.4759, -13.4962,  -8.1303,  -8.8426,  -8.9636, -11.9132,
        -12.1462, -15.5068, -12.1842, -16.2961, -16.3319,  -8.3935, -12.3675,
        -12.2517, -10.9466,  -8.2334, -15.1010, -12.5676,  -7.6552],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-5.7081e-02, -2.8187e+00, -1.6651e+00, -1.2651e+00, -2.6917e+00,
         -3.9172e-04],
        [-2.8779e+00, -4.8978e-01, -2.8311e+00, -4.0340e+00, -2.2365e+00,
         -3.0452e+00],
        [-3.4478e+00, -3.4581e+00, -1.9065e+00, -1.3633e-01, -5.6762e-01,
         -5.9441e-02],
        [-1.0796e+00, -1.2618e+00, -1.1797e+00, -2.3938e-01, -3.6132e-02,
         -1.4987e-01],
        [-2.0312e+00, -9.3563e-03, -2.1537e+00, -5.6337e-01, -3.3041e-03,
         -1.6424e-01],
        [-2.3731e+00, -2.5605e-01, -5.6544e-01, -1.9361e-03, -3.8370e-01,
         -4.0172e-01],
        [-3.2925e+00, -2.9640e+00, -1.2323e+00, -8.4187e-01, -1.1219e-01,
         -1.4029e+00],
        [-1.1716e+00, -1.5615e+00, -1.7196e+00, -2.5614e+00, -2.0859e+00,
         -1.7336e+00],
        [-2.9839e+00, -1.8776e+00, -3.3428e+00, -2.2326e+00, -2.4152e+00,
         -3.6718e+00],
        [-3.8412e+00, -3.8914e-01, -2.3260e+00, -1.0124e+00, -2.7930e+00,
         -1.7576e+00],
        [-4.4318e+00, -1.7830e+00, -4.2201e+00, -7.9883e-01, -2.9069e+00,
         -3.3800e+00],
        [-3.3897e+00, -2.0957e+00, -3.0687e+00, -3.0049e+00, -1.8754e+00,
         -3.3343e+00],
        [-2.2665e-02, -1.1904e+00, -2.2988e+00, -6.5796e-02, -4.3283e-01,
         -1.2310e-02],
        [-2.0303e+00, -1.2835e-01, -2.5792e+00, -2.5216e-01, -2.2937e+00,
         -2.0089e+00],
        [-3.1852e+00, -1.4731e-01, -1.2311e+00, -1.1188e-01, -4.1837e+00,
         -2.2216e+00],
        [-1.7691e+00, -1.1139e-01, -2.3896e+00, -1.4418e+00, -3.2544e+00,
         -4.5149e-01],
        [-6.0390e-01, -4.9994e-03, -1.1631e+00, -5.0756e-01, -2.3895e+00,
         -4.2166e-01],
        [-3.7184e+00, -4.2771e+00, -2.8352e+00, -1.9413e+00, -4.5713e+00,
         -1.5326e+00],
        [-2.7715e+00, -2.3224e+00, -2.5545e+00, -7.7890e-02, -1.1228e+00,
         -7.6652e-04],
        [-1.6794e+00, -2.9547e-03, -1.3496e+00, -2.0974e-01, -7.4065e-01,
         -4.1364e-02]], device='cuda:0')
tensor([ -8.4980, -15.5144,  -9.5758,  -3.9465,  -4.9251,  -3.9820,  -9.8457,
        -10.8336, -16.5238, -12.1194, -17.5207, -16.7689,  -4.0228,  -9.2927,
        -11.0807,  -9.4178,  -5.0907, -18.8759,  -8.8498,  -4.0237],
       device='cuda:0')
tensor([-1.4603, -0.9614, -3.9204, -4.1839, -3.9175, -4.9816, -2.0674, -1.3125,
         1.0170, -0.0648,  1.2246,  0.4370, -4.3707, -3.0748, -1.1710, -1.5288,
        -3.1427,  3.7750, -3.7178, -3.6315], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.7109, -1.6731, -2.1597, -2.0877, -1.0470, -0.9992],
        [-4.1953, -1.4465, -4.4353, -2.5563, -1.9122, -1.4235],
        [-2.3543, -1.1207, -1.2998, -1.0570, -1.8625, -1.1619],
        [-3.9385, -1.9931, -3.0864, -1.4959, -2.7088, -1.2758],
        [-2.4723, -2.1886, -2.1366, -2.8715, -2.0184, -2.5242],
        [-1.6242, -1.1493, -1.3321, -1.5413, -1.4582, -1.3196],
        [-3.4057, -1.1368, -3.2961, -1.2993, -1.8516, -1.9713],
        [-1.4686, -1.5329, -2.3346, -2.4049, -1.7814, -0.8507],
        [-1.0071, -2.6410, -1.5469, -2.7399, -1.5404, -1.2342],
        [-4.1904, -2.3963, -3.5493, -3.7745, -2.3187, -1.6182],
        [-3.1586, -3.0411, -2.0400, -1.6647, -1.8336, -1.1788],
        [-1.2740, -1.5596, -3.1578, -2.4417, -1.7912, -1.0106],
        [-2.3767, -1.8089, -1.6197, -1.2830, -1.3740, -0.9173],
        [-1.5195, -1.2697, -1.6661, -1.3251, -2.5251, -2.0049],
        [-2.7203, -1.8157, -2.4618, -2.3719, -1.7072, -1.0493],
        [-3.0211, -2.4654, -1.7953, -4.1149, -1.9705, -1.3885],
        [-5.4435, -1.1975, -2.5305, -2.9322, -1.5885, -2.1219],
        [-1.7979, -1.0779, -1.7613, -0.9709, -2.6753, -1.3536],
        [-3.1442, -1.9458, -2.4314, -1.4564, -2.0577, -1.9682],
        [-2.5925, -2.0892, -2.3089, -1.4832, -2.6310, -1.6137]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.6776, -15.9691,  -8.8561, -14.4984, -14.2117,  -8.4247, -12.9608,
        -10.3732, -10.7096, -17.8474, -12.9169, -11.2349,  -9.3796, -10.3104,
        -12.1263, -14.7557, -15.8142,  -9.6370, -13.0035, -12.7185],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.3975e+00, -1.1881e+00, -2.1724e+00, -1.4794e+00, -1.6764e-01,
         -1.3804e-04],
        [-5.3447e+00, -6.9037e-01, -4.9685e+00, -2.6172e+00, -1.5653e+00,
         -8.9158e-01],
        [-3.1398e+00, -9.6998e-01, -2.8501e-01, -1.0504e-01, -1.8313e+00,
         -6.0740e-01],
        [-3.7263e+00, -1.1760e+00, -4.3844e+00, -7.0982e-01, -4.0150e+00,
         -1.0377e+00],
        [-1.9328e+00, -2.0621e+00, -2.5914e+00, -1.6626e+00, -2.3127e+00,
         -3.5924e+00],
        [-2.0618e+00, -1.7340e-01, -2.3500e-01, -1.2451e+00, -1.0637e+00,
         -6.6535e-01],
        [-4.6227e+00, -1.5404e-01, -3.8492e+00, -2.7150e-01, -1.7301e+00,
         -1.4070e+00],
        [-1.5202e+00, -2.4376e-01, -2.6823e+00, -2.6027e+00, -2.1220e+00,
         -1.4045e-02],
        [-5.8632e-02, -2.1085e+00, -1.3393e+00, -2.7684e+00, -1.4138e+00,
         -4.4700e-01],
        [-3.7864e+00, -2.8237e+00, -5.3057e+00, -4.1509e+00, -2.9700e+00,
         -1.2371e+00],
        [-3.0730e+00, -3.2117e+00, -2.2711e+00, -6.5513e-01, -1.7434e+00,
         -2.4391e-02],
        [-5.9780e-02, -1.5895e+00, -4.0090e+00, -2.0038e+00, -1.2619e+00,
         -1.2131e-02],
        [-2.8943e+00, -2.4195e+00, -8.3333e-01, -2.2485e-01, -4.7152e-01,
         -2.0590e-01],
        [-8.0747e-01, -1.5057e-01, -1.2041e+00, -4.3559e-01, -2.1577e+00,
         -1.6530e+00],
        [-2.2694e+00, -2.2016e+00, -3.4103e+00, -2.0613e+00, -1.4006e+00,
         -3.2068e-01],
        [-4.3637e+00, -3.4719e+00, -1.7312e+00, -2.9840e+00, -1.9438e+00,
         -7.7465e-01],
        [-5.8987e+00, -8.8546e-02, -3.1302e+00, -3.1446e+00, -1.1404e+00,
         -1.9931e+00],
        [-2.0538e+00, -5.5017e-01, -1.4686e+00, -1.9355e-03, -2.9245e+00,
         -8.9241e-01],
        [-2.8643e+00, -1.2490e+00, -3.3562e+00, -6.5157e-01, -2.5169e+00,
         -2.2498e+00],
        [-2.0643e+00, -1.8119e+00, -2.8558e+00, -3.1467e-01, -4.4587e+00,
         -1.1785e+00]], device='cuda:0')
tensor([ -7.4051, -16.0777,  -6.9385, -15.0492, -14.1540,  -5.4443, -12.0346,
         -9.1851,  -8.1356, -20.2738, -10.9787,  -8.9361,  -7.0494,  -6.4085,
        -11.6639, -15.2693, -15.3956,  -7.8914, -12.8878, -12.6839],
       device='cuda:0')
tensor([-2.2726,  0.1086, -1.9176,  0.5508, -0.0577, -2.9804, -0.9262, -1.1881,
        -2.5740,  2.4264, -1.9382, -2.2988, -2.3302, -3.9019, -0.4624,  0.5136,
        -0.4186, -1.7456, -0.1158, -0.0346], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.7696, -1.2030, -1.9964, -1.0222, -1.9382, -1.1552],
        [-2.4853, -1.4727, -2.6047, -1.2471, -1.5894, -1.3799],
        [-3.5060, -1.4027, -2.3066, -2.1320, -1.5237, -1.0553],
        [-1.4514, -2.2410, -1.1084, -3.3287, -1.4729, -1.5126],
        [-4.6806, -2.8410, -2.1658, -1.5878, -2.7033, -1.1495],
        [-2.5855, -3.1191, -1.8389, -1.9622, -2.0086, -1.2420],
        [-1.3383, -1.3181, -1.6233, -1.1654, -1.2952, -1.3539],
        [-1.8020, -1.2763, -1.7273, -1.6660, -2.0854, -3.5990],
        [-1.0562, -2.1927, -1.7745, -5.7598, -1.3859, -1.1392],
        [-3.7341, -2.6811, -3.0048, -3.2341, -2.5223, -1.9543],
        [-3.0436, -1.4345, -1.9498, -1.1045, -2.1392, -2.2058],
        [-1.3184, -2.0268, -1.7382, -3.8466, -1.4200, -1.3252],
        [-3.2670, -1.4255, -2.8494, -1.2407, -1.7833, -1.1573],
        [-5.0721, -2.0044, -1.7815, -1.3225, -2.0965, -1.3327],
        [-2.0702, -1.5237, -1.3418, -1.2470, -1.2622, -1.1030],
        [-1.3496, -2.9378, -1.8146, -3.2938, -1.3709, -1.0626],
        [-2.9672, -1.0080, -2.4400, -5.3790, -1.2806, -1.1418],
        [-2.4580, -1.2814, -2.5138, -2.4167, -1.2869, -1.3607],
        [-2.0154, -1.1741, -1.5532, -1.0922, -1.6132, -1.3409],
        [-3.1670, -1.8493, -2.4673, -1.1988, -1.8811, -1.4849]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.0847, -10.7791, -11.9264, -11.1148, -15.1279, -12.7563,  -8.0943,
        -12.1559, -13.3084, -17.1307, -11.8774, -11.6752, -11.7231, -13.6099,
         -8.5479, -11.8295, -14.2166, -11.3175,  -8.7890, -12.0484],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.5081e+00, -5.7460e-02, -2.0715e+00, -9.3793e-02, -2.1527e+00,
         -6.6564e-02],
        [-3.5844e+00, -1.8983e-01, -3.7240e+00, -1.9667e-02, -1.0242e+00,
         -4.6530e-01],
        [-3.8746e+00, -1.5862e-01, -2.5718e+00, -2.0263e+00, -1.6186e+00,
         -8.5783e-03],
        [-1.2075e+00, -1.5436e+00, -5.4506e-02, -3.7055e+00, -1.0640e+00,
         -1.2576e+00],
        [-4.3878e+00, -3.5007e+00, -2.9960e+00, -1.3702e+00, -3.7571e+00,
         -1.3367e-02],
        [-2.1061e+00, -4.0752e+00, -1.2265e+00, -7.9986e-01, -2.2219e+00,
         -3.8961e-02],
        [-1.7628e-01, -7.2807e-01, -1.1460e+00, -2.0810e-01, -5.9514e-02,
         -8.4603e-01],
        [-2.2388e+00, -1.2793e-02, -1.2548e+00, -1.1348e+00, -2.5387e+00,
         -2.6509e+00],
        [-4.1230e-01, -1.3837e+00, -1.7170e+00, -5.5690e+00, -1.1836e+00,
         -4.3303e-01],
        [-3.4392e+00, -2.6980e+00, -3.9746e+00, -3.2892e+00, -2.9180e+00,
         -1.6419e+00],
        [-3.2328e+00, -2.6955e-01, -2.0971e+00, -7.8339e-01, -2.3956e+00,
         -2.7217e+00],
        [-5.5841e-01, -2.7759e+00, -1.4295e+00, -3.4458e+00, -3.1450e-01,
         -8.7573e-01],
        [-3.5470e+00, -2.5539e-01, -4.0123e+00, -1.7818e-01, -2.0233e+00,
         -6.5811e-01],
        [-3.9564e+00, -1.1032e+00, -1.7245e+00, -5.0610e-01, -1.7375e+00,
         -1.1807e-01],
        [-1.2232e+00, -1.3782e+00, -3.3135e-01, -6.3071e-01, -2.3863e-02,
         -5.0940e-01],
        [-1.1923e+00, -4.1531e+00, -1.6018e+00, -3.4029e+00, -9.2325e-01,
         -5.1377e-02],
        [-3.2403e+00, -1.5162e-02, -2.7409e+00, -5.7922e+00, -4.9134e-01,
         -3.6928e-01],
        [-3.1913e+00, -1.1401e+00, -3.1321e+00, -2.5063e+00, -6.5942e-01,
         -1.0216e+00],
        [-2.3547e+00, -2.3394e-01, -1.3471e+00, -2.8107e-03, -1.3278e+00,
         -5.7494e-01],
        [-3.0909e+00, -9.2199e-01, -4.1229e+00, -1.1657e-03, -1.8010e+00,
         -1.0407e+00]], device='cuda:0')
tensor([ -6.9500,  -9.0074, -10.2585,  -8.8326, -16.0252, -10.4686,  -3.1640,
         -9.8308, -10.6986, -17.9609, -11.5001,  -9.3999, -10.6744,  -9.1458,
         -4.0967, -11.3248, -12.6492, -11.6508,  -5.8413, -10.9787],
       device='cuda:0')
tensor([-2.1347, -1.7717, -1.6678, -2.2822,  0.8974, -2.2877, -4.9302, -2.3251,
        -2.6098,  0.8302, -0.3773, -2.2753, -1.0487, -4.4641, -4.4511, -0.5047,
        -1.5674,  0.3333, -2.9477, -1.0696], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.9890, -1.3545, -2.3753, -1.2323, -1.6002, -0.8735],
        [-3.5180, -1.1619, -3.6108, -1.1745, -1.5158, -1.6201],
        [-2.4292, -1.2229, -1.4005, -1.1892, -2.1751, -1.2558],
        [-4.0920, -2.3571, -2.1454, -1.1494, -1.8421, -1.4664],
        [-3.4590, -2.6717, -2.2948, -1.7935, -1.5537, -2.1661],
        [-1.4791, -1.2256, -1.4185, -0.9620, -1.3008, -1.4978],
        [-2.0885, -1.4194, -2.9778, -1.5682, -2.1376, -1.6245],
        [-3.9000, -1.1161, -1.7379, -1.0677, -1.6677, -1.3924],
        [-4.2193, -1.1390, -1.7712, -1.6402, -1.5055, -1.2433],
        [-3.9989, -2.8603, -1.9750, -1.7787, -2.1367, -1.4195],
        [-2.4925, -2.9829, -2.5554, -1.0579, -1.6811, -1.3789],
        [-1.2109, -1.3008, -1.3852, -1.1099, -1.4593, -1.1720],
        [-3.5624, -1.3985, -2.2598, -1.3048, -1.2779, -0.8886],
        [-2.8417, -1.3119, -2.6636, -1.0853, -1.5375, -1.1375],
        [-3.6208, -0.9402, -2.6924, -1.3675, -1.4833, -1.6053],
        [-2.1118, -1.1862, -1.4292, -1.3069, -1.8333, -2.5493],
        [-2.5985, -2.5185, -1.8783, -1.1173, -1.7744, -1.3106],
        [-1.8944, -2.4113, -2.5176, -1.0376, -2.3867, -1.4408],
        [-1.8872, -1.4823, -1.7005, -1.3243, -1.4880, -1.2695],
        [-2.9637, -1.4136, -2.1596, -1.2841, -2.3531, -1.0550]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.4248, -12.6011,  -9.6726, -13.0525, -13.9388,  -7.8837, -11.8161,
        -10.8819, -11.5185, -14.1690, -12.1488,  -7.6380, -10.6921, -10.5775,
        -11.7094, -10.4167, -11.1975, -11.6884,  -9.1519, -11.2291],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.4416e+00, -1.9532e-03, -2.6027e+00, -3.5431e-02, -1.4589e+00,
         -4.7312e-03],
        [-3.8675e+00, -5.6477e-01, -3.6232e+00, -1.8394e-01, -1.2299e+00,
         -1.3200e+00],
        [-3.2017e+00, -8.2442e-02, -1.1330e+00, -2.9567e-01, -2.2974e+00,
         -3.8352e-01],
        [-4.3149e+00, -2.3296e+00, -2.9273e+00, -3.2168e-01, -1.8093e+00,
         -3.9306e-01],
        [-3.5798e+00, -2.7527e+00, -2.6694e+00, -8.4646e-01, -7.6099e-01,
         -2.5559e+00],
        [-1.5455e+00, -3.2059e-01, -6.8253e-01, -7.2956e-05, -3.1598e-02,
         -9.2834e-01],
        [-3.3475e+00, -9.9385e-04, -4.0871e+00, -8.5743e-01, -2.6951e+00,
         -4.6132e-01],
        [-4.2636e+00, -4.3100e-01, -1.3745e+00, -7.4411e-02, -2.0008e+00,
         -1.0969e+00],
        [-4.1365e+00, -1.1101e-01, -1.7490e+00, -1.1532e+00, -1.3073e+00,
         -5.3397e-01],
        [-3.9487e+00, -4.0932e+00, -2.2984e+00, -1.9560e+00, -2.7864e+00,
         -8.6846e-01],
        [-1.9586e+00, -3.2892e+00, -3.2542e+00, -1.1079e-02, -1.3600e+00,
         -5.9029e-01],
        [-5.0402e-02, -6.5576e-01, -6.7623e-01, -7.3360e-02, -6.1374e-01,
         -6.5188e-02],
        [-5.0108e+00, -6.6138e-02, -2.6347e+00, -1.1385e-01, -1.6331e-01,
         -2.8343e-03],
        [-3.2810e+00, -1.1089e-01, -3.6322e+00, -1.3389e-02, -1.4818e+00,
         -2.3964e-01],
        [-3.9510e+00, -3.7644e-01, -2.5534e+00, -5.4715e-01, -1.1309e+00,
         -1.4621e+00],
        [-2.6566e+00, -3.1484e-02, -1.3393e+00, -4.2229e-01, -1.7938e+00,
         -1.8580e+00],
        [-2.2633e+00, -2.7575e+00, -1.6936e+00, -1.9631e-02, -1.5685e+00,
         -7.9423e-03],
        [-8.1123e-01, -2.1746e+00, -3.2285e+00, -3.3574e-02, -3.2104e+00,
         -1.0264e+00],
        [-2.5979e+00, -1.1738e+00, -1.5427e+00, -3.6688e-01, -4.3886e-01,
         -2.6956e-01],
        [-4.0708e+00, -4.7881e-02, -2.6176e+00, -7.0682e-02, -2.8520e+00,
         -6.5160e-01]], device='cuda:0')
tensor([ -7.5454, -10.7893,  -7.3938, -12.0959, -13.1652,  -3.5087, -11.4496,
         -9.2413,  -8.9910, -15.9511, -10.4634,  -2.1347,  -7.9917,  -8.7589,
        -10.0209,  -8.1015,  -8.3105, -10.4846,  -6.3896, -10.3106],
       device='cuda:0')
tensor([-2.8794, -1.8118, -2.2788, -0.9566, -0.7736, -4.3751, -0.3665, -1.6406,
        -2.5274,  1.7821, -1.6854, -5.5033, -2.7004, -1.8186, -1.6886, -2.3151,
        -2.8870, -1.2038, -2.7622, -0.9185], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-5.7145, -1.0543, -2.4835, -1.1053, -1.3263, -1.2808],
        [-1.3497, -1.9975, -2.6305, -1.8211, -1.1836, -0.8165],
        [-4.4541, -1.4408, -2.6662, -2.6138, -2.8985, -1.7814],
        [-1.9694, -1.2424, -1.1952, -0.9249, -3.7890, -1.2604],
        [-1.8779, -1.2727, -1.4740, -1.2088, -1.6008, -2.9135],
        [-3.7922, -2.6717, -1.8916, -1.2189, -2.0204, -1.6480],
        [-2.9939, -4.2327, -2.4115, -1.8888, -1.7748, -1.8795],
        [-1.4077, -1.2113, -1.2780, -1.1949, -1.6632, -1.4278],
        [-4.3161, -1.1756, -2.9214, -1.3447, -2.7258, -1.5271],
        [-2.8009, -1.3748, -2.4759, -1.1863, -1.7125, -1.0601],
        [-3.9394, -1.3271, -1.2127, -1.4021, -2.7941, -2.0398],
        [-4.2431, -2.1304, -2.6950, -1.2574, -1.9525, -1.5331],
        [-3.4558, -3.7583, -2.1370, -1.0254, -2.3573, -1.5120],
        [-1.8547, -1.1996, -1.6876, -1.5675, -1.6434, -1.2329],
        [-2.8091, -1.4162, -1.9578, -1.3266, -1.4000, -0.9303],
        [-2.7449, -1.1753, -1.3394, -1.3840, -1.4035, -1.1059],
        [-3.4235, -1.4584, -1.0590, -1.4133, -1.2646, -1.2305],
        [-1.6798, -1.7273, -1.3570, -3.0996, -1.4831, -1.0345],
        [-2.3510, -1.6757, -2.4530, -3.9927, -1.2289, -1.6311],
        [-2.0559, -2.6350, -2.4083, -2.0741, -4.4670, -2.1336]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.9647,  -9.7989, -15.8549, -10.3813, -10.3477, -13.2428, -15.1812,
         -8.1829, -14.0108, -10.6105, -12.7152, -13.8116, -14.2458,  -9.1858,
         -9.8400,  -9.1530,  -9.8494, -10.3813, -13.3324, -15.7739],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-5.1169e+00, -4.6660e-01, -2.9621e+00, -9.1868e-02, -4.6653e-02,
         -6.3455e-04],
        [-7.8597e-01, -2.9585e+00, -3.2193e+00, -1.4481e+00, -1.2600e-04,
         -7.1312e-01],
        [-5.2965e+00, -3.8864e-01, -3.1217e+00, -2.6467e+00, -2.7954e+00,
         -1.1265e+00],
        [-2.3826e+00, -1.1294e+00, -5.3150e-02, -7.5183e-02, -4.4305e+00,
         -9.5733e-01],
        [-2.4951e+00, -7.5627e-01, -1.1305e+00, -3.6548e-02, -9.4123e-01,
         -2.2737e+00],
        [-3.9776e+00, -3.2678e+00, -2.0167e+00, -4.7986e-01, -2.4615e+00,
         -1.7216e+00],
        [-3.1558e+00, -4.8593e+00, -2.8634e+00, -9.9198e-01, -1.8225e+00,
         -1.9754e+00],
        [-1.2766e+00, -3.5509e-02, -9.3555e-02, -7.5783e-01, -1.3077e+00,
         -8.8918e-01],
        [-5.7400e+00, -2.5636e-02, -3.6402e+00, -4.4365e-01, -3.2106e+00,
         -8.8468e-01],
        [-3.3315e+00, -5.9907e-01, -3.1880e+00, -6.6901e-02, -1.9764e+00,
         -2.9257e-01],
        [-4.2445e+00, -5.7221e-01, -1.4317e-01, -6.6261e-01, -4.0615e+00,
         -1.6781e+00],
        [-3.9044e+00, -1.9146e+00, -4.0002e+00, -6.1249e-01, -2.0385e+00,
         -1.1332e+00],
        [-3.6513e+00, -4.7026e+00, -2.2461e+00, -9.9200e-03, -3.7367e+00,
         -8.7775e-01],
        [-2.8099e+00, -1.2331e-01, -1.3770e+00, -8.5862e-01, -1.1633e+00,
         -8.9019e-02],
        [-3.8192e+00, -2.8503e-04, -2.0115e+00, -1.7219e-01, -5.6873e-01,
         -4.1122e-02],
        [-2.1479e+00, -1.8581e-02, -3.8000e-01, -6.3707e-01, -6.3656e-02,
         -1.2286e-01],
        [-2.7665e+00, -1.3198e+00, -2.3243e-01, -4.9178e-01, -2.5304e-01,
         -1.0803e+00],
        [-1.9613e+00, -1.7070e+00, -7.9453e-01, -2.5483e+00, -9.6565e-01,
         -7.3159e-04],
        [-2.5980e+00, -1.1284e+00, -2.7369e+00, -3.9626e+00, -4.2343e-01,
         -1.2620e+00],
        [-2.7764e+00, -1.9213e+00, -2.5561e+00, -1.4265e+00, -4.8042e+00,
         -2.1133e+00]], device='cuda:0')
tensor([ -8.6847,  -9.1251, -15.3755,  -9.0281,  -7.6332, -13.9251, -15.6683,
         -4.3603, -13.9448,  -9.4545, -11.3621, -13.6034, -15.2243,  -6.4211,
         -6.6131,  -3.3700,  -6.1438,  -7.9775, -12.1113, -15.5978],
       device='cuda:0')
tensor([-4.2800, -0.6738, -0.4794, -1.3532, -2.7144,  0.6822,  0.4872, -3.8225,
        -0.0660, -1.1560, -1.3532, -0.2081,  0.9785, -2.7648, -3.2269, -5.7830,
        -3.7057, -2.4037, -1.2211, -0.1761], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.3427, -3.9637, -1.8827, -2.2336, -2.2281, -1.2090],
        [-1.3800, -1.3019, -2.7129, -2.0846, -1.4512, -1.3147],
        [-1.6557, -1.2439, -3.4947, -1.2401, -1.7021, -1.4676],
        [-1.9222, -1.2683, -2.1317, -1.1671, -1.9913, -1.3794],
        [-3.4930, -1.1391, -1.0981, -1.8748, -1.5347, -1.2101],
        [-3.7451, -2.9246, -2.4406, -1.1968, -2.2993, -2.3660],
        [-2.7692, -4.7677, -1.8674, -1.5439, -2.2441, -1.6471],
        [-2.3250, -1.0810, -1.1248, -1.2613, -1.6596, -1.2099],
        [-3.7340, -1.4578, -2.0544, -1.2865, -2.9261, -2.2042],
        [-4.5923, -1.3271, -3.2991, -1.4163, -2.2596, -1.0433],
        [-1.0154, -1.2341, -1.4508, -2.3785, -1.9979, -1.0849],
        [-2.8658, -2.7921, -1.8464, -2.4586, -2.4904, -2.1598],
        [-2.7911, -4.6875, -2.3918, -1.5917, -1.8912, -1.9319],
        [-1.9142, -1.3150, -1.4196, -1.0176, -1.9615, -0.9736],
        [-2.4704, -1.4064, -2.7182, -1.2278, -1.1881, -0.8534],
        [-3.4329, -1.5906, -1.6895, -1.4282, -1.1767, -1.1351],
        [-1.8094, -1.2011, -1.1294, -1.7414, -1.8594, -2.2124],
        [-3.7047, -2.0559, -1.5124, -1.4115, -2.5335, -1.2947],
        [-1.9815, -3.4298, -1.9059, -1.5186, -2.2518, -1.2733],
        [-1.3796, -1.2165, -2.0285, -2.6373, -1.3920, -1.1100]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.8598, -10.2454, -10.8040,  -9.8599, -10.3498, -14.9723, -14.8395,
         -8.6615, -13.6628, -13.9377,  -9.1617, -14.6131, -15.2852,  -8.6014,
         -9.8643, -10.4530,  -9.9531, -12.5127, -12.3609,  -9.7640],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.3738e+00, -5.2576e+00, -1.8639e+00, -1.2992e+00, -2.8147e+00,
         -3.8604e-01],
        [-4.6105e-01, -2.7028e-01, -3.4673e+00, -1.2160e+00, -5.7803e-02,
         -6.6825e-01],
        [-1.9525e+00, -1.1001e-02, -4.2356e+00, -7.4788e-02, -1.6222e+00,
         -8.3635e-01],
        [-2.3100e+00, -1.6870e-01, -2.1791e+00, -2.6708e-02, -2.6236e+00,
         -8.3653e-01],
        [-3.5381e+00, -2.6813e-02, -1.9705e-01, -1.4811e+00, -1.3741e+00,
         -4.6213e-01],
        [-3.7803e+00, -3.6734e+00, -3.7949e+00, -4.6277e-01, -3.1568e+00,
         -2.9653e+00],
        [-2.4595e+00, -6.0586e+00, -1.4932e+00, -3.7171e-01, -3.2203e+00,
         -1.4906e+00],
        [-3.3664e+00, -8.1450e-02, -3.6403e-02, -5.3730e-01, -1.5098e+00,
         -3.9346e-01],
        [-5.5079e+00, -2.8428e-01, -2.2161e+00, -2.4477e-01, -3.5885e+00,
         -1.4872e+00],
        [-4.9952e+00, -9.8891e-03, -4.6707e+00, -7.7533e-01, -3.1599e+00,
         -1.7153e-01],
        [-6.6541e-02, -1.7089e-02, -1.2413e+00, -2.5615e+00, -2.6072e+00,
         -1.2589e-04],
        [-2.7975e+00, -4.0928e+00, -2.1279e+00, -3.4042e+00, -3.5495e+00,
         -2.1191e+00],
        [-2.7901e+00, -4.9390e+00, -2.9681e+00, -5.4558e-01, -2.1847e+00,
         -2.1401e+00],
        [-2.8497e+00, -9.1823e-02, -2.7159e-01, -3.2542e-02, -1.9390e+00,
         -6.8591e-02],
        [-3.5558e+00, -1.2530e-01, -3.8215e+00, -8.5043e-03, -2.9899e-03,
         -3.5048e-02],
        [-3.9837e+00, -5.4462e-01, -1.2542e+00, -6.2221e-01, -5.8623e-01,
         -1.6599e-01],
        [-1.8807e+00, -1.3379e-01, -6.9614e-02, -1.5747e+00, -2.0613e+00,
         -2.6521e+00],
        [-3.3120e+00, -1.6797e+00, -1.1488e+00, -1.0335e+00, -3.7215e+00,
         -5.8591e-01],
        [-7.2249e-01, -4.2053e+00, -1.8918e+00, -4.9567e-01, -2.8830e+00,
         -5.8333e-01],
        [-7.9103e-01, -2.2085e-02, -2.2048e+00, -2.0507e+00, -1.9972e-03,
         -7.3659e-04]], device='cuda:0')
tensor([-12.9952,  -6.1407,  -8.7324,  -8.1447,  -7.0792, -17.8336, -15.0940,
         -5.9248, -13.3287, -13.7825,  -6.4937, -18.0910, -15.5675,  -5.2532,
         -7.5491,  -7.1569,  -8.3723, -11.4814, -10.7816,  -5.0713],
       device='cuda:0')
tensor([-0.8647, -4.1047, -2.0716, -1.7152, -3.2706,  2.8612,  0.2545, -2.7368,
        -0.3341, -0.1551, -2.6680,  3.4779,  0.2823, -3.3483, -2.3152, -3.2960,
        -1.5809, -1.0314, -1.5793, -4.6927], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.9819, -1.2299, -2.6642, -1.5273, -3.1029, -2.1336],
        [-3.3468, -1.4153, -2.1522, -1.8931, -1.7427, -1.3331],
        [-2.1330, -1.6063, -1.5209, -3.9485, -1.6280, -1.4172],
        [-1.9508, -1.8001, -2.1211, -6.0422, -2.6289, -2.0517],
        [-4.5186, -3.8869, -2.0541, -1.2881, -1.5889, -1.2447],
        [-1.3677, -1.1265, -2.0398, -2.7175, -1.0946, -1.3046],
        [-2.0085, -1.6758, -2.3990, -1.1837, -1.4365, -1.2944],
        [-3.1860, -2.0229, -2.8536, -1.3341, -2.2758, -1.2031],
        [-2.1528, -1.6100, -1.9395, -1.8261, -1.1804, -1.0949],
        [-5.4019, -1.1676, -1.9574, -1.4648, -1.1489, -1.4095],
        [-1.4568, -1.4500, -2.1174, -1.7993, -2.5942, -1.2011],
        [-2.1968, -1.0879, -1.6997, -1.2377, -1.5663, -1.4497],
        [-3.8434, -2.3440, -1.8722, -1.2137, -1.9483, -1.7814],
        [-3.0276, -2.6043, -2.7736, -1.3621, -2.4515, -2.0541],
        [-1.2152, -1.1970, -1.1119, -1.3807, -2.3412, -1.8071],
        [-2.4649, -1.1703, -2.1110, -1.3397, -2.2225, -3.1323],
        [-2.4658, -1.4445, -3.5341, -1.6297, -2.2336, -1.9448],
        [-2.6818, -1.7096, -1.3375, -3.5574, -1.6802, -1.9707],
        [-4.5421, -1.4126, -2.2043, -2.3625, -2.6801, -1.5005],
        [-2.5061, -2.7796, -1.8048, -1.2500, -2.7645, -1.4634]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.6398, -11.8832, -12.2539, -16.5947, -14.5812,  -9.6508,  -9.9980,
        -12.8755,  -9.8036, -12.5501, -10.6188,  -9.2382, -13.0029, -14.2733,
         -9.0532, -12.4407, -13.2526, -12.9372, -14.7020, -12.5683],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.5075e+00, -4.3118e-01, -3.3206e+00, -7.7835e-01, -3.4284e+00,
         -1.4668e+00],
        [-4.0616e+00, -6.4861e-03, -2.2250e+00, -1.7897e+00, -2.1978e+00,
         -6.6682e-01],
        [-2.2979e+00, -5.7870e-01, -1.2746e+00, -4.0270e+00, -1.5632e+00,
         -4.4635e-01],
        [-1.4171e+00, -9.2589e-01, -2.9512e+00, -6.7107e+00, -3.6282e+00,
         -2.2323e+00],
        [-3.5944e+00, -3.8408e+00, -1.7678e+00, -2.5215e-01, -7.3785e-01,
         -4.7014e-02],
        [-4.0556e-01, -4.7759e-01, -1.7966e+00, -2.2167e+00, -6.0047e-01,
         -9.1971e-01],
        [-2.2549e+00, -1.8752e+00, -2.9963e+00, -3.0867e-02, -1.0225e+00,
         -9.8325e-01],
        [-2.6567e+00, -2.9632e+00, -3.6894e+00, -6.2700e-01, -2.0743e+00,
         -7.2464e-02],
        [-3.6748e+00, -1.5738e+00, -2.3883e+00, -1.5677e+00, -1.0547e-01,
         -1.0204e-02],
        [-6.1857e+00, -3.5057e-01, -2.0072e+00, -1.0779e+00, -1.6662e-01,
         -9.5164e-01],
        [-1.1233e+00, -1.4226e+00, -2.1177e+00, -1.7425e+00, -3.0359e+00,
         -7.1574e-01],
        [-2.4085e+00, -2.1674e-02, -1.6526e+00, -5.3115e-02, -1.2081e+00,
         -9.0680e-01],
        [-4.3747e+00, -2.2907e+00, -1.3323e+00, -6.1280e-02, -2.2463e+00,
         -1.8978e+00],
        [-3.1894e+00, -2.6839e+00, -3.6448e+00, -2.7556e-01, -3.5779e+00,
         -2.3340e+00],
        [-7.0172e-01, -4.6650e-02, -1.1675e-01, -7.1934e-01, -2.7173e+00,
         -1.5827e+00],
        [-3.3260e+00, -7.3910e-06, -2.1523e+00, -4.7831e-01, -2.4471e+00,
         -2.5334e+00],
        [-3.0608e+00, -1.0547e-02, -5.0283e+00, -1.1600e+00, -3.2056e+00,
         -1.2473e+00],
        [-2.9178e+00, -8.3955e-01, -1.0103e-01, -3.8168e+00, -1.8897e+00,
         -1.7570e+00],
        [-4.2343e+00, -4.7128e-01, -2.3735e+00, -2.3691e+00, -3.2914e+00,
         -9.0620e-01],
        [-2.2336e+00, -3.7512e+00, -6.4982e-01, -1.2410e-04, -4.4073e+00,
         -9.1760e-01]], device='cuda:0')
tensor([-11.9329, -10.9474, -10.1877, -17.8654, -10.2400,  -6.4167,  -9.1630,
        -12.0831,  -9.3203, -10.7395, -10.1578,  -6.2508, -12.2031, -15.7055,
         -5.8844, -10.9370, -13.7126, -11.3219, -13.6458, -11.9596],
       device='cuda:0')
tensor([-0.7069, -0.9358, -2.0662,  1.2707, -4.3412, -3.2341, -0.8350, -0.7924,
        -0.4834, -1.8106, -0.4610, -2.9874, -0.7998,  1.4322, -3.1688, -1.5037,
         0.4600, -1.6153, -1.0562, -0.6087], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.4480, -1.2007, -2.1485, -1.0143, -1.3767, -1.1650],
        [-2.8098, -1.6391, -3.2233, -1.2346, -1.2001, -0.8198],
        [-3.4816, -1.4564, -2.1072, -1.2382, -1.3300, -1.0934],
        [-4.1358, -1.1965, -3.2629, -1.2309, -1.3448, -1.8969],
        [-2.3365, -1.2052, -1.3233, -1.2346, -1.9072, -2.0480],
        [-4.8355, -1.7551, -2.3276, -1.0565, -2.0692, -1.1185],
        [-2.6960, -2.7152, -1.9121, -2.1015, -1.8434, -1.3631],
        [-2.4291, -1.2030, -1.4125, -1.0991, -1.6225, -1.5040],
        [-1.9694, -1.3010, -2.7673, -1.1742, -2.2889, -0.9314],
        [-3.5821, -1.5381, -2.4875, -1.1418, -1.3590, -1.0862],
        [-3.5172, -1.1301, -2.7049, -1.2182, -1.4311, -1.5575],
        [-2.5304, -1.1164, -1.0956, -1.1963, -2.2656, -1.6064],
        [-3.8481, -3.6058, -1.8811, -1.0684, -1.6042, -1.1310],
        [-3.0935, -2.4085, -1.9040, -1.3180, -2.4136, -1.5131],
        [-1.5407, -1.8007, -1.4749, -1.3276, -1.5892, -1.1077],
        [-2.0763, -1.4653, -2.1637, -1.3303, -1.9663, -0.9057],
        [-3.8981, -1.1612, -3.3469, -1.3293, -1.4626, -1.3383],
        [-2.7760, -1.2933, -1.7389, -2.1093, -1.7687, -1.5636],
        [-2.8135, -2.4044, -1.7921, -1.1841, -1.7797, -1.4153],
        [-2.8049, -2.9408, -2.2334, -1.0945, -2.2567, -1.6938]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.3533, -10.9267, -10.7068, -13.0678, -10.0548, -13.1625, -12.6313,
         -9.2704, -10.4321, -11.1947, -11.5589,  -9.8108, -13.1386, -12.6507,
         -8.8408,  -9.9077, -12.5364, -11.2498, -11.3890, -13.0241],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.2689e+00, -2.3140e-01, -2.2685e+00, -3.4193e-01, -3.2462e-01,
         -1.3149e-01],
        [-4.1511e+00, -6.7228e-01, -4.3017e+00, -2.4005e-02, -1.0631e-03,
         -1.6253e-02],
        [-4.0375e+00, -3.5534e-01, -2.3680e+00, -4.8105e-01, -9.5711e-01,
         -5.5537e-02],
        [-4.3856e+00, -6.2572e-01, -3.0110e+00, -1.7378e-01, -5.0300e-01,
         -1.8185e+00],
        [-3.0854e+00, -2.0012e-01, -1.7373e-01, -8.1468e-02, -2.1190e+00,
         -1.1651e+00],
        [-4.5934e+00, -1.0859e+00, -2.8126e+00, -9.0946e-03, -2.3132e+00,
         -7.1856e-02],
        [-2.1485e+00, -3.1080e+00, -1.6674e+00, -9.8180e-01, -1.9616e+00,
         -6.3658e-05],
        [-3.9832e+00, -7.9846e-04, -7.9192e-01, -9.6661e-02, -1.0722e+00,
         -9.4879e-01],
        [-2.6183e+00, -1.4791e-01, -3.7152e+00, -1.5360e-01, -2.4230e+00,
         -1.6734e-02],
        [-3.8396e+00, -4.2493e-01, -2.8078e+00, -5.3898e-03, -7.8960e-01,
         -2.9885e-01],
        [-4.1606e+00, -3.3361e-02, -2.7203e+00, -1.9893e-01, -1.0012e+00,
         -1.0280e+00],
        [-3.0214e+00, -6.9309e-03, -1.1898e-02, -2.2224e-01, -2.5594e+00,
         -1.0790e+00],
        [-3.9988e+00, -4.3632e+00, -2.0179e+00, -3.1257e-02, -9.0357e-01,
         -2.0850e-04],
        [-2.7884e+00, -2.4621e+00, -1.7411e+00, -2.8199e-01, -3.4824e+00,
         -1.0515e+00],
        [-1.7239e+00, -1.8996e+00, -9.3215e-01, -4.4037e-01, -1.0357e+00,
         -2.7734e-02],
        [-3.1194e+00, -5.2343e-02, -2.6131e+00, -9.4827e-02, -2.0992e+00,
         -1.6267e-03],
        [-4.3695e+00, -2.6749e-02, -4.2982e+00, -5.6848e-01, -1.3002e+00,
         -7.2932e-01],
        [-3.3683e+00, -2.1906e-01, -1.8235e+00, -2.0305e+00, -1.7790e+00,
         -1.2849e+00],
        [-2.4634e+00, -2.7664e+00, -1.0125e+00, -4.1274e-01, -1.4951e+00,
         -3.2095e-01],
        [-2.2444e+00, -2.8490e+00, -2.5192e+00, -9.2797e-02, -2.7578e+00,
         -1.5551e+00]], device='cuda:0')
tensor([ -7.5669,  -9.1665,  -8.2546, -10.5176,  -6.8249, -10.8861,  -9.8673,
         -6.8935,  -9.0747,  -8.1662,  -9.1423,  -6.9010, -11.3149, -11.8076,
         -6.0595,  -7.9804, -11.2924, -10.5054,  -8.4710, -12.0184],
       device='cuda:0')
tensor([-1.7864, -1.7602, -2.4522, -2.5502, -3.2299, -2.2764, -2.7640, -2.3768,
        -1.3575, -3.0285, -2.4166, -2.9098, -1.8237, -0.8431, -2.7813, -1.9272,
        -1.2439, -0.7444, -2.9180, -1.0057], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.9141, -1.3000, -1.2805, -0.9871, -1.6839, -1.3133],
        [-2.2561, -1.6205, -4.1572, -1.4081, -1.4589, -0.9352],
        [-3.9678, -1.7753, -2.0895, -1.1566, -2.3638, -1.8909],
        [-1.4654, -1.1883, -1.4567, -1.3326, -1.8196, -1.3359],
        [-1.6221, -1.2617, -2.0098, -1.3412, -1.1739, -1.4275],
        [-1.9765, -1.2427, -1.7631, -1.5990, -1.4941, -2.1395],
        [-4.9318, -1.7454, -3.2814, -1.6043, -1.4658, -1.2857],
        [-2.4115, -1.1788, -1.2280, -1.6524, -1.7778, -2.5774],
        [-3.8196, -2.0873, -1.3189, -1.2716, -1.7463, -1.4013],
        [-2.8462, -2.7130, -2.2699, -1.3550, -2.4316, -1.3259],
        [-1.3790, -1.5467, -1.9973, -1.2644, -1.5055, -1.2269],
        [-4.2677, -1.2106, -3.2010, -1.2810, -2.3540, -1.8261],
        [-3.2741, -1.2517, -3.7529, -1.1645, -1.8284, -1.4338],
        [-1.1826, -1.4982, -1.1755, -1.6157, -2.9318, -1.6199],
        [-4.0062, -1.9788, -1.7022, -1.2712, -1.6450, -1.9112],
        [-2.4964, -3.5489, -2.2983, -1.6072, -1.9926, -1.5505],
        [-1.4252, -1.1211, -2.1125, -1.2500, -1.4664, -1.1245],
        [-1.6468, -1.6543, -2.2646, -1.2939, -1.2852, -0.9675],
        [-2.4653, -1.0756, -1.7334, -1.1209, -1.7248, -1.3614],
        [-5.1305, -1.2285, -6.8334, -1.1725, -1.3404, -1.0548]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.4789, -11.8359, -13.2438,  -8.5985,  -8.8362, -10.2149, -14.3144,
        -10.8259, -11.6452, -12.9417,  -8.9199, -14.1404, -12.7055, -10.0237,
        -12.5145, -13.4940,  -8.4997,  -9.1122,  -9.4813, -16.7600],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.7642e+00, -1.2293e+00, -2.9011e-01, -2.6390e-02, -1.5046e+00,
         -9.4297e-01],
        [-1.3744e+00, -4.3700e-01, -6.3284e+00, -3.7410e-01, -2.9647e-01,
         -4.2025e-01],
        [-4.4785e+00, -1.0560e+00, -2.3018e+00, -6.7183e-03, -3.6071e+00,
         -1.8932e+00],
        [-5.2703e-02, -7.5773e-02, -7.8655e-01, -9.2212e-01, -1.8564e+00,
         -5.4587e-01],
        [-1.8666e+00, -1.2592e-03, -1.9778e+00, -5.7222e-01, -2.4206e-01,
         -1.2500e+00],
        [-2.2945e+00, -7.5222e-02, -1.7684e+00, -7.0707e-01, -1.3574e+00,
         -1.8751e+00],
        [-4.9140e+00, -8.7365e-01, -3.8299e+00, -1.0894e+00, -9.0443e-01,
         -3.3838e-01],
        [-3.3088e+00, -4.6242e-03, -5.2021e-03, -1.2961e+00, -1.5833e+00,
         -2.4108e+00],
        [-3.5535e+00, -1.7774e+00, -3.4321e-01, -5.8786e-01, -1.3314e+00,
         -5.9724e-01],
        [-2.3761e+00, -2.8746e+00, -2.8512e+00, -3.3968e-01, -3.2477e+00,
         -5.8011e-01],
        [-8.6748e-01, -1.3707e+00, -2.0433e+00, -3.2719e-01, -6.8571e-01,
         -2.4309e-01],
        [-5.6386e+00, -1.7099e-01, -4.2837e+00, -9.5719e-02, -2.7147e+00,
         -9.6746e-01],
        [-3.5995e+00, -2.4467e-02, -5.1471e+00, -2.3340e-02, -2.3637e+00,
         -1.1728e+00],
        [-3.6687e-01, -4.3503e-01, -3.3252e-03, -1.0173e+00, -3.6442e+00,
         -1.2722e+00],
        [-4.0114e+00, -1.6335e+00, -1.5305e+00, -8.2523e-01, -1.1469e+00,
         -1.6823e+00],
        [-2.2112e+00, -4.4438e+00, -3.3844e+00, -3.1984e-01, -2.2125e+00,
         -1.2168e+00],
        [-1.1714e+00, -5.6923e-02, -2.1865e+00, -3.9869e-01, -9.9155e-01,
         -1.7193e-01],
        [-2.1317e+00, -6.2971e-01, -2.7981e+00, -3.9364e-02, -1.9892e-01,
         -6.7924e-01],
        [-2.6994e+00, -1.4693e-03, -1.4750e+00, -9.2960e-02, -1.9574e+00,
         -7.8740e-01],
        [-5.0939e+00, -3.1212e-01, -6.6684e+00, -1.6462e-02, -6.6687e-01,
         -7.7451e-02]], device='cuda:0')
tensor([ -7.7576,  -9.2306, -13.3433,  -4.2394,  -5.9099,  -8.0777, -11.9498,
         -8.6088,  -8.1906, -12.2694,  -5.5375, -13.8712, -12.3308,  -6.7389,
        -10.8300, -13.7885,  -4.9769,  -6.4770,  -7.0136, -12.8352],
       device='cuda:0')
tensor([-1.7214, -2.6053,  0.0995, -4.3591, -2.9263, -2.1371, -2.3645, -2.2171,
        -3.4546, -0.6722, -3.3825, -0.2693, -0.3746, -3.2848, -1.6846,  0.2946,
        -3.5228, -2.6353, -2.4677, -3.9248], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.9555, -1.0265, -4.8188, -1.1441, -1.2222, -2.5099],
        [-1.2847, -1.7350, -1.2161, -1.0894, -2.7339, -3.2858],
        [-2.8052, -3.5410, -1.6845, -1.0363, -2.0921, -2.2749],
        [-2.3773, -2.9914, -3.0560, -2.3592, -3.0137, -1.7017],
        [-1.6768, -1.8718, -1.5275, -6.1212, -1.3457, -1.0267],
        [-2.1986, -1.7630, -3.9482, -1.3460, -2.4349, -4.1205],
        [-2.2580, -1.0529, -1.4368, -1.0958, -3.7088, -2.4720],
        [-2.8480, -1.3899, -3.1094, -2.4842, -1.2978, -2.4649],
        [-3.5910, -2.7403, -1.6318, -1.6005, -2.5128, -1.6917],
        [-1.9537, -2.8286, -2.7592, -0.9080, -1.4464, -1.2537],
        [-1.7707, -1.4463, -1.4896, -1.0932, -1.4383, -0.9162],
        [-1.4834, -1.6106, -1.7768, -1.6207, -1.1640, -1.0495],
        [-2.4184, -0.9657, -1.4989, -1.2107, -1.3962, -1.2000],
        [-2.3113, -1.3225, -1.3187, -1.4482, -1.2839, -2.6697],
        [-1.1887, -1.4001, -2.4109, -3.0029, -1.6456, -1.1615],
        [-2.3652, -1.2852, -3.0493, -2.1770, -1.6110, -1.6039],
        [-2.9730, -4.9587, -1.4722, -1.2785, -3.3704, -3.1744],
        [-3.3359, -2.2382, -1.5023, -1.1987, -1.6197, -0.9808],
        [-2.5612, -2.5542, -2.1079, -1.3198, -2.1804, -2.0731],
        [-1.3958, -1.1765, -1.5726, -0.9708, -1.5721, -1.5677]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.6771, -11.3448, -13.4340, -15.4993, -13.5697, -15.8112, -12.0243,
        -13.5942, -13.7681, -11.1497,  -8.1543,  -8.7050,  -8.6899, -10.3543,
        -10.8098, -12.0917, -17.2273, -10.8757, -12.7966,  -8.2556],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.6001e+00, -1.4724e-02, -4.7614e+00, -8.9516e-02, -3.0249e-01,
         -2.7957e+00],
        [-8.3113e-01, -8.7646e-01, -1.1139e-01, -7.8262e-02, -3.1499e+00,
         -1.5639e+00],
        [-2.6659e+00, -4.5517e+00, -1.3641e+00, -1.3602e-02, -2.5874e+00,
         -2.7978e+00],
        [-1.5994e+00, -3.1907e+00, -4.1431e+00, -1.7293e+00, -4.0575e+00,
         -1.6538e+00],
        [-6.4230e-01, -2.1780e+00, -1.0238e+00, -4.7411e+00, -1.7743e-02,
         -8.5024e-02],
        [-3.1211e+00, -6.6650e-01, -4.6740e+00, -1.3239e-01, -3.4953e+00,
         -2.6483e+00],
        [-2.4360e+00, -5.9725e-02, -5.7968e-02, -4.9440e-02, -4.7413e+00,
         -3.4997e+00],
        [-2.3957e+00, -2.0125e-01, -3.8614e+00, -2.6374e+00, -2.3784e-02,
         -2.4102e+00],
        [-4.6391e+00, -3.6096e+00, -5.8526e-01, -2.3368e+00, -3.8303e+00,
         -1.2703e+00],
        [-7.7130e-01, -2.4281e+00, -2.7687e+00, -9.6691e-04, -1.2131e-01,
         -5.7715e-02],
        [-2.2638e+00, -7.2209e-03, -2.2471e-01, -2.7179e-03, -9.3574e-01,
         -8.6133e-02],
        [-1.2438e+00, -5.3820e-01, -1.4477e+00, -8.8699e-01, -3.0576e-03,
         -3.1077e-01],
        [-2.1690e+00, -1.6501e-01, -8.4223e-01, -2.9149e-01, -3.4807e-01,
         -2.9683e-04],
        [-1.4475e+00, -3.0118e-01, -7.6192e-01, -2.1659e-01, -1.7962e-01,
         -3.0191e+00],
        [-4.7356e-01, -4.4165e-01, -2.9707e+00, -2.0982e+00, -1.1333e+00,
         -1.4260e-01],
        [-2.5805e+00, -4.6339e-01, -3.2827e+00, -2.1415e+00, -1.2522e+00,
         -1.4384e+00],
        [-4.0237e+00, -3.5031e+00, -8.7403e-01, -3.7666e-01, -3.9416e+00,
         -3.9351e+00],
        [-3.3891e+00, -1.7817e+00, -7.3009e-01, -1.4463e-01, -1.2234e+00,
         -1.1994e-01],
        [-2.4206e+00, -3.4515e+00, -2.7196e+00, -1.4546e-02, -2.5445e+00,
         -2.7717e+00],
        [-2.3233e-02, -2.8208e-03, -1.0901e+00, -1.4028e-01, -7.4162e-01,
         -1.0501e+00]], device='cuda:0')
tensor([-11.5639,  -6.6111, -13.9805, -16.3737,  -8.6880, -14.7375, -10.8442,
        -11.5298, -16.2714,  -6.1481,  -3.5203,  -4.4306,  -3.8161,  -5.9260,
         -7.2600, -11.1588, -16.6542,  -7.3889, -13.9225,  -3.0481],
       device='cuda:0')
tensor([-2.1132, -4.7338,  0.5466,  0.8744, -4.8817, -1.0737, -1.1801, -2.0644,
         2.5033, -5.0016, -4.6340, -4.2745, -4.8739, -4.4284, -3.5498, -0.9329,
        -0.5731, -3.4868,  1.1259, -5.2075], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.9407, -1.3258, -3.0518, -1.3270, -1.5104, -1.2434],
        [-3.2897, -1.1903, -2.3914, -1.1107, -1.6659, -1.8019],
        [-1.8990, -1.1925, -1.2712, -1.4039, -1.7192, -3.6835],
        [-1.1394, -1.3104, -1.4568, -1.0636, -1.7414, -1.4391],
        [-4.0403, -2.6135, -1.6149, -1.0110, -2.4437, -1.3314],
        [-4.2460, -2.9582, -2.1968, -2.9084, -2.1047, -1.4230],
        [-1.6886, -1.3405, -1.2840, -1.0555, -1.3034, -1.0572],
        [-1.8191, -1.4807, -2.1537, -1.2529, -1.1846, -0.9502],
        [-3.5867, -1.3199, -2.2004, -1.2010, -1.7910, -1.0878],
        [-2.1176, -1.0979, -1.0615, -1.9176, -2.1483, -2.7358],
        [-3.2299, -2.5066, -1.8624, -1.5010, -1.9449, -1.1484],
        [-3.0731, -2.7798, -2.0080, -1.7412, -1.8012, -1.2825],
        [-1.4895, -1.1674, -1.2399, -0.9749, -1.7913, -1.0255],
        [-1.8003, -1.2161, -2.5810, -1.2971, -2.1645, -3.5963],
        [-4.1472, -1.3769, -2.0236, -1.2790, -2.0730, -1.6112],
        [-2.6228, -1.1311, -1.4472, -2.7865, -1.9479, -1.6310],
        [-1.9006, -2.7780, -2.7980, -2.0623, -1.7390, -1.9892],
        [-3.6326, -2.9133, -1.7588, -0.9632, -2.3451, -1.5353],
        [-1.5222, -1.1293, -1.7893, -0.9985, -1.9510, -1.0593],
        [-1.3282, -1.5179, -2.4039, -1.4085, -1.4530, -0.8404]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.3990, -11.4500, -11.1692,  -8.1507, -13.0548, -15.8371,  -7.7292,
         -8.8411, -11.1868, -11.0788, -12.1932, -12.6858,  -7.6885, -12.6553,
        -12.5108, -11.5666, -13.2671, -13.1482,  -8.4497,  -8.9519],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.8726e+00, -4.0105e-02, -3.6210e+00, -2.4500e-01, -1.0375e+00,
         -4.4239e-01],
        [-3.9412e+00, -1.4095e-01, -3.0221e+00, -3.1411e-02, -1.8154e+00,
         -1.6659e+00],
        [-1.9352e+00, -4.3802e-03, -1.3630e-01, -6.2574e-01, -1.8534e+00,
         -3.7631e+00],
        [-2.7215e-01, -1.8259e-02, -9.1586e-01, -6.5379e-03, -1.4628e+00,
         -7.7587e-01],
        [-3.8316e+00, -3.2797e+00, -8.9748e-01, -3.2091e-01, -3.4410e+00,
         -1.3804e-01],
        [-4.6252e+00, -3.0200e+00, -2.3705e+00, -2.3104e+00, -2.6215e+00,
         -8.7603e-01],
        [-1.9235e+00, -5.1581e-01, -1.9129e-02, -3.5572e-04, -3.8207e-01,
         -1.1500e-01],
        [-2.1693e+00, -2.9826e-01, -2.5009e+00, -2.1156e-01, -3.8647e-01,
         -1.5761e-01],
        [-3.9354e+00, -2.2570e-01, -2.4740e+00, -3.4323e-01, -2.0952e+00,
         -7.5197e-04],
        [-2.2477e+00, -4.8359e-02, -2.5908e-02, -1.7890e+00, -2.5190e+00,
         -3.3329e+00],
        [-2.7124e+00, -2.7604e+00, -2.2187e+00, -1.0871e+00, -2.1398e+00,
         -6.6696e-01],
        [-3.3489e+00, -3.1948e+00, -2.1906e+00, -5.2284e-01, -1.5902e+00,
         -2.9782e-01],
        [-1.5618e+00, -1.0323e-01, -1.7464e-02, -2.6491e-01, -1.8443e+00,
         -5.6748e-02],
        [-2.3228e+00, -1.6691e-01, -3.1947e+00, -3.0271e-01, -2.4052e+00,
         -2.5296e+00],
        [-4.7848e+00, -1.1921e-07, -1.9550e+00, -4.5575e-01, -2.9436e+00,
         -1.1177e+00],
        [-2.6917e+00, -3.0477e-03, -1.2303e+00, -3.1269e+00, -2.3718e+00,
         -9.5375e-01],
        [-1.4864e+00, -3.7410e+00, -4.8140e+00, -2.8140e+00, -1.3740e+00,
         -2.0841e+00],
        [-2.9724e+00, -2.7598e+00, -1.2372e+00, -4.6849e-05, -3.1531e+00,
         -9.5605e-01],
        [-1.9012e+00, -2.9271e-01, -1.4087e+00, -1.5369e-01, -2.1292e+00,
         -9.5310e-03],
        [-1.3333e+00, -2.1034e-01, -3.2006e+00, -2.6037e-01, -5.9467e-01,
         -1.3981e-01]], device='cuda:0')
tensor([-10.2586, -10.6170,  -8.3181,  -3.4515, -11.9088, -15.8236,  -2.9558,
         -5.7241,  -9.0743,  -9.9629, -11.5854, -11.1452,  -3.8485, -10.9219,
        -11.2569, -10.3775, -16.3135, -11.0785,  -5.8950,  -5.7391],
       device='cuda:0')
tensor([-2.1404, -0.8330, -2.8511, -4.6992, -1.1460, -0.0134, -4.7734, -3.1171,
        -2.1125, -1.1159, -0.6078, -1.5405, -3.8400, -1.7334, -1.2540, -1.1892,
         3.0463, -2.0697, -2.5547, -3.2129], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.7155, -1.9729, -1.7269, -1.8545, -4.2978, -2.1851],
        [-2.4078, -1.9341, -1.5938, -1.0082, -1.8361, -1.3156],
        [-1.4577, -2.1059, -1.7605, -1.8734, -1.3215, -1.1629],
        [-1.7414, -1.6046, -2.0100, -1.6009, -1.1389, -1.3399],
        [-1.6363, -1.2622, -2.3842, -1.0181, -1.1022, -1.4685],
        [-1.3668, -1.4479, -2.3463, -1.4995, -1.1818, -1.1093],
        [-2.1299, -1.1024, -1.3772, -1.1361, -1.2942, -1.4878],
        [-1.3412, -1.7954, -1.5827, -1.0761, -1.3367, -2.6852],
        [-2.8553, -3.3406, -1.4686, -0.9261, -2.1927, -1.5920],
        [-2.0234, -1.1740, -2.9639, -1.6853, -2.9247, -1.4797],
        [-1.4337, -3.1850, -1.6604, -1.6211, -1.3243, -1.5315],
        [-1.7372, -1.5098, -2.7453, -1.3471, -1.6084, -0.8978],
        [-2.4160, -0.9028, -1.5233, -1.1735, -2.3560, -2.0471],
        [-1.4270, -1.3877, -1.2179, -1.9687, -2.1804, -0.9109],
        [-5.5537, -2.1504, -1.8174, -1.2468, -1.6984, -1.4062],
        [-4.0257, -3.4938, -2.0751, -1.2943, -2.1339, -1.4730],
        [-1.3060, -1.5788, -2.4280, -1.1438, -1.4176, -1.1521],
        [-1.7400, -1.3662, -2.0792, -1.5597, -1.5762, -1.4852],
        [-4.6869, -1.0389, -1.9470, -1.4672, -1.6823, -1.5263],
        [-3.8124, -1.1719, -1.9548, -2.5118, -1.9510, -1.8304]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-15.7527, -10.0957,  -9.6820,  -9.4357,  -8.8716,  -8.9517,  -8.5276,
         -9.8173, -12.3753, -12.2509, -10.7559,  -9.8457, -10.4187,  -9.0927,
        -13.8728, -14.4958,  -9.0263,  -9.8066, -12.3486, -13.2323],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.1999e+00, -1.1268e+00, -1.3882e+00, -2.4956e+00, -6.1640e+00,
         -2.7713e+00],
        [-1.3866e+00, -1.0466e+00, -9.4162e-01, -9.1683e-02, -1.7782e+00,
         -4.1181e-02],
        [-5.7577e-02, -1.4724e+00, -1.3083e+00, -1.4495e+00, -7.6788e-01,
         -8.1215e-01],
        [-1.5529e+00, -7.8087e-01, -2.1602e+00, -1.2779e+00, -1.6062e-02,
         -5.8963e-01],
        [-1.7422e+00, -1.2234e-01, -3.0821e+00, -4.5502e-03, -1.6984e-01,
         -5.4169e-01],
        [-1.0311e+00, -2.7063e-01, -2.6029e+00, -7.2211e-01, -1.0914e-02,
         -1.8476e-01],
        [-2.9731e+00, -8.3379e-02, -5.1243e-01, -2.2172e-01, -3.6599e-01,
         -6.2554e-01],
        [-1.7162e-01, -1.2280e+00, -1.3896e+00, -6.5362e-04, -8.7931e-02,
         -2.3452e+00],
        [-2.3516e+00, -4.4741e+00, -6.5713e-01, -6.6293e-03, -2.7866e+00,
         -1.1570e+00],
        [-1.2797e+00, -6.1395e-02, -3.1130e+00, -1.2328e+00, -3.4955e+00,
         -9.8169e-01],
        [-1.4384e-01, -5.1169e+00, -1.1079e+00, -8.0848e-01, -7.5196e-01,
         -1.1069e+00],
        [-2.0465e+00, -3.3204e-01, -3.4850e+00, -3.3781e-01, -1.4841e+00,
         -7.8050e-02],
        [-2.8852e+00, -3.6526e-01, -7.4642e-01, -2.3924e-01, -3.0306e+00,
         -2.7967e+00],
        [-1.6331e+00, -2.5209e-01, -5.8316e-01, -1.7965e+00, -2.4120e+00,
         -1.1014e-02],
        [-5.4830e+00, -2.0230e+00, -1.7172e+00, -7.6826e-01, -1.2813e+00,
         -4.1195e-01],
        [-4.4921e+00, -3.9184e+00, -2.2979e+00, -1.7668e-01, -2.6934e+00,
         -1.0977e+00],
        [-6.7363e-01, -1.4374e+00, -2.7415e+00, -2.4049e-01, -7.9584e-01,
         -3.9133e-01],
        [-2.1085e+00, -1.7088e-02, -2.2556e+00, -7.8399e-01, -1.2564e+00,
         -5.3121e-01],
        [-4.9085e+00, -1.5417e-01, -1.7125e+00, -8.8596e-01, -2.0473e+00,
         -7.7419e-01],
        [-3.9696e+00, -7.1700e-02, -1.9591e+00, -2.3801e+00, -2.3017e+00,
         -2.0118e+00]], device='cuda:0')
tensor([-18.1458,  -5.2859,  -5.8678,  -6.3776,  -5.6627,  -4.8224,  -4.7822,
         -5.2230, -11.4331, -10.1641,  -9.0360,  -7.7634, -10.0633,  -6.6879,
        -11.6848, -14.6761,  -6.2802,  -6.9528, -10.4826, -12.6940],
       device='cuda:0')
tensor([ 2.3931, -4.8098, -3.8142, -3.0582, -3.2088, -4.1293, -3.7454, -4.5942,
        -0.9422, -2.0868, -1.7199, -2.0823, -0.3554, -2.4048, -2.1880,  0.1803,
        -2.7461, -2.8538, -1.8660, -0.5383], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.9207, -2.2026, -2.0256, -3.1065, -2.5502, -2.0669],
        [-2.3282, -4.5802, -2.0700, -1.0942, -2.3769, -1.2203],
        [-1.3541, -1.7235, -1.6455, -1.4529, -1.2483, -1.1225],
        [-1.9974, -1.3370, -3.2161, -1.5910, -1.3785, -1.1107],
        [-2.3580, -1.2433, -2.4716, -1.1554, -1.7370, -1.0571],
        [-2.3764, -1.0313, -3.1177, -1.1335, -2.1551, -2.7140],
        [-3.9871, -1.1821, -1.6986, -1.1597, -2.0243, -2.3843],
        [-4.0325, -2.6652, -2.0234, -1.1470, -1.7078, -1.3274],
        [-2.2488, -2.9447, -1.6438, -1.2701, -2.7032, -1.4856],
        [-1.8239, -1.1772, -1.4481, -1.0116, -1.4996, -1.7947],
        [-3.6122, -1.2473, -2.2426, -1.2159, -1.9273, -2.3364],
        [-2.4372, -1.4436, -1.8130, -1.6655, -1.5908, -0.9285],
        [-1.3466, -1.2725, -1.3232, -2.4286, -2.3217, -1.3525],
        [-6.1416, -2.9813, -1.9171, -1.6712, -2.1487, -1.7095],
        [-3.0113, -2.6589, -1.9880, -1.9388, -1.8873, -1.4499],
        [-1.3852, -1.2572, -1.4030, -0.9229, -1.3101, -1.1238],
        [-2.0476, -1.3831, -4.3490, -1.3353, -1.7302, -1.5270],
        [-1.9627, -1.1441, -2.3574, -1.0700, -1.9446, -1.8083],
        [-4.7356, -1.2289, -1.6070, -1.2012, -1.8535, -1.9378],
        [-1.3181, -1.2312, -1.9228, -1.1752, -2.4068, -1.5984]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-15.8725, -13.6698,  -8.5467, -10.6307, -10.0224, -12.5280, -12.4360,
        -12.9032, -12.2963,  -8.7551, -12.5817,  -9.8786, -10.0451, -16.5694,
        -12.9342,  -7.4023, -12.3723, -10.2872, -12.5639,  -9.6523],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.3857e+00, -2.2505e+00, -2.6553e+00, -3.3729e+00, -3.2330e+00,
         -2.2471e+00],
        [-1.7479e+00, -5.6004e+00, -2.4760e+00, -1.9286e-02, -3.3065e+00,
         -8.4017e-02],
        [-8.7072e-01, -1.7696e+00, -1.1717e+00, -8.5830e-01, -4.3815e-02,
         -2.9935e-01],
        [-2.4179e+00, -1.2267e-04, -4.4359e+00, -9.5178e-01, -8.3100e-01,
         -1.8525e-01],
        [-2.5890e+00, -3.2401e-02, -2.9118e+00, -1.7232e-03, -2.0605e+00,
         -2.1572e-01],
        [-2.8409e+00, -3.9910e-01, -3.4202e+00, -8.4826e-02, -2.4659e+00,
         -3.1637e+00],
        [-4.5629e+00, -5.0702e-02, -1.6563e+00, -1.4812e-01, -2.0881e+00,
         -2.0715e+00],
        [-4.2765e+00, -3.3060e+00, -2.2942e+00, -1.7734e-01, -1.1566e+00,
         -5.6028e-05],
        [-1.5142e+00, -3.0651e+00, -9.6792e-01, -3.1800e-01, -3.9999e+00,
         -1.0244e+00],
        [-2.4822e+00, -1.0517e-02, -8.4219e-01, -3.1209e-04, -8.1219e-01,
         -1.5440e+00],
        [-4.6782e+00, -1.6002e-02, -2.8595e+00, -9.5129e-04, -1.8079e+00,
         -1.8379e+00],
        [-2.7655e+00, -2.8802e-02, -1.2053e+00, -1.4180e+00, -1.6471e+00,
         -5.4002e-05],
        [-9.3193e-01, -2.0040e-01, -8.2715e-01, -2.4078e+00, -2.9996e+00,
         -7.2347e-01],
        [-5.6293e+00, -3.9804e+00, -2.2689e+00, -1.6798e+00, -2.7425e+00,
         -1.5065e+00],
        [-3.0665e+00, -2.7568e+00, -2.0827e+00, -8.5416e-01, -1.9449e+00,
         -1.0660e+00],
        [-1.4507e-02, -5.3590e-01, -5.0251e-01, -1.2801e-01, -3.8231e-01,
         -1.9972e-01],
        [-3.1773e+00, -6.3564e-02, -5.1674e+00, -1.2298e-01, -1.5768e+00,
         -4.0349e-01],
        [-2.0072e+00, -6.5960e-03, -2.7666e+00, -1.5891e-02, -2.4218e+00,
         -2.0001e+00],
        [-4.5187e+00, -5.5203e-01, -8.8660e-01, -3.1684e-01, -2.1861e+00,
         -2.0637e+00],
        [-1.0263e+00, -4.5942e-03, -2.2271e+00, -8.6506e-02, -2.6931e+00,
         -8.9070e-01]], device='cuda:0')
tensor([-17.1446, -13.2342,  -5.0134,  -8.8220,  -7.8112, -12.3747, -10.5777,
        -11.2108, -10.8895,  -5.6913, -11.2005,  -7.0647,  -8.0904, -17.8074,
        -11.7711,  -1.7630, -10.5116,  -9.2182, -10.5240,  -6.9283],
       device='cuda:0')
tensor([ 1.2721, -0.4356, -3.5333, -1.8087, -2.2112, -0.1533, -1.8584, -1.6925,
        -1.4068, -3.0638, -1.3813, -2.8139, -1.9548,  1.2380, -1.1631, -5.6394,
        -1.8607, -1.0690, -2.0399, -2.7240], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.5707, -1.8673, -1.3704, -1.9760, -1.2966, -1.2079],
        [-2.0423, -1.5920, -2.3975, -2.3817, -1.8657, -1.4290],
        [-2.2252, -0.9649, -2.4934, -1.2677, -2.9172, -2.2832],
        [-5.2788, -2.4265, -2.2567, -1.5139, -2.2763, -2.6680],
        [-2.3890, -2.4519, -2.0738, -2.9324, -1.8714, -2.4146],
        [-1.3765, -1.1719, -1.6156, -1.0804, -1.3887, -1.1090],
        [-2.2706, -1.2108, -2.5958, -1.4886, -2.3857, -3.8986],
        [-3.2664, -1.3693, -2.3637, -1.2253, -1.7170, -0.9428],
        [-2.3209, -1.2671, -1.1564, -1.6917, -2.1159, -1.2040],
        [-5.2372, -1.9718, -2.1522, -1.6357, -2.3091, -1.8079],
        [-2.8175, -2.7632, -2.2675, -1.1088, -2.2779, -1.5183],
        [-1.4165, -1.3208, -1.3084, -1.2644, -1.5499, -1.1495],
        [-1.7699, -1.2216, -1.5964, -2.0885, -1.3804, -1.0990],
        [-4.0219, -1.4414, -1.3297, -1.3190, -1.4112, -1.0949],
        [-2.9448, -1.2558, -1.2254, -1.6848, -1.1031, -1.6026],
        [-1.3338, -1.6572, -1.6536, -2.4379, -1.3483, -1.2011],
        [-2.9820, -1.9976, -2.2927, -3.1179, -2.2737, -1.5077],
        [-2.6824, -2.8112, -2.3199, -1.5486, -2.7431, -1.9811],
        [-4.7303, -3.1408, -1.6463, -1.3066, -1.6856, -1.8949],
        [-3.0385, -2.3928, -2.3609, -1.4445, -1.9483, -2.4920]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.2890, -11.7082, -12.1516, -16.4202, -14.1331,  -7.7421, -13.8501,
        -10.8846,  -9.7560, -15.1140, -12.7533,  -8.0096,  -9.1557, -10.6181,
         -9.8164,  -9.6319, -14.1716, -14.0863, -14.4045, -13.6769],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.9595e+00, -2.2762e+00, -7.5985e-01, -1.3420e+00, -8.1188e-01,
         -1.2727e-01],
        [-2.2122e+00, -8.1712e-01, -2.9189e+00, -2.6377e+00, -1.6871e+00,
         -9.8458e-01],
        [-2.9860e+00, -3.2279e-01, -2.8740e+00, -4.0934e-01, -3.3083e+00,
         -2.4869e+00],
        [-5.3629e+00, -2.0388e+00, -3.1366e+00, -8.6887e-01, -3.2319e+00,
         -3.0291e+00],
        [-1.7001e+00, -2.6211e+00, -2.3384e+00, -1.8037e+00, -1.6701e+00,
         -3.2474e+00],
        [-9.2578e-01, -1.1750e-01, -1.0298e+00, -5.9582e-01, -5.3776e-01,
         -3.1939e-02],
        [-3.3955e+00, -1.1613e-01, -2.9936e+00, -7.3303e-01, -2.8849e+00,
         -2.6360e+00],
        [-4.0243e+00, -7.1863e-01, -2.8187e+00, -1.4178e-01, -1.9599e+00,
         -2.4455e-01],
        [-2.3513e+00, -6.0831e-01, -9.6311e-02, -1.1750e+00, -2.8208e+00,
         -3.3631e-01],
        [-4.9363e+00, -1.5010e+00, -2.9805e+00, -1.5448e+00, -2.9834e+00,
         -1.6965e+00],
        [-2.3801e+00, -3.2346e+00, -2.6211e+00, -5.3385e-02, -3.4733e+00,
         -8.6089e-01],
        [-1.5244e+00, -1.2362e-01, -2.5789e-01, -3.0570e-01, -9.7277e-01,
         -2.2650e-06],
        [-1.8751e+00, -4.3203e-02, -1.0889e+00, -1.8325e+00, -6.8665e-01,
         -4.1677e-01],
        [-3.4347e+00, -2.0882e-01, -1.6747e-01, -2.8303e-01, -9.4348e-02,
         -3.7825e-02],
        [-2.3584e+00, -1.0131e+00, -8.2900e-01, -1.1793e+00, -5.5065e-02,
         -1.2583e+00],
        [-1.0416e+00, -6.9430e-01, -1.4835e+00, -1.9563e+00, -6.1899e-01,
         -2.8147e-02],
        [-3.5013e+00, -1.6019e+00, -2.5927e+00, -3.3339e+00, -2.2027e+00,
         -1.1533e+00],
        [-3.7523e+00, -1.8663e+00, -2.3773e+00, -7.9823e-01, -3.2524e+00,
         -2.2932e+00],
        [-4.9353e+00, -2.7624e+00, -7.8861e-01, -8.1601e-02, -1.2811e+00,
         -1.8248e+00],
        [-3.3325e+00, -2.5291e+00, -3.3049e+00, -8.5972e-02, -1.9193e+00,
         -3.2036e+00]], device='cuda:0')
tensor([ -7.2767, -11.2577, -12.3874, -17.6681, -13.3809,  -3.2386, -12.7591,
         -9.9079,  -7.3881, -15.6425, -12.6233,  -3.1844,  -5.9431,  -4.2262,
         -6.6932,  -5.8229, -14.3857, -14.3397, -11.6738, -14.3753],
       device='cuda:0')
tensor([-2.0123, -0.4505,  0.2358,  1.2479, -0.7522, -4.5035, -1.0910, -0.9767,
        -2.3679,  0.5285, -0.1300, -4.8252, -3.2126, -6.3919, -3.1233, -3.8090,
         0.2141,  0.2534, -2.7306,  0.6984], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.1884, -0.9928, -1.6180, -1.0379, -2.3961, -4.1872],
        [-1.1067, -1.5287, -1.6042, -1.1932, -1.3168, -1.6586],
        [-4.4432, -3.6833, -1.9725, -1.0361, -2.1250, -1.7701],
        [-3.4893, -4.6860, -2.3609, -1.3274, -1.5685, -1.9144],
        [-1.4086, -1.9357, -1.3665, -1.4899, -1.3054, -1.0713],
        [-1.9500, -1.2955, -2.7317, -1.2355, -2.9035, -3.1541],
        [-2.5802, -1.1209, -2.2585, -1.0982, -2.2087, -1.2028],
        [-4.2569, -1.0639, -1.6980, -1.6720, -1.9601, -1.1406],
        [-4.9876, -2.3662, -1.8661, -1.2527, -2.0203, -1.9596],
        [-3.0026, -2.6137, -2.2085, -1.3332, -2.4717, -1.9387],
        [-1.3203, -1.0670, -2.2675, -0.8386, -1.3606, -1.2747],
        [-1.2152, -1.1710, -2.8230, -1.2398, -1.3976, -1.2958],
        [-8.0501, -1.4586, -1.6735, -1.2418, -1.5916, -0.9806],
        [-2.6942, -1.0607, -3.1916, -1.1790, -1.8430, -2.8965],
        [-2.2405, -1.1134, -1.7165, -1.1462, -1.9145, -1.9536],
        [-3.4614, -2.6508, -2.0150, -1.1406, -2.1978, -1.2345],
        [-2.4667, -2.5826, -2.3844, -1.6961, -2.6622, -1.5458],
        [-1.3238, -1.8444, -1.9949, -1.8250, -1.3799, -1.2165],
        [-1.8178, -1.2989, -2.0964, -1.2570, -1.3833, -1.1300],
        [-2.4645, -1.2964, -1.5024, -1.2234, -1.2489, -1.0665]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-14.4204,  -8.4081, -15.0302, -15.3464,  -8.5774, -13.2704, -10.4692,
        -11.7915, -14.4525, -13.5684,  -8.1287,  -9.1424, -14.9963, -12.8651,
        -10.0848, -12.7001, -13.3378,  -9.5845,  -8.9834,  -8.8021],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.9859e+00, -8.9005e-03, -1.2052e+00, -6.0979e-02, -2.2525e+00,
         -4.0383e+00],
        [-1.4772e-01, -5.9359e-01, -1.3781e+00, -5.5483e-01, -3.3818e-01,
         -1.2288e+00],
        [-4.4902e+00, -5.0218e+00, -2.3726e+00, -1.6123e-03, -2.7534e+00,
         -1.5723e+00],
        [-3.0433e+00, -4.3128e+00, -2.3492e+00, -6.9400e-01, -1.1867e+00,
         -1.7271e+00],
        [-1.0696e+00, -2.3732e+00, -5.5112e-01, -6.5847e-01, -2.6681e-03,
         -1.0530e-01],
        [-2.5182e+00, -1.4865e-04, -3.6942e+00, -5.0282e-02, -3.9245e+00,
         -2.2380e+00],
        [-3.0369e+00, -2.6582e-01, -2.5852e+00, -2.6030e-02, -2.9538e+00,
         -4.7890e-01],
        [-4.2630e+00, -1.0810e-02, -1.6272e+00, -1.2082e+00, -2.3398e+00,
         -3.4070e-03],
        [-5.7377e+00, -2.5097e+00, -1.4512e+00, -5.8373e-01, -2.4893e+00,
         -2.3160e+00],
        [-3.0101e+00, -2.6674e+00, -2.6050e+00, -2.0261e-01, -3.6979e+00,
         -2.2126e+00],
        [-2.6320e-01, -1.9060e-02, -2.1669e+00, -8.8552e-03, -1.1863e+00,
         -7.7460e-01],
        [-7.3135e-01, -1.3921e-01, -3.4968e+00, -7.7241e-02, -6.5925e-01,
         -5.0774e-01],
        [-6.9366e+00, -1.0452e-01, -8.7336e-01, -6.7919e-02, -1.6349e+00,
         -1.9898e-01],
        [-3.2800e+00, -2.3603e-01, -3.4040e+00, -2.7247e-01, -1.8884e+00,
         -3.5009e+00],
        [-2.5569e+00, -1.4931e-01, -1.6699e+00, -4.4993e-02, -1.9660e+00,
         -1.6721e+00],
        [-3.5564e+00, -3.2703e+00, -2.4376e+00, -1.1241e-01, -3.1450e+00,
         -1.0562e-02],
        [-2.0064e+00, -2.5030e+00, -2.8193e+00, -8.3832e-01, -3.9999e+00,
         -1.1852e+00],
        [-8.9001e-01, -2.1442e+00, -2.1846e+00, -1.1499e+00, -2.3489e-03,
         -1.3214e-01],
        [-2.2517e+00, -1.9280e-01, -2.2875e+00, -4.4727e-02, -7.3031e-01,
         -2.0739e-01],
        [-1.8975e+00, -3.0406e-03, -6.1273e-01, -3.3086e-01, -4.1459e-03,
         -5.5375e-02]], device='cuda:0')
tensor([-12.5518,  -4.2412, -16.2120, -13.3131,  -4.7604, -12.4253,  -9.3467,
         -9.4525, -15.0877, -14.3956,  -4.4189,  -5.6116,  -9.8163, -12.5819,
         -8.0591, -12.5323, -13.3521,  -6.5033,  -5.7145,  -2.9037],
       device='cuda:0')
tensor([-1.8685, -4.1669,  1.1818, -2.0333, -3.8170, -0.8450, -1.1225, -2.3391,
         0.6352,  0.8271, -3.7098, -3.5309, -5.1800, -0.2832, -2.0257, -0.1679,
         0.0143, -3.0813, -3.2689, -5.8984], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.9077, -1.3328, -1.6400, -2.3812, -1.6885, -1.5430],
        [-3.2421, -1.6527, -1.6316, -1.2202, -2.3394, -1.3510],
        [-2.9161, -2.3155, -1.8575, -1.6108, -1.8878, -1.1906],
        [-1.3398, -1.6419, -2.6488, -0.9288, -1.2671, -1.0395],
        [-2.8050, -1.5001, -2.3358, -1.2494, -1.8246, -0.8906],
        [-2.0228, -0.9900, -3.2393, -1.2819, -1.5402, -1.3841],
        [-2.1099, -1.5215, -1.6707, -1.8714, -1.4281, -1.1464],
        [-4.0669, -2.1118, -1.7129, -1.2112, -1.7554, -1.1464],
        [-4.8606, -2.2091, -2.9086, -1.1265, -1.6772, -1.7802],
        [-1.5208, -1.1777, -1.2409, -1.0684, -1.3662, -1.9957],
        [-2.5370, -1.1899, -1.5771, -1.1648, -1.2708, -1.3614],
        [-1.9208, -1.5884, -2.3946, -2.6825, -1.4558, -0.9547],
        [-1.8831, -3.2085, -1.4058, -2.1622, -1.6825, -1.5510],
        [-1.9922, -2.3163, -1.6138, -1.5978, -2.1918, -1.2684],
        [-2.6258, -4.2578, -1.8544, -1.8942, -1.8152, -1.7131],
        [-1.2810, -2.5489, -2.4572, -2.1903, -1.3585, -1.1583],
        [-2.8740, -1.3259, -1.8536, -1.3540, -1.3869, -0.9095],
        [-1.4996, -1.2132, -3.0630, -1.3765, -1.3133, -1.0468],
        [-1.8420, -0.9060, -1.2992, -1.1769, -2.1335, -2.1107],
        [-3.9928, -1.8541, -2.8358, -1.2361, -2.4493, -1.6134]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.4932, -11.4370, -11.7782,  -8.8658, -10.6055, -10.4584,  -9.7480,
        -12.0047, -14.5623,  -8.3696,  -9.1010, -10.9967, -11.8931, -10.9802,
        -14.1605, -10.9943,  -9.7040,  -9.5125,  -9.4683, -13.9815],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.3488, -0.3513, -1.2763, -2.0190, -1.3873, -1.1042],
        [-2.9889, -0.4693, -0.7702, -0.6378, -3.2495, -0.1612],
        [-2.7923, -2.2810, -1.6822, -0.4820, -1.9192, -0.0843],
        [-0.8741, -1.4921, -2.9539, -0.0611, -0.4922, -0.2032],
        [-3.6833, -0.3606, -2.9441, -0.0579, -1.9200, -0.0170],
        [-2.2199, -0.0812, -3.9349, -0.4357, -1.3688, -0.2440],
        [-2.2125, -0.5394, -1.6101, -1.4835, -1.1540, -0.3671],
        [-3.7578, -1.7946, -0.3561, -0.5839, -1.6132, -0.0400],
        [-4.6093, -1.9493, -3.8212, -0.0870, -1.3664, -1.7441],
        [-1.7283, -0.0231, -0.1838, -0.1119, -0.0359, -1.9372],
        [-3.1530, -0.2039, -1.0239, -0.0343, -0.5382, -0.6622],
        [-2.3359, -0.3437, -2.7362, -3.0548, -0.9368, -0.1384],
        [-1.9400, -2.2132, -0.9388, -1.9076, -1.5866, -1.4319],
        [-1.2304, -2.4968, -0.7734, -1.5324, -2.8003, -0.0063],
        [-2.1887, -4.8916, -1.6795, -0.9178, -1.5841, -1.8786],
        [-0.0517, -3.4107, -3.0031, -1.4622, -0.0448, -0.2344],
        [-3.5932, -0.8578, -1.3813, -0.3279, -0.4969, -0.3717],
        [-1.2491, -0.0974, -4.1558, -0.5574, -0.6488, -0.0061],
        [-2.0427, -0.0397, -0.4540, -0.1890, -2.1973, -2.2633],
        [-3.4951, -0.9390, -5.2202, -0.5263, -3.3445, -0.7331]],
       device='cuda:0')
tensor([ -8.4869,  -8.2768,  -9.2409,  -6.0767,  -8.9829,  -8.2845,  -7.3667,
         -8.1457, -13.5773,  -4.0202,  -5.6154,  -9.5457, -10.0181,  -8.8396,
        -13.1403,  -8.2069,  -7.0288,  -6.7145,  -7.1860, -14.2582],
       device='cuda:0')
tensor([-2.0064, -3.1602, -2.5374, -2.7891, -1.6226, -2.1739, -2.3813, -3.8591,
        -0.9850, -4.3494, -3.4855, -1.4510, -1.8749, -2.1407, -1.0202, -2.7874,
        -2.6752, -2.7980, -2.2823,  0.2767], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-5.7311, -1.6426, -3.3647, -1.3250, -1.9477, -0.9620],
        [-2.9388, -1.0505, -1.1777, -1.7891, -2.1536, -1.5473],
        [-5.3700, -2.9801, -1.5054, -1.3672, -1.8149, -1.2835],
        [-2.2787, -3.2148, -2.2496, -1.2211, -1.9933, -1.4977],
        [-1.3253, -1.2377, -1.2190, -1.0249, -1.9688, -1.9027],
        [-2.8290, -1.1953, -1.5319, -1.6043, -3.6764, -2.3096],
        [-1.6759, -1.5551, -2.2154, -1.5032, -1.6819, -0.8430],
        [-1.3718, -1.4666, -1.7041, -2.8950, -1.7178, -1.4022],
        [-2.1409, -2.3009, -2.6090, -3.3954, -2.7629, -2.5648],
        [-2.7459, -3.0331, -2.3100, -1.1312, -2.1319, -1.2708],
        [-1.2998, -1.3503, -2.4889, -1.3333, -1.3035, -1.2010],
        [-2.5077, -1.4054, -2.7578, -1.3239, -1.5596, -0.8799],
        [-4.1049, -1.3719, -2.3334, -2.3237, -1.4271, -1.1315],
        [-2.5324, -1.9088, -1.3048, -1.4539, -1.6227, -1.2190],
        [-3.6681, -2.1894, -2.0056, -1.0556, -2.4211, -1.8128],
        [-2.5384, -2.5859, -2.7333, -3.0646, -2.3162, -1.1713],
        [-1.2778, -1.7742, -1.3028, -5.3918, -1.2296, -0.9375],
        [-2.9110, -2.0166, -2.0135, -1.3339, -2.1921, -1.0583],
        [-4.7512, -0.9823, -3.5329, -1.0400, -2.5032, -2.2213],
        [-1.2646, -2.0026, -1.5361, -2.6973, -1.5381, -1.0913]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-14.9730, -10.6570, -14.3212, -12.4552,  -8.6784, -13.1464,  -9.4746,
        -10.5575, -15.7740, -12.6229,  -8.9768, -10.4343, -12.6925, -10.0415,
        -13.1526, -14.4097, -11.9136, -11.5254, -15.0309, -10.1301],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-5.7342e+00, -5.0762e-01, -4.7869e+00, -3.7499e-01, -2.7477e+00,
         -1.1846e-01],
        [-3.5389e+00, -1.5163e-04, -5.1357e-03, -1.3843e+00, -2.5588e+00,
         -1.2466e+00],
        [-4.8519e+00, -3.4799e+00, -1.1503e+00, -8.5412e-01, -1.6274e+00,
         -5.7082e-01],
        [-1.4890e+00, -3.9959e+00, -3.0255e+00, -4.1432e-02, -2.2964e+00,
         -1.0428e+00],
        [-7.2559e-01, -1.1921e-07, -2.2937e-03, -5.5685e-02, -2.2294e+00,
         -1.8320e+00],
        [-4.0367e+00, -3.3365e-01, -9.4459e-01, -9.5068e-01, -4.2297e+00,
         -1.6249e+00],
        [-2.0877e+00, -3.9628e-02, -2.4518e+00, -8.2739e-01, -1.7157e+00,
         -1.7243e-02],
        [-9.1607e-01, -4.0139e-01, -1.7568e+00, -3.0615e+00, -1.8340e+00,
         -9.1674e-01],
        [-1.2568e+00, -2.4474e+00, -4.2075e+00, -3.8238e+00, -3.9662e+00,
         -3.0478e+00],
        [-2.3785e+00, -3.2874e+00, -2.9029e+00, -4.8288e-03, -2.6248e+00,
         -3.0182e-01],
        [-2.6396e-01, -8.8458e-01, -3.0176e+00, -4.4091e-01, -2.4952e-02,
         -7.8658e-02],
        [-3.2068e+00, -8.4196e-01, -3.6980e+00, -2.3475e-01, -1.3454e+00,
         -2.4915e-05],
        [-4.3363e+00, -6.3447e-02, -2.8372e+00, -2.2904e+00, -1.2140e+00,
         -2.3316e-02],
        [-2.8457e+00, -1.0913e+00, -5.3717e-01, -7.3920e-01, -1.4564e+00,
         -5.2509e-01],
        [-3.3449e+00, -2.0965e+00, -2.5776e+00, -2.3754e-01, -3.1667e+00,
         -1.3399e+00],
        [-1.6828e+00, -2.5616e+00, -3.7271e+00, -2.3523e+00, -2.8322e+00,
         -4.2598e-02],
        [-5.6310e-03, -1.8845e+00, -2.8517e-01, -3.9869e+00, -2.1613e-01,
         -2.1522e-02],
        [-4.0396e+00, -1.0481e+00, -2.3530e+00, -1.3066e-01, -2.8065e+00,
         -2.4356e-01],
        [-4.8732e+00, -3.2372e-01, -4.5603e+00, -6.8818e-03, -3.3340e+00,
         -3.0249e+00],
        [-1.2366e+00, -1.4561e+00, -1.3111e+00, -2.9632e+00, -1.2067e+00,
         -1.6678e-01]], device='cuda:0')
tensor([-14.2699,  -8.7339, -12.5345, -11.8909,  -4.8449, -12.1202,  -7.1395,
         -8.8865, -18.7495, -11.5002,  -4.7107,  -9.3269, -10.7646,  -7.1948,
        -12.7632, -13.1985,  -6.3999, -10.6214, -16.1230,  -8.3405],
       device='cuda:0')
tensor([-0.7031, -1.9231, -1.7867, -0.5643, -3.8335, -1.0263, -2.3351, -1.6710,
         2.9755, -1.1226, -4.2661, -1.1074, -1.9279, -2.8467, -0.3894, -1.2111,
        -5.5138, -0.9040,  1.0920, -1.7896], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.2443, -1.7639, -2.2082, -1.4364, -1.2904, -1.0133],
        [-3.4176, -1.2278, -2.6722, -1.2375, -1.2452, -1.2828],
        [-3.5577, -1.1357, -1.8025, -1.4337, -1.5565, -1.1798],
        [-1.8451, -1.6598, -1.7114, -4.4279, -1.6162, -1.4533],
        [-4.5384, -2.0819, -1.7136, -1.5762, -1.7125, -1.3191],
        [-2.0487, -3.3814, -1.6257, -1.3299, -1.7489, -1.4446],
        [-1.2379, -1.5105, -1.4215, -1.0826, -1.3604, -0.9955],
        [-3.7165, -1.2655, -2.7169, -1.5546, -2.3307, -1.5388],
        [-2.7536, -1.3424, -1.8459, -1.1982, -2.1915, -1.1008],
        [-1.4486, -1.2326, -2.3310, -3.7017, -1.0841, -1.0804],
        [-2.9764, -1.9980, -1.9876, -2.9113, -2.2751, -1.4074],
        [-3.3928, -3.5798, -2.6261, -1.3046, -1.9248, -1.2757],
        [-1.3659, -2.6861, -2.4848, -1.7288, -1.3117, -1.4025],
        [-2.9571, -1.4190, -2.5512, -1.7405, -1.4009, -1.4087],
        [-2.5319, -1.0846, -2.0748, -1.1012, -1.5243, -1.2823],
        [-3.3271, -1.1417, -1.6317, -1.1090, -1.5959, -3.1183],
        [-2.2375, -1.1650, -1.4549, -1.0637, -1.8591, -3.3905],
        [-2.8164, -2.9001, -1.8998, -1.1745, -1.9582, -1.5628],
        [-1.9035, -2.8770, -2.0900, -1.4206, -1.9986, -1.5085],
        [-1.2321, -1.3863, -1.6789, -1.2342, -1.3380, -1.3097]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.9566, -11.0830, -10.6659, -12.7137, -12.9417, -11.5791,  -7.6083,
        -13.1230, -10.4324, -10.8784, -13.5558, -14.1038, -10.9797, -11.4773,
         -9.5990, -11.9236, -11.1706, -12.3117, -11.7981,  -8.1792],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.6491e-01, -1.7399e+00, -2.4116e+00, -6.3101e-01, -1.3409e-02,
         -5.2448e-02],
        [-3.9271e+00, -2.0572e-01, -3.2113e+00, -9.5759e-02, -4.6366e-01,
         -5.7848e-01],
        [-3.7730e+00, -1.6726e-02, -1.6234e+00, -7.0097e-01, -1.5256e+00,
         -1.4854e-02],
        [-2.0354e+00, -7.8081e-01, -1.4939e+00, -4.0087e+00, -1.3520e+00,
         -1.1524e+00],
        [-4.0370e+00, -1.7186e+00, -8.1501e-01, -1.3013e+00, -1.4117e+00,
         -3.5265e-01],
        [-8.8429e-01, -3.7458e+00, -8.7872e-01, -2.1310e-01, -1.3752e+00,
         -1.0443e+00],
        [-4.2381e-02, -1.1318e+00, -7.6098e-01, -6.5930e-02, -9.6638e-02,
         -1.0800e-03],
        [-4.8204e+00, -7.3946e-04, -3.1370e+00, -8.4467e-01, -3.0070e+00,
         -6.6900e-01],
        [-3.0651e+00, -1.0857e+00, -1.4553e+00, -3.2012e-01, -2.9554e+00,
         -3.6910e-03],
        [-1.0627e+00, -1.0300e-01, -2.8212e+00, -3.2396e+00, -3.1418e-02,
         -5.6327e-02],
        [-2.4188e+00, -1.3978e+00, -2.3244e+00, -3.5256e+00, -3.0067e+00,
         -8.3662e-01],
        [-2.7127e+00, -3.4411e+00, -2.1988e+00, -5.1070e-01, -2.0649e+00,
         -2.1207e-01],
        [-1.0928e+00, -3.2273e+00, -2.7972e+00, -1.0482e+00, -2.5569e-01,
         -6.5352e-01],
        [-3.7227e+00, -9.9668e-02, -3.0666e+00, -1.2607e+00, -8.1340e-01,
         -1.2099e+00],
        [-3.0646e+00, -7.6641e-01, -2.3349e+00, -1.2572e-01, -1.3043e+00,
         -3.3654e-02],
        [-3.8907e+00, -2.7953e-01, -1.0407e+00, -6.5410e-02, -1.5432e+00,
         -2.9870e+00],
        [-3.2578e+00, -2.8357e-02, -1.1045e+00, -6.7383e-02, -1.5488e+00,
         -2.2538e+00],
        [-2.5669e+00, -3.5286e+00, -1.8926e+00, -4.1255e-01, -2.2556e+00,
         -7.6803e-01],
        [-8.4180e-01, -2.8358e+00, -2.0061e+00, -6.4360e-01, -2.4252e+00,
         -9.9211e-01],
        [-2.2868e-03, -8.2074e-01, -1.5062e+00, -1.9742e-01, -3.4337e-02,
         -4.8997e-01]], device='cuda:0')
tensor([ -5.1133,  -8.4820,  -7.6546, -10.8232,  -9.6362,  -8.1414,  -2.0988,
        -12.4788,  -8.8853,  -7.3142, -13.5100, -11.1402,  -9.0747, -10.1731,
         -7.6295,  -9.8065,  -8.2606, -11.4243,  -9.7446,  -3.0509],
       device='cuda:0')
tensor([-3.8433, -2.6010, -3.0113, -1.8905, -3.3055, -3.4377, -5.5095, -0.6441,
        -1.5471, -3.5641, -0.0457, -2.9636, -1.9051, -1.3042, -1.9695, -2.1171,
        -2.9100, -0.8875, -2.0536, -5.1283], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.1139, -1.2457, -1.1951, -1.2192, -2.2336, -3.3651],
        [-2.6871, -2.8602, -1.6742, -1.1143, -2.0280, -1.1576],
        [-2.5126, -3.2576, -2.1762, -1.7562, -1.9075, -1.4193],
        [-1.6517, -1.5752, -1.7552, -1.8390, -1.4409, -1.2887],
        [-2.8236, -1.3177, -2.4230, -1.2894, -1.8669, -1.3445],
        [-2.9797, -1.1226, -2.3888, -1.7262, -1.8225, -1.1040],
        [-2.6597, -1.5704, -2.4585, -4.2962, -1.7571, -1.2102],
        [-3.4358, -2.2804, -2.3402, -1.5811, -2.0143, -1.8599],
        [-2.4670, -3.3818, -2.4154, -2.6454, -2.3109, -1.3763],
        [-1.3675, -1.1559, -1.8013, -0.7738, -1.0269, -0.9617],
        [-1.7180, -1.2475, -3.1066, -1.4720, -2.4665, -1.2359],
        [-2.4186, -1.1797, -2.5940, -1.0696, -1.5411, -1.1044],
        [-1.3949, -1.0457, -1.2272, -1.5064, -1.3116, -1.3675],
        [-2.7235, -2.0149, -2.1253, -1.1587, -1.9735, -1.5085],
        [-3.5182, -2.3933, -2.0556, -1.4036, -2.1075, -1.4398],
        [-1.5626, -1.1819, -1.5084, -1.0652, -1.4857, -1.4679],
        [-2.0563, -1.4125, -2.2661, -1.2795, -1.7675, -1.2183],
        [-3.4746, -1.1618, -4.1706, -1.1269, -1.7081, -1.2497],
        [-4.3767, -1.1817, -2.2291, -1.1560, -1.5603, -1.1978],
        [-1.9287, -1.1914, -1.7655, -1.0952, -1.5973, -1.2913]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.3727, -11.5215, -13.0294,  -9.5507, -11.0652, -11.1437, -13.9520,
        -13.5117, -14.5968,  -7.0872, -11.2464,  -9.9074,  -7.8533, -11.5044,
        -12.9180,  -8.2716, -10.0003, -12.8916, -11.7015,  -8.8695],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.0269e+00, -6.9082e-01, -9.3774e-03, -4.4049e-02, -2.6517e+00,
         -1.8274e+00],
        [-2.3550e+00, -3.6345e+00, -1.2798e+00, -1.2061e-01, -2.3517e+00,
         -1.0729e-02],
        [-1.9487e+00, -3.7623e+00, -2.2740e+00, -8.1599e-01, -2.3854e+00,
         -7.1368e-01],
        [-2.0334e+00, -1.4534e+00, -1.6275e+00, -1.1308e+00, -1.3339e-01,
         -1.0661e-01],
        [-4.1714e+00, -3.7726e-03, -3.0624e+00, -9.5404e-02, -1.8928e+00,
         -4.1453e-01],
        [-3.4473e+00, -1.9953e-01, -2.8717e+00, -1.4159e+00, -2.3325e+00,
         -2.5310e-01],
        [-2.9731e+00, -6.1418e-01, -3.1433e+00, -4.7315e+00, -1.9048e+00,
         -3.2606e-01],
        [-3.5625e+00, -2.4183e+00, -3.7772e+00, -1.7906e+00, -2.3547e+00,
         -1.8142e+00],
        [-1.9530e+00, -3.7140e+00, -3.1832e+00, -1.7741e+00, -3.1467e+00,
         -8.6391e-01],
        [-4.4551e-01, -5.3745e-01, -1.2281e+00, -6.5840e-03, -5.0353e-01,
         -2.9137e-01],
        [-2.2349e+00, -1.6085e-01, -4.0976e+00, -6.3086e-01, -3.2725e+00,
         -3.4057e-01],
        [-2.8714e+00, -8.1359e-01, -3.5594e+00, -5.7552e-02, -1.7513e+00,
         -1.7536e-01],
        [-9.9959e-01, -3.8542e-02, -4.7690e-01, -8.1327e-01, -7.0898e-01,
         -7.5614e-01],
        [-2.2671e+00, -1.4546e+00, -3.0421e+00, -4.9786e-01, -2.2255e+00,
         -1.0045e+00],
        [-3.7846e+00, -2.4352e+00, -2.1087e+00, -2.8110e-01, -2.8008e+00,
         -7.3549e-01],
        [-1.7688e+00, -2.1198e-01, -9.4676e-01, -7.2058e-02, -8.3531e-01,
         -9.3609e-01],
        [-2.7409e+00, -1.3146e-01, -2.8869e+00, -4.2933e-02, -1.5449e+00,
         -3.2704e-01],
        [-3.7873e+00, -8.7757e-02, -5.5794e+00, -4.7722e-02, -2.0387e+00,
         -2.3236e-01],
        [-4.2394e+00, -2.7779e-01, -1.7568e+00, -9.6633e-02, -1.3538e+00,
         -2.8277e-01],
        [-2.4096e+00, -6.7382e-03, -1.8596e+00, -1.8399e-02, -1.1444e+00,
         -4.8671e-01]], device='cuda:0')
tensor([ -8.2503,  -9.7524, -11.9001,  -6.4851,  -9.6403, -10.5200, -13.6929,
        -15.7175, -14.6349,  -3.0126, -10.7373,  -9.2286,  -3.7934, -10.4916,
        -12.1458,  -4.7710,  -7.6742, -11.7733,  -8.0071,  -5.9254],
       device='cuda:0')
tensor([-3.1224, -1.7691, -1.1293, -3.0656, -1.4249, -0.6236, -0.2591,  2.2058,
         0.0381, -4.0746, -0.5091, -0.6788, -4.0598, -1.0128, -0.7722, -3.5005,
        -2.3261, -1.1183, -3.6943, -2.9441], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.6251, -1.4215, -2.1138, -1.3765, -1.7368, -0.9182],
        [-1.1742, -1.1233, -3.2643, -1.1095, -1.3625, -1.3985],
        [-3.8741, -1.1926, -1.5136, -1.1958, -1.9988, -1.7455],
        [-1.7198, -1.2602, -1.4064, -1.1172, -1.9079, -2.5992],
        [-3.0663, -1.8331, -1.8243, -1.0646, -1.8820, -1.3575],
        [-2.6780, -2.2669, -1.7462, -1.3045, -2.5098, -2.2771],
        [-1.5727, -1.2236, -1.2802, -1.0603, -1.9047, -1.1708],
        [-2.7163, -1.3232, -2.9121, -1.2567, -1.3595, -2.2744],
        [-2.5279, -1.4418, -1.8630, -1.3069, -1.7553, -1.2250],
        [-4.2422, -1.1160, -3.4217, -1.1055, -1.7335, -1.5845],
        [-4.8189, -1.2336, -1.3263, -1.1976, -2.0152, -1.7309],
        [-3.1358, -2.0302, -1.7835, -1.1531, -2.0899, -1.2379],
        [-2.9990, -2.6374, -2.8719, -2.2965, -2.5635, -1.2440],
        [-1.5248, -1.2364, -1.7491, -1.0641, -1.5311, -1.2314],
        [-1.6113, -1.1197, -2.5665, -1.2594, -1.2462, -1.0539],
        [-1.5382, -1.2603, -2.5553, -1.5198, -1.6304, -1.2322],
        [-5.2021, -1.0750, -3.1516, -1.1950, -1.3199, -1.4821],
        [-1.6663, -1.4304, -1.1643, -1.4436, -1.7970, -3.9832],
        [-3.6104, -2.7867, -1.4394, -1.0183, -1.6283, -1.5949],
        [-2.4376, -3.3906, -1.9130, -1.4488, -1.9984, -1.4630]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.1919,  -9.4324, -11.5205, -10.0106, -11.0277, -12.7825,  -8.2123,
        -11.8420, -10.1199, -13.2034, -12.3226, -11.4304, -14.6123,  -8.3369,
         -8.8570,  -9.7362, -13.4257, -11.4848, -12.0779, -12.6515],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.9518e+00, -3.9439e-02, -2.3402e+00, -2.4542e-01, -1.5887e+00,
         -1.9763e-03],
        [-6.7105e-01, -5.6709e-02, -4.1546e+00, -5.4240e-04, -4.4147e-01,
         -1.0593e+00],
        [-4.8163e+00, -9.8363e-01, -7.4594e-01, -1.7760e-01, -2.1275e+00,
         -1.4136e+00],
        [-1.9439e+00, -4.6802e-04, -8.8348e-01, -1.8959e-03, -1.8462e+00,
         -2.3455e+00],
        [-2.7761e+00, -9.6831e-01, -1.6885e+00, -2.3868e-01, -1.9804e+00,
         -1.6810e-01],
        [-2.3530e+00, -2.1011e+00, -1.3383e+00, -2.9162e-01, -3.6001e+00,
         -2.6416e+00],
        [-1.8018e+00, -4.5916e-02, -3.5467e-01, -4.9317e-04, -1.9865e+00,
         -6.6710e-04],
        [-3.9964e+00, -9.5049e-03, -4.0255e+00, -3.3913e-02, -4.9591e-01,
         -1.3301e+00],
        [-2.9653e+00, -1.5921e-02, -1.4046e+00, -2.7798e-01, -2.0633e+00,
         -5.9060e-01],
        [-4.6991e+00, -3.2737e-01, -3.2898e+00, -2.5536e-03, -1.7628e+00,
         -1.2120e+00],
        [-4.9282e+00, -5.1981e-01, -8.7815e-01, -1.5899e-01, -2.1130e+00,
         -1.3869e+00],
        [-2.9869e+00, -1.4015e+00, -1.2740e+00, -7.6210e-04, -2.7487e+00,
         -3.6936e-01],
        [-3.2138e+00, -2.6736e+00, -3.6812e+00, -1.4853e+00, -3.9935e+00,
         -7.5257e-03],
        [-1.8819e+00, -7.8347e-02, -1.3966e+00, -1.3849e-01, -9.1486e-01,
         -2.5488e-01],
        [-1.6466e+00, -2.4019e-02, -3.4998e+00, -2.1964e-01, -9.1807e-02,
         -2.3568e-01],
        [-1.4163e+00, -1.1505e-02, -3.0145e+00, -8.2032e-01, -1.7283e+00,
         -4.3191e-01],
        [-5.0764e+00, -1.5334e-02, -3.2308e+00, -1.8045e-01, -7.2141e-01,
         -1.1779e+00],
        [-2.1574e+00, -3.6531e-01, -1.0998e-01, -7.5180e-01, -1.4014e+00,
         -2.6242e+00],
        [-3.5555e+00, -3.3393e+00, -7.6162e-01, -8.0825e-02, -1.2067e+00,
         -1.2140e+00],
        [-1.7312e+00, -4.6608e+00, -2.1434e+00, -4.5432e-01, -2.2146e+00,
         -1.1849e+00]], device='cuda:0')
tensor([ -8.1675,  -6.3836, -10.2645,  -7.0215,  -7.8201, -12.3258,  -4.1900,
         -9.8913,  -7.3178, -11.2936,  -9.9851,  -8.7812, -15.0550,  -4.6650,
         -5.7176,  -7.4228, -10.4023,  -7.4100, -10.1579, -12.3892],
       device='cuda:0')
tensor([-2.0244, -3.0488, -1.2560, -2.9891, -3.2076, -0.4568, -4.0223, -1.9507,
        -2.8021, -1.9098, -2.3374, -2.6492,  0.4427, -3.6719, -3.1394, -2.3134,
        -3.0233, -4.0747, -1.9200, -0.2623], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.5339, -1.4521, -2.0248, -2.4361, -1.3967, -1.1995],
        [-1.8657, -1.3830, -2.0483, -1.5330, -1.4317, -0.9632],
        [-1.9653, -1.1408, -1.4324, -1.1167, -1.5521, -2.1634],
        [-3.8087, -1.9428, -2.0773, -1.6135, -1.6734, -1.1069],
        [-1.8889, -2.2756, -1.7868, -5.4066, -1.4665, -2.4807],
        [-4.7349, -1.3850, -2.0946, -3.9955, -1.9280, -2.9551],
        [-2.1694, -0.9608, -1.5760, -1.3517, -2.9407, -1.7787],
        [-5.8007, -2.1690, -2.1844, -1.4818, -2.6439, -2.0018],
        [-2.8263, -3.0683, -2.6904, -2.4425, -1.8649, -2.4518],
        [-1.3737, -1.3396, -1.4673, -0.9666, -1.6027, -1.1395],
        [-2.5908, -1.3716, -3.2632, -1.4242, -2.1628, -3.7296],
        [-2.0254, -1.1831, -2.5823, -1.1205, -2.5621, -1.2319],
        [-3.9347, -1.3317, -1.1682, -2.9217, -2.1815, -1.3336],
        [-2.4825, -2.2235, -2.2766, -1.4745, -2.7499, -2.0146],
        [-2.6436, -1.7029, -2.1863, -1.4862, -2.1144, -2.3976],
        [-1.4957, -1.1304, -1.3343, -0.8982, -1.5694, -1.1289],
        [-2.5746, -1.6254, -2.5035, -1.5327, -1.7682, -0.7843],
        [-4.0279, -1.1166, -3.0568, -1.1444, -1.6365, -1.3999],
        [-1.4573, -1.3208, -2.3670, -1.4882, -2.3843, -1.0348],
        [-1.1177, -1.2707, -1.7085, -1.1520, -1.4097, -1.0532]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.0431,  -9.2251,  -9.3706, -12.2227, -15.3051, -17.0930, -10.7773,
        -16.2817, -15.3442,  -7.8894, -14.5422, -10.7052, -12.8714, -13.2217,
        -12.5310,  -7.5569, -10.7886, -12.3822, -10.0523,  -7.7118],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.7649e+00, -1.1063e+00, -2.1565e+00, -1.9705e+00, -2.2521e-01,
         -6.7216e-02],
        [-2.3721e+00, -3.2949e-03, -2.1801e+00, -7.3464e-01, -8.1428e-01,
         -3.1613e-02],
        [-1.2505e+00, -1.5280e-02, -6.3020e-01, -4.8921e-02, -4.1703e-01,
         -1.7585e+00],
        [-3.8660e+00, -2.5768e+00, -2.4815e+00, -7.8113e-01, -1.6255e+00,
         -6.1738e-01],
        [-2.5070e+00, -3.2196e+00, -1.8273e+00, -4.3892e+00, -8.4392e-01,
         -2.2390e+00],
        [-6.2926e+00, -4.0766e-01, -2.5093e+00, -4.0659e+00, -1.6067e+00,
         -2.8131e+00],
        [-3.0715e+00, -1.0838e-01, -9.9653e-01, -5.2553e-01, -3.3416e+00,
         -1.6794e+00],
        [-5.7639e+00, -1.5173e+00, -2.7281e+00, -7.6955e-01, -3.9901e+00,
         -2.0993e+00],
        [-2.7410e+00, -3.7602e+00, -4.1007e+00, -1.2626e+00, -1.5591e+00,
         -3.3116e+00],
        [-8.5646e-01, -7.1672e-01, -5.5987e-01, -2.0358e-01, -1.2669e+00,
         -7.9163e-02],
        [-4.0136e+00, -4.9484e-02, -4.3549e+00, -4.8204e-01, -2.4298e+00,
         -2.0763e+00],
        [-2.3750e+00, -3.8049e-01, -3.2189e+00, -6.9273e-03, -3.8167e+00,
         -7.4828e-01],
        [-3.8662e+00, -2.6702e-02, -3.6864e-01, -3.0616e+00, -3.1698e+00,
         -4.7376e-01],
        [-2.5325e+00, -2.1979e+00, -3.1131e+00, -1.4883e+00, -3.9796e+00,
         -2.0249e+00],
        [-2.0771e+00, -8.0099e-01, -2.6056e+00, -3.5583e-01, -2.8131e+00,
         -3.2715e+00],
        [-1.8607e+00, -1.3059e-01, -3.9595e-01, -1.4207e-03, -1.2856e+00,
         -1.7014e-01],
        [-4.0186e+00, -5.3315e-01, -3.2037e+00, -7.5403e-01, -1.6173e+00,
         -5.5210e-02],
        [-4.6260e+00, -5.7220e-06, -4.1591e+00, -2.0399e-03, -1.6375e+00,
         -1.0363e+00],
        [-1.3350e+00, -1.2809e-03, -2.2514e+00, -8.1647e-01, -3.2888e+00,
         -2.6522e-02],
        [-3.1086e-01, -1.7584e-01, -1.5770e+00, -2.3813e-01, -7.5841e-01,
         -7.8785e-04]], device='cuda:0')
tensor([ -7.2906,  -6.1361,  -4.1205, -11.9483, -15.0260, -17.6954,  -9.7229,
        -16.8682, -16.7352,  -3.6827, -13.4060, -10.5463, -10.9667, -15.3363,
        -11.9242,  -3.8445, -10.1819, -11.4610,  -7.7194,  -3.0610],
       device='cuda:0')
tensor([-2.7525, -3.0890, -5.2501, -0.2744, -0.2791,  0.6024, -1.0544,  0.5866,
         1.3909, -4.2068, -1.1361, -0.1589, -1.9046,  2.1146, -0.6068, -3.7124,
        -0.6067, -0.9212, -2.3329, -4.6508], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-5.2925, -1.2842, -2.0644, -1.1754, -1.9061, -1.0567],
        [-2.2152, -1.0860, -3.4616, -1.1357, -1.9683, -1.4445],
        [-2.0918, -1.2140, -1.7661, -1.0998, -1.7868, -1.9432],
        [-2.6741, -2.1354, -2.2809, -1.1911, -1.8263, -1.4161],
        [-2.7461, -2.8583, -3.1502, -2.4313, -1.9525, -1.4942],
        [-1.5152, -1.5369, -2.6261, -1.1273, -1.9245, -1.1475],
        [-2.2069, -1.6589, -2.9846, -1.2782, -2.1806, -0.9521],
        [-4.1833, -1.0453, -2.6157, -1.0450, -2.4814, -1.3062],
        [-3.0050, -1.3279, -1.1751, -1.7277, -1.4678, -1.1403],
        [-2.1513, -1.6913, -1.9332, -1.0567, -2.1213, -1.1712],
        [-3.2284, -2.6445, -2.2558, -1.3553, -2.3694, -1.7565],
        [-1.3636, -1.1770, -1.5037, -1.0840, -1.4222, -1.1210],
        [-1.6966, -1.4052, -3.2306, -1.2359, -1.3488, -0.8692],
        [-3.4849, -1.2717, -2.3320, -1.0824, -1.7490, -1.2838],
        [-2.8908, -1.6315, -1.2794, -2.5828, -2.1133, -1.1688],
        [-3.3023, -2.1187, -1.8468, -1.3151, -2.0192, -1.3661],
        [-3.6450, -4.1731, -2.5415, -2.3188, -2.7523, -1.2539],
        [-1.4376, -1.1355, -1.2601, -0.9296, -1.3160, -1.6541],
        [-2.5625, -1.2080, -2.2238, -1.1762, -1.8730, -1.3406],
        [-2.6084, -1.3502, -2.3002, -1.1460, -1.7226, -1.0154]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.7792, -11.3113,  -9.9018, -11.5239, -14.6326,  -9.8776, -11.2612,
        -12.6768,  -9.8438, -10.1248, -13.6099,  -7.6715,  -9.7863, -11.2038,
        -11.6666, -11.9683, -16.6846,  -7.7331, -10.3841, -10.1428],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-5.1705e+00, -1.0668e-02, -2.0512e+00, -1.7858e-02, -2.5566e+00,
         -2.6818e-01],
        [-2.6352e+00, -3.1921e-01, -3.7570e+00, -9.2552e-02, -2.2639e+00,
         -8.4300e-01],
        [-2.6380e+00, -5.7146e-01, -1.8540e+00, -6.4025e-02, -1.5715e+00,
         -1.3575e+00],
        [-2.4226e+00, -1.6798e+00, -3.6873e+00, -2.6267e-01, -1.5926e+00,
         -9.3328e-01],
        [-2.7447e+00, -3.0604e+00, -4.1983e+00, -1.6126e+00, -2.4038e+00,
         -1.0227e+00],
        [-1.6722e+00, -1.3772e+00, -3.2645e+00, -6.3441e-02, -2.0290e+00,
         -1.8716e-03],
        [-3.3806e+00, -5.3736e-01, -4.4875e+00, -1.7852e-03, -2.6034e+00,
         -1.4849e-02],
        [-4.5321e+00, -4.7502e-01, -3.0610e+00, -3.7074e-05, -3.5755e+00,
         -2.7610e-02],
        [-3.0287e+00, -9.8417e-02, -4.5558e-01, -1.2802e+00, -1.1648e+00,
         -1.9142e-01],
        [-1.5368e+00, -6.5371e-01, -2.2812e+00, -7.0451e-02, -2.7608e+00,
         -1.5022e-01],
        [-3.0994e+00, -2.8402e+00, -2.8945e+00, -2.3578e-01, -3.3504e+00,
         -1.8628e+00],
        [-6.9597e-01, -4.6944e-02, -9.6754e-01, -1.2413e-01, -1.9418e-01,
         -6.7158e-02],
        [-2.2174e+00, -2.0658e-01, -4.2386e+00, -1.1880e-03, -4.1815e-01,
         -7.1073e-04],
        [-3.8477e+00, -3.9325e-02, -2.6550e+00, -3.3438e-04, -2.1216e+00,
         -7.9625e-01],
        [-3.2721e+00, -6.1053e-01, -6.3241e-01, -2.6955e+00, -2.6739e+00,
         -2.4900e-01],
        [-3.2103e+00, -1.8156e+00, -1.9300e+00, -7.6778e-01, -2.4759e+00,
         -4.4772e-01],
        [-3.6917e+00, -4.7076e+00, -3.3379e+00, -1.4062e+00, -4.2805e+00,
         -9.6235e-02],
        [-1.2051e+00, -6.5578e-02, -6.9975e-02, -1.5560e-01, -4.8198e-01,
         -1.3985e+00],
        [-3.5928e+00, -1.4116e-02, -2.4688e+00, -2.7162e-03, -1.9149e+00,
         -7.7632e-01],
        [-3.1062e+00, -1.2646e-03, -2.6527e+00, -4.5589e-03, -2.0817e+00,
         -1.7866e-01]], device='cuda:0')
tensor([-10.0750,  -9.9108,  -8.0565, -10.5783, -15.0426,  -8.4083, -11.0255,
        -11.6713,  -6.2191,  -7.4532, -14.2830,  -2.0959,  -7.0826,  -9.4602,
        -10.1334, -10.6473, -17.5202,  -3.3767,  -8.7696,  -8.0252],
       device='cuda:0')
tensor([-2.7042, -1.4004, -1.8453, -0.9456,  0.4099, -1.4693, -0.2357, -1.0055,
        -3.6247, -2.6716,  0.6731, -5.5755, -2.7037, -1.7436, -1.5332, -1.3210,
         0.8356, -4.3564, -1.6145, -2.1176], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.4189, -1.1037, -1.4552, -1.2592, -2.9328, -1.5447],
        [-5.5182, -2.3993, -2.6328, -1.1436, -1.7714, -1.2371],
        [-2.3637, -3.0780, -1.8423, -1.0649, -2.4138, -1.4106],
        [-2.1443, -1.4711, -1.1625, -1.2431, -1.8258, -1.2828],
        [-1.7790, -1.8086, -1.6437, -1.4606, -2.3957, -1.7820],
        [-2.6255, -1.1320, -2.4317, -1.4075, -2.2002, -1.1558],
        [-2.3074, -1.5787, -2.0641, -2.5804, -1.9220, -1.1646],
        [-4.8342, -2.2907, -2.2190, -2.2851, -2.2544, -2.0028],
        [-4.5056, -2.5240, -3.1028, -1.0160, -1.9712, -1.2016],
        [-1.3854, -1.3992, -1.5354, -0.9136, -1.1595, -1.1198],
        [-2.6761, -1.3486, -3.0676, -1.7544, -1.6783, -0.8901],
        [-2.8044, -1.0838, -3.3751, -1.1292, -1.9115, -1.2061],
        [-2.5797, -1.1404, -1.6181, -1.1852, -2.4726, -1.7044],
        [-1.8189, -1.4029, -1.7617, -1.0987, -1.7426, -1.5275],
        [-3.5967, -2.7716, -2.5323, -1.0616, -1.7850, -1.7671],
        [-2.1449, -2.4716, -2.4823, -1.0430, -2.0711, -1.2162],
        [-1.4569, -1.1556, -2.2304, -0.9758, -1.2370, -1.1100],
        [-2.4421, -1.2955, -2.0771, -1.2979, -1.5997, -1.0372],
        [-1.8607, -1.5220, -2.0885, -2.1900, -1.3853, -1.3546],
        [-1.9075, -1.3106, -1.4352, -1.4809, -1.5699, -1.6081]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.7145, -14.7023, -12.1733,  -9.1296, -10.8697, -10.9527, -11.6172,
        -15.8863, -14.3213,  -7.5129, -11.4151, -11.5101, -10.7004,  -9.3523,
        -13.5143, -11.4290,  -8.1657,  -9.7494, -10.4011,  -9.3121],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.9931e+00, -1.6973e-01, -1.2093e+00, -2.5103e-01, -3.7470e+00,
         -9.3876e-01],
        [-6.1961e+00, -2.3887e+00, -4.6405e+00, -7.4025e-03, -1.3859e+00,
         -5.5040e-02],
        [-1.8256e+00, -3.2410e+00, -1.3715e+00, -4.6672e-02, -3.4813e+00,
         -9.0478e-01],
        [-3.0508e+00, -1.1393e+00, -4.9366e-02, -1.6667e-01, -1.6666e+00,
         -3.0740e-01],
        [-2.1511e+00, -1.0370e+00, -9.5112e-01, -5.6026e-01, -2.8571e+00,
         -9.1117e-01],
        [-2.9457e+00, -4.0925e-01, -2.6679e+00, -7.7661e-01, -3.2075e+00,
         -7.2554e-02],
        [-2.4703e+00, -5.6340e-01, -2.5017e+00, -2.8232e+00, -2.5548e+00,
         -4.1213e-01],
        [-5.2106e+00, -2.4708e+00, -3.3708e+00, -2.9520e+00, -3.0579e+00,
         -2.2240e+00],
        [-3.9282e+00, -2.4609e+00, -2.9589e+00, -2.5595e-02, -2.1386e+00,
         -3.0761e-03],
        [-2.6274e-02, -1.2002e+00, -6.9457e-01, -2.1110e-01, -9.2835e-02,
         -6.0362e-01],
        [-3.3853e+00, -1.4515e-03, -4.0509e+00, -1.3733e+00, -1.8101e+00,
         -1.2027e-01],
        [-3.1043e+00, -2.5825e-01, -4.6873e+00, -2.3513e-02, -2.4682e+00,
         -5.3119e-02],
        [-3.4392e+00, -7.1170e-01, -1.0029e+00, -1.8205e-01, -2.7942e+00,
         -1.2878e+00],
        [-2.5394e+00, -2.3649e-01, -1.9238e+00, -1.1766e-01, -1.3945e+00,
         -5.4123e-01],
        [-3.5744e+00, -3.5022e+00, -4.1118e+00, -2.1014e-01, -1.6380e+00,
         -1.5681e+00],
        [-1.1655e+00, -2.2686e+00, -2.3921e+00, -1.9501e-02, -2.2902e+00,
         -1.0850e-01],
        [-1.4598e+00, -8.4581e-03, -2.3438e+00, -9.1015e-02, -4.8549e-01,
         -3.5732e-01],
        [-3.0006e+00, -1.3172e-01, -2.3829e+00, -2.2889e-01, -1.2098e+00,
         -3.1917e-01],
        [-1.9449e+00, -3.6873e-01, -2.1471e+00, -2.1096e+00, -7.4635e-01,
         -9.6239e-01],
        [-2.4605e+00, -2.8203e-01, -6.2030e-01, -7.7181e-01, -1.0912e+00,
         -1.3481e+00]], device='cuda:0')
tensor([ -9.3088, -14.6736, -10.8708,  -6.3802,  -8.4678, -10.0796, -11.3255,
        -19.2861, -11.5153,  -2.8286, -10.7413, -10.5947,  -9.4179,  -6.7531,
        -14.6045,  -8.2443,  -4.7459,  -7.2731,  -8.2790,  -6.5740],
       device='cuda:0')
tensor([-1.4057, -0.0287, -1.3025, -2.7495, -2.4019, -0.8732, -0.2917,  3.3998,
        -2.8060, -4.6843, -0.6738, -0.9155, -1.2825, -2.5992,  1.0902, -3.1847,
        -3.4198, -2.4763, -2.1221, -2.7381], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.9958, -1.3237, -2.4104, -1.2430, -2.0909, -1.0326],
        [-1.8201, -1.5619, -3.6233, -1.6552, -1.4941, -1.1034],
        [-1.2507, -4.1037, -1.1069, -2.3420, -1.3248, -1.7433],
        [-6.3179, -3.2343, -1.5005, -1.5378, -2.4108, -1.1161],
        [-2.1284, -3.4081, -1.9054, -1.9769, -2.6465, -1.1868],
        [-1.3432, -1.9101, -1.9820, -2.1278, -1.4524, -1.4127],
        [-2.8069, -1.2767, -2.2042, -1.3095, -1.3345, -1.1905],
        [-2.8517, -1.2972, -2.5416, -1.3491, -1.2455, -1.3038],
        [-3.9409, -1.3941, -1.9866, -0.9972, -2.6480, -2.3318],
        [-4.6522, -1.2875, -1.1204, -1.2053, -2.1535, -1.4513],
        [-2.9104, -2.4124, -1.7047, -1.1571, -1.6958, -1.1822],
        [-3.1127, -2.7553, -2.1271, -1.7918, -2.2890, -1.4051],
        [-2.3826, -1.2479, -1.2035, -1.1406, -1.9780, -1.4923],
        [-1.9489, -1.4277, -2.1019, -1.1482, -2.1199, -1.0535],
        [-1.9472, -1.3577, -3.8750, -1.1716, -1.3570, -1.0609],
        [-2.3396, -1.2966, -2.3047, -1.2266, -2.8928, -1.5425],
        [-2.7833, -1.1235, -1.1511, -1.2792, -1.8147, -5.5821],
        [-4.6213, -3.3636, -2.1110, -1.1178, -2.0562, -1.5774],
        [-2.6350, -3.4646, -2.1083, -1.5621, -2.1708, -1.3635],
        [-1.1561, -1.3734, -1.2335, -1.1853, -1.5259, -1.8039]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.0964, -11.2580, -11.8714, -16.1175, -13.2521, -10.2282, -10.1224,
        -10.5890, -13.2988, -11.8702, -11.0627, -13.4809,  -9.4450,  -9.8001,
        -10.7694, -11.6029, -13.7339, -14.8473, -13.3043,  -8.2782],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.8749e+00, -1.2631e-02, -2.8759e+00, -4.6602e-02, -2.1742e+00,
         -9.3161e-02],
        [-1.8366e+00, -4.9903e-01, -4.7316e+00, -1.2607e+00, -1.3221e+00,
         -7.1287e-05],
        [-6.6974e-01, -3.5316e+00, -1.0079e-02, -2.2941e+00, -6.8656e-01,
         -1.8347e+00],
        [-5.6120e+00, -4.2549e+00, -8.5062e-01, -1.1551e+00, -3.1022e+00,
         -5.3334e-02],
        [-1.1460e+00, -4.6845e+00, -1.9834e+00, -9.3250e-01, -3.7923e+00,
         -1.1381e-01],
        [-3.8183e-01, -2.2072e+00, -2.0647e+00, -1.4583e+00, -2.8587e-01,
         -7.1070e-01],
        [-3.4630e+00, -1.7316e-02, -2.1849e+00, -1.5935e-01, -1.6180e-01,
         -5.8583e-01],
        [-2.9116e+00, -1.2813e-01, -3.1844e+00, -8.4656e-01, -6.3166e-01,
         -7.2732e-01],
        [-5.9170e+00, -1.3461e+00, -1.8917e+00, -5.3644e-06, -2.7208e+00,
         -2.1966e+00],
        [-5.5788e+00, -9.0690e-01, -1.9967e-01, -1.2212e-03, -2.4690e+00,
         -5.6888e-01],
        [-2.8533e+00, -2.3252e+00, -1.3700e+00, -9.6051e-03, -1.0711e+00,
         -1.0249e-01],
        [-3.2552e+00, -2.9489e+00, -2.1636e+00, -8.7923e-01, -3.4802e+00,
         -6.9129e-01],
        [-3.5941e+00, -1.9801e-01, -1.0358e-01, -3.5265e-02, -2.0865e+00,
         -9.3865e-01],
        [-2.6757e+00, -3.0737e-01, -2.4293e+00, -7.7137e-03, -2.1794e+00,
         -2.1127e-01],
        [-2.0142e+00, -1.9101e-01, -5.3791e+00, -1.8241e-01, -9.1422e-01,
         -9.7819e-02],
        [-2.9243e+00, -1.0022e+00, -1.8301e+00, -3.1602e-01, -3.9307e+00,
         -9.6332e-01],
        [-3.6079e+00, -1.0767e-01, -4.2980e-01, -2.9216e-01, -1.7246e+00,
         -3.7946e+00],
        [-5.1346e+00, -3.8939e+00, -2.7107e+00, -6.5158e-02, -2.6327e+00,
         -1.3944e+00],
        [-2.4010e+00, -3.9075e+00, -2.1825e+00, -4.9098e-01, -3.0503e+00,
         -7.4108e-01],
        [-2.5546e-01, -8.9219e-01, -2.5144e-01, -7.0680e-02, -8.5524e-01,
         -1.7257e+00]], device='cuda:0')
tensor([ -8.0774,  -9.6500,  -9.0267, -15.0281, -12.6525,  -7.1086,  -6.5722,
         -8.4297, -14.0722,  -9.7245,  -7.7317, -13.4184,  -6.9560,  -7.8107,
         -8.7788, -10.9667,  -9.9567, -15.8313, -12.7733,  -4.0507],
       device='cuda:0')
tensor([-2.0190, -1.6080, -2.8447, -1.0893, -0.5996, -3.1195, -3.5501, -2.1593,
         0.7735, -2.1457, -3.3310, -0.0625, -2.4889, -1.9894, -1.9906, -0.6362,
        -3.7772,  0.9840, -0.5310, -4.2274], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.0368, -1.3139, -1.4176, -1.5331, -1.4292, -1.5051],
        [-3.0586, -1.3825, -1.1362, -1.4659, -1.2637, -1.2569],
        [-1.1862, -1.7978, -2.0262, -3.6249, -1.9018, -1.6051],
        [-2.4013, -1.5331, -2.1986, -3.4009, -1.6534, -1.8667],
        [-1.9791, -2.2984, -1.1030, -1.0988, -2.6139, -1.3356],
        [-3.3161, -2.0722, -2.9101, -1.2143, -1.9807, -1.4402],
        [-2.5310, -2.6952, -2.0726, -1.6805, -2.6990, -1.8416],
        [-1.7920, -1.2182, -1.2051, -0.9165, -1.6901, -1.1559],
        [-2.1819, -1.3159, -2.2715, -1.1945, -3.5335, -1.2016],
        [-4.2578, -1.5313, -2.6585, -1.1596, -1.6703, -1.5082],
        [-1.1962, -1.5614, -1.2245, -2.3421, -1.6812, -1.9577],
        [-2.3035, -2.4381, -2.0890, -1.5738, -2.8912, -1.1819],
        [-2.5924, -2.5650, -2.1485, -1.9150, -2.0559, -1.3191],
        [-1.4945, -1.2920, -1.3592, -1.0455, -1.4825, -1.1908],
        [-2.9810, -1.1998, -3.1438, -1.2017, -1.8633, -1.2968],
        [-7.8827, -1.3792, -2.9718, -1.2291, -1.4539, -1.2849],
        [-2.5832, -1.1864, -1.9402, -1.2438, -1.9559, -2.3063],
        [-1.6827, -1.1570, -1.4336, -1.2270, -1.5434, -2.7978],
        [-3.0429, -3.4583, -1.8339, -1.0744, -2.3832, -1.5174],
        [-2.0807, -5.1984, -1.9465, -1.7526, -2.2622, -1.4946]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.2356,  -9.5639, -12.1420, -13.0539, -10.4287, -12.9336, -13.5198,
         -7.9778, -11.6990, -12.7858,  -9.9633, -12.4775, -12.5960,  -7.8645,
        -11.6864, -16.2015, -11.2159,  -9.8416, -13.3102, -14.7350],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.4650e+00, -1.2656e-01, -7.6079e-01, -9.1378e-01, -2.7033e-02,
         -5.7775e-01],
        [-2.6810e+00, -9.6812e-01, -3.8656e-01, -7.0346e-01, -1.7527e-01,
         -8.2405e-01],
        [-5.7997e-01, -1.9518e+00, -1.9634e+00, -3.3442e+00, -1.8699e+00,
         -8.2563e-01],
        [-3.1560e+00, -7.6722e-01, -2.5744e+00, -3.5513e+00, -1.1410e+00,
         -1.4526e+00],
        [-2.4013e+00, -1.5569e+00, -8.3939e-03, -1.8406e-01, -2.7771e+00,
         -1.0479e+00],
        [-3.3587e+00, -1.4522e+00, -3.3531e+00, -1.6339e-01, -2.1787e+00,
         -9.2994e-01],
        [-2.4979e+00, -3.8411e+00, -2.5331e+00, -2.6662e-01, -4.2439e+00,
         -2.0327e+00],
        [-2.4907e+00, -5.8862e-01, -2.6037e-01, -2.0218e-02, -1.6288e+00,
         -7.7305e-02],
        [-3.0585e+00, -2.9911e-02, -2.5170e+00, -3.2624e-02, -4.1140e+00,
         -2.6029e-01],
        [-4.2973e+00, -4.7753e-01, -3.1067e+00, -1.0207e-01, -1.8409e+00,
         -1.3262e+00],
        [-6.9394e-01, -5.4461e-01, -1.7876e-02, -2.2909e+00, -1.7162e+00,
         -1.9987e+00],
        [-1.7609e+00, -2.4177e+00, -2.4817e+00, -1.2610e+00, -4.1477e+00,
         -1.8037e-01],
        [-2.3544e+00, -2.9946e+00, -2.8783e+00, -6.7703e-01, -2.3638e+00,
         -4.2149e-01],
        [-1.6166e+00, -4.9699e-01, -3.8177e-01, -9.4712e-04, -6.4483e-01,
         -5.1418e-02],
        [-4.2250e+00, -5.8683e-03, -3.8742e+00, -8.1062e-06, -1.8185e+00,
         -4.6477e-01],
        [-6.8122e+00, -3.1026e-02, -3.9527e+00, -1.3849e-01, -1.1482e+00,
         -7.5134e-01],
        [-3.1779e+00, -5.7851e-01, -1.4979e+00, -3.7481e-01, -2.2081e+00,
         -2.3621e+00],
        [-1.8050e+00, -1.0343e-02, -1.1047e+00, -2.0034e-01, -1.0216e+00,
         -2.4891e+00],
        [-2.9771e+00, -4.3399e+00, -1.6679e+00, -2.0373e-02, -3.6771e+00,
         -9.9010e-01],
        [-1.1149e+00, -5.7807e+00, -1.8031e+00, -7.2279e-01, -3.1510e+00,
         -1.1533e+00]], device='cuda:0')
tensor([ -5.8710,  -5.7385, -10.5349, -12.6425,  -7.9757, -11.4360, -15.4153,
         -5.0660, -10.0123, -11.1507,  -7.2623, -12.2495, -11.6896,  -3.1925,
        -10.3883, -12.8339, -10.1992,  -6.6311, -13.6725, -13.7257],
       device='cuda:0')
tensor([-5.3646, -3.8255, -1.6071, -0.4114, -2.4530, -1.4976,  1.8955, -2.9117,
        -1.6866, -1.6351, -2.7010, -0.2280, -0.9064, -4.6719, -1.2981, -3.3676,
        -1.0167, -3.2105,  0.3623, -1.0093], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.9152, -1.1852, -2.3966, -1.6499, -1.4745, -2.0720],
        [-2.1993, -1.4063, -1.2966, -1.1142, -1.8294, -1.9043],
        [-2.9457, -2.3970, -1.6666, -1.0947, -1.8652, -1.4899],
        [-3.6538, -3.6802, -3.1061, -1.5989, -2.3547, -1.9094],
        [-1.4423, -1.3262, -1.8445, -1.6466, -1.4665, -1.2155],
        [-2.1959, -1.3013, -2.7471, -1.2247, -1.5769, -0.9570],
        [-5.2956, -1.0866, -1.5694, -1.3414, -1.7768, -1.2836],
        [-2.1581, -1.4008, -1.4572, -2.9202, -2.0871, -1.0296],
        [-5.0123, -2.5704, -1.8348, -1.2083, -1.7248, -1.2610],
        [-3.1253, -3.6857, -1.9237, -1.3471, -2.3647, -1.5564],
        [-1.4232, -1.3191, -2.4789, -1.1235, -1.4464, -1.0839],
        [-2.0981, -1.5325, -2.1316, -1.2639, -1.8257, -1.2354],
        [-2.1640, -1.1011, -3.6372, -1.0575, -1.5768, -1.1803],
        [-1.3429, -1.1813, -1.4720, -1.4942, -1.7731, -1.1016],
        [-2.7042, -2.5021, -1.8440, -1.1847, -1.9505, -1.2520],
        [-2.5842, -3.0695, -3.1724, -1.2317, -1.8264, -1.2694],
        [-1.3917, -1.6474, -1.6912, -1.4217, -1.3784, -1.1842],
        [-2.3525, -1.7074, -3.6937, -1.2296, -1.4291, -0.9730],
        [-2.5575, -1.0112, -2.6445, -1.1179, -1.9981, -1.4574],
        [-2.3283, -1.7765, -1.2577, -2.9098, -1.4662, -0.9438]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.6936,  -9.7500, -11.4592, -16.3031,  -8.9416, -10.0028, -12.3535,
        -11.0530, -13.6116, -14.0029,  -8.8751, -10.0871, -10.7169,  -8.3652,
        -11.4374, -13.1537,  -8.7145, -11.3853, -10.7865, -10.6824],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.6511e+00, -8.7107e-03, -2.3109e+00, -1.0677e+00, -1.0291e+00,
         -1.8664e+00],
        [-2.9006e+00, -3.2108e-01, -8.4855e-01, -2.4164e-02, -1.7159e+00,
         -1.2673e+00],
        [-2.6850e+00, -2.7542e+00, -5.2489e-01, -2.5973e-01, -1.9348e+00,
         -6.1294e-01],
        [-3.2755e+00, -3.5462e+00, -3.7993e+00, -8.8334e-01, -2.8642e+00,
         -1.8298e+00],
        [-1.1522e+00, -5.9332e-01, -1.8039e+00, -7.8815e-01, -2.5252e-01,
         -2.9648e-01],
        [-2.9342e+00, -1.7786e-04, -3.7954e+00, -1.1276e-03, -1.1719e+00,
         -3.6184e-01],
        [-5.3842e+00, -1.9728e-03, -9.6851e-01, -5.3999e-01, -2.1340e+00,
         -5.2737e-01],
        [-2.6605e+00, -2.9996e-01, -1.1046e+00, -3.3006e+00, -2.4476e+00,
         -1.0724e-01],
        [-5.0322e+00, -3.0201e+00, -1.8597e+00, -6.2802e-01, -1.3761e+00,
         -6.4373e-04],
        [-2.8393e+00, -4.0116e+00, -1.7875e+00, -3.0453e-01, -3.1644e+00,
         -1.2697e+00],
        [-1.2271e+00, -6.3898e-01, -3.0363e+00, -1.0313e-01, -5.0932e-01,
         -5.7304e-02],
        [-2.9805e+00, -3.2397e-01, -2.4674e+00, -3.5107e-03, -1.8142e+00,
         -3.0276e-01],
        [-2.3416e+00, -3.1109e-03, -4.8708e+00, -1.6976e-02, -1.6730e+00,
         -8.3054e-02],
        [-1.0837e+00, -1.3872e-02, -1.2806e+00, -8.1805e-01, -1.9423e+00,
         -1.2743e-04],
        [-2.4261e+00, -2.8428e+00, -1.7639e+00, -2.2123e-01, -2.1868e+00,
         -3.7541e-02],
        [-1.9866e+00, -3.1615e+00, -4.3753e+00, -2.0592e-01, -1.9562e+00,
         -3.4029e-01],
        [-1.2336e+00, -1.6753e+00, -1.3920e+00, -6.6816e-01, -3.7599e-02,
         -6.3996e-02],
        [-3.2291e+00, -8.0075e-01, -5.2375e+00, -2.8793e-03, -7.7344e-01,
         -1.0330e-01],
        [-2.8403e+00, -5.7999e-02, -3.2503e+00, -1.5944e-01, -2.5003e+00,
         -1.1476e+00],
        [-2.7666e+00, -9.6347e-01, -4.2758e-01, -3.2609e+00, -1.0742e+00,
         -3.3848e-02]], device='cuda:0')
tensor([ -9.9339,  -7.0775,  -8.7716, -16.1983,  -4.8866,  -8.2647,  -9.5561,
         -9.9206, -11.9168, -13.3770,  -5.5722,  -7.8924,  -8.9886,  -5.1388,
         -9.4784, -12.0258,  -5.0707, -10.1470,  -9.9560,  -8.5265],
       device='cuda:0')
tensor([-1.7596, -2.6725, -2.6876, -0.1048, -4.0550, -1.7381, -2.7974, -1.1324,
        -1.6948, -0.6259, -3.3029, -2.1947, -1.7283, -3.2265, -1.9590, -1.1279,
        -3.6438, -1.2383, -0.8305, -2.1558], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.5065, -3.2663, -1.9060, -1.2516, -1.8299, -1.2994],
        [-1.6148, -1.1889, -1.1622, -1.0759, -1.9933, -1.4719],
        [-4.8109, -1.3545, -1.9062, -1.1521, -1.9998, -2.0028],
        [-3.7058, -1.4410, -3.8238, -1.2060, -1.9242, -1.4759],
        [-1.2517, -1.0934, -1.5017, -1.8586, -2.5374, -1.6831],
        [-1.5287, -1.5351, -1.3179, -1.2064, -1.3691, -1.0512],
        [-4.6547, -2.5401, -1.9754, -1.0541, -1.6537, -1.5208],
        [-1.2937, -1.1946, -1.2334, -0.9546, -1.8154, -2.0088],
        [-1.7115, -1.4776, -2.9362, -1.1492, -1.7736, -1.1155],
        [-4.0457, -1.8590, -2.1681, -1.4785, -1.5053, -1.8161],
        [-1.7394, -2.3400, -1.2915, -2.1082, -1.8968, -2.1882],
        [-3.6652, -1.6588, -2.5668, -3.1083, -2.1209, -1.6610],
        [-2.1290, -2.9548, -1.7609, -1.5263, -1.6446, -1.2890],
        [-1.2572, -1.8373, -2.5428, -2.1444, -1.4249, -1.1984],
        [-1.5017, -1.4742, -2.0370, -1.3532, -1.3699, -1.1131],
        [-3.7881, -1.2723, -2.2067, -1.1984, -1.3543, -1.7896],
        [-1.4527, -1.6321, -1.2558, -2.2250, -1.0952, -0.9869],
        [-4.7391, -1.5408, -3.0783, -4.2762, -2.6282, -1.9720],
        [-3.4278, -1.7364, -1.3094, -0.9992, -2.4256, -1.4266],
        [-5.2620, -2.0067, -2.9245, -1.4536, -2.1887, -2.5737]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.0596,  -8.5070, -13.2263, -13.5768,  -9.9258,  -8.0084, -13.3988,
         -8.5005, -10.1636, -12.8729, -11.5641, -14.7811, -11.3046, -10.4050,
         -8.8492, -11.6095,  -8.6477, -18.2346, -11.3250, -16.4093],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.5199e+00, -3.8045e+00, -1.6990e+00, -1.5477e-01, -1.7889e+00,
         -3.3173e-01],
        [-2.0782e+00, -2.7166e-02, -2.3506e-01, -2.9716e-02, -2.2058e+00,
         -8.3427e-01],
        [-6.0635e+00, -1.0414e-01, -1.9908e+00, -7.8201e-04, -1.8324e+00,
         -1.3218e+00],
        [-3.9272e+00, -2.1222e-01, -5.1525e+00, -5.1936e-02, -2.5711e+00,
         -1.1555e+00],
        [-9.5478e-01, -7.8519e-02, -5.7153e-01, -1.5345e+00, -3.4822e+00,
         -1.3773e+00],
        [-6.0792e-01, -6.4934e-01, -7.6824e-01, -2.2101e-01, -3.1984e-01,
         -1.1614e-01],
        [-5.1199e+00, -3.0465e+00, -2.1155e+00, -4.4680e-03, -8.3733e-01,
         -9.1361e-01],
        [-3.7079e-01, -7.6485e-03, -3.1717e-01, -4.1723e-05, -1.7098e+00,
         -1.9562e+00],
        [-2.0281e+00, -1.1101e+00, -3.5888e+00, -1.1027e-01, -1.3242e+00,
         -1.8313e-01],
        [-4.8537e+00, -9.5157e-01, -2.0477e+00, -7.4473e-01, -5.7493e-01,
         -1.0467e+00],
        [-1.8386e+00, -1.6291e+00, -5.1800e-02, -1.7376e+00, -1.9440e+00,
         -2.5940e+00],
        [-3.1123e+00, -8.6848e-01, -3.4757e+00, -3.2606e+00, -2.3647e+00,
         -1.2616e+00],
        [-1.3711e+00, -3.8438e+00, -8.7087e-01, -1.4589e-01, -5.9886e-01,
         -1.8161e-01],
        [-2.8714e-01, -1.9944e+00, -2.8882e+00, -1.6528e+00, -4.1087e-01,
         -1.1144e-01],
        [-1.7086e+00, -1.1646e+00, -2.0349e+00, -2.5724e-01, -4.9251e-01,
         -7.9966e-02],
        [-3.1364e+00, -6.5466e-01, -2.8067e+00, -1.5337e-01, -8.1427e-03,
         -1.1593e+00],
        [-1.2798e+00, -1.3421e+00, -4.7924e-01, -1.7641e+00, -1.7751e-01,
         -4.0981e-02],
        [-4.8632e+00, -7.2136e-01, -3.6768e+00, -3.6754e+00, -2.9427e+00,
         -1.7357e+00],
        [-4.8649e+00, -1.9335e+00, -3.7764e-01, -6.6078e-04, -2.8497e+00,
         -1.3791e+00],
        [-4.9353e+00, -1.1904e+00, -4.5110e+00, -5.9134e-01, -2.9042e+00,
         -2.6092e+00]], device='cuda:0')
tensor([-11.2987,  -5.4102, -11.3134, -13.0705,  -7.9988,  -2.6825, -12.0373,
         -4.3616,  -8.3446, -10.2194,  -9.7952, -14.3434,  -7.0122,  -7.3450,
         -5.7378,  -7.9185,  -5.0837, -17.6153, -11.4056, -16.7415],
       device='cuda:0')
tensor([-1.7609, -3.0968, -1.9128, -0.5063, -1.9269, -5.3260, -1.3615, -4.1389,
        -1.8190, -2.6535, -1.7689, -0.4377, -4.2924, -3.0601, -3.1114, -3.6910,
        -3.5640, -0.6193,  0.0806,  0.3322], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.1113, -2.7781, -1.7951, -1.7492, -1.6435, -1.5156],
        [-1.4595, -1.1985, -1.3066, -0.9844, -1.5194, -1.1593],
        [-2.7265, -1.4399, -3.9636, -1.3104, -1.8206, -0.8595],
        [-2.5679, -1.3567, -1.8778, -1.2757, -1.8234, -1.2497],
        [-1.9354, -1.2528, -1.1885, -2.5045, -2.5760, -5.1487],
        [-3.8037, -2.8093, -2.0666, -1.8926, -1.8071, -1.3718],
        [-2.3935, -2.7604, -2.0091, -1.1566, -2.0996, -1.3961],
        [-1.4384, -1.5520, -2.0131, -1.1460, -1.6627, -1.1839],
        [-2.6496, -1.3654, -2.2846, -1.3263, -2.4240, -2.7549],
        [-2.1445, -1.3666, -2.2852, -1.1914, -1.4770, -1.5382],
        [-2.9934, -1.2932, -2.4041, -1.5919, -2.2886, -1.7224],
        [-1.9166, -1.1570, -1.3099, -1.2141, -1.5085, -1.7896],
        [-3.5934, -2.4155, -2.1800, -1.1120, -1.9377, -1.4209],
        [-3.3377, -3.6623, -2.3682, -1.4155, -2.2523, -1.6429],
        [-1.5715, -1.7371, -2.0156, -1.1613, -1.4057, -1.7026],
        [-2.5882, -1.6914, -2.6202, -1.3343, -1.5919, -1.1082],
        [-3.6379, -1.0360, -2.1515, -1.0462, -1.6557, -1.3209],
        [-1.7342, -1.2680, -1.2036, -1.7063, -1.4581, -1.1344],
        [-3.4835, -2.2146, -1.9893, -1.1110, -2.5195, -1.3539],
        [-2.9552, -3.2189, -2.6540, -1.7935, -2.0375, -1.3647]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.5927,  -7.6277, -12.1205, -10.1512, -14.6058, -13.7510, -11.8154,
         -8.9962, -12.8049, -10.0030, -12.2935,  -8.8956, -12.6596, -14.6788,
         -9.5938, -10.9343, -10.8482,  -8.5046, -12.6718, -14.0239],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.4232e+00, -3.4758e+00, -9.5339e-01, -4.4889e-01, -4.4595e-01,
         -9.4164e-01],
        [-1.3782e+00, -4.3851e-03, -3.8275e-01, -1.3416e-01, -6.4272e-01,
         -1.6848e-03],
        [-4.0034e+00, -2.3892e-01, -5.2857e+00, -2.8241e-01, -1.6277e+00,
         -1.0852e-03],
        [-2.8137e+00, -1.5022e-01, -1.6959e+00, -3.2831e-01, -2.2256e+00,
         -7.0997e-01],
        [-2.0550e+00, -3.2496e-03, -1.7785e-01, -2.6065e+00, -3.5238e+00,
         -4.7256e+00],
        [-3.5417e+00, -3.2611e+00, -2.6827e+00, -1.9375e+00, -1.5848e+00,
         -8.2490e-01],
        [-1.7422e+00, -3.1371e+00, -2.1201e+00, -4.0076e-02, -2.7169e+00,
         -5.9268e-01],
        [-1.4350e+00, -1.4226e+00, -2.1090e+00, -8.6849e-02, -1.2662e+00,
         -2.9683e-05],
        [-4.0917e+00, -5.8585e-03, -2.7107e+00, -1.7643e-01, -2.8930e+00,
         -1.3458e+00],
        [-2.3772e+00, -1.3796e-01, -2.5040e+00, -1.0366e-01, -1.2714e+00,
         -1.3421e+00],
        [-3.4038e+00, -8.9215e-01, -2.0086e+00, -1.2938e+00, -2.9424e+00,
         -1.5582e+00],
        [-2.3449e+00, -2.6501e-02, -6.5788e-01, -1.5337e-02, -8.6872e-01,
         -1.1822e+00],
        [-3.4682e+00, -2.7749e+00, -3.1979e+00, -2.3780e-01, -2.2429e+00,
         -8.2523e-01],
        [-3.7072e+00, -4.1306e+00, -2.7205e+00, -4.2754e-01, -3.2121e+00,
         -1.4394e+00],
        [-1.7848e+00, -1.9303e+00, -2.2094e+00, -1.1089e-01, -3.9991e-01,
         -1.4603e+00],
        [-3.5374e+00, -6.9052e-01, -3.7406e+00, -1.8149e-01, -1.2851e+00,
         -1.9168e-01],
        [-4.0033e+00, -2.7318e-01, -2.2810e+00, -9.0599e-06, -1.8879e+00,
         -7.0178e-01],
        [-1.9818e+00, -3.8147e-05, -6.8812e-03, -1.2718e+00, -1.0802e+00,
         -2.1406e-01],
        [-3.4423e+00, -2.0519e+00, -2.3186e+00, -2.1512e-01, -3.4795e+00,
         -6.8528e-01],
        [-2.7579e+00, -4.1284e+00, -3.9800e+00, -5.8040e-01, -2.5457e+00,
         -1.4105e-01]], device='cuda:0')
tensor([ -9.6889,  -2.5439, -11.4392,  -7.9237, -13.0920, -13.8326, -10.3490,
         -6.3197, -11.2235,  -7.7363, -12.0988,  -5.0955, -12.7470, -15.6374,
         -7.8956,  -9.6268,  -9.1472,  -4.5548, -12.1926, -14.1335],
       device='cuda:0')
tensor([-2.9038, -5.0838, -0.6813, -2.2275, -1.5138,  0.0816, -1.4663, -2.6765,
        -1.5814, -2.2667, -0.1947, -3.8001,  0.0874,  0.9586, -1.6982, -1.3075,
        -1.7010, -3.9498, -0.4792,  0.1096], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.1304, -1.6623, -1.1687, -1.4948, -1.7153, -1.6817],
        [-2.8687, -2.0779, -2.5007, -3.1319, -2.6027, -0.9899],
        [-4.0124, -2.8499, -2.1478, -1.1502, -2.1343, -2.3260],
        [-1.3074, -1.4286, -1.5775, -0.9903, -1.2647, -1.0456],
        [-2.7048, -1.2049, -4.0692, -1.2974, -3.1750, -1.1799],
        [-1.4572, -1.5426, -1.2834, -1.6666, -2.0413, -1.6138],
        [-1.9539, -3.2143, -1.5715, -4.2587, -1.1305, -1.2550],
        [-6.4173, -3.3230, -2.1703, -3.6476, -1.6678, -1.1776],
        [-1.6717, -2.6445, -3.4101, -1.5741, -1.3763, -1.1337],
        [-2.3496, -1.4456, -2.3925, -1.0674, -1.4673, -1.0430],
        [-0.9593, -1.1689, -1.0719, -1.2585, -1.1791, -1.1488],
        [-1.8692, -1.0514, -1.1605, -0.9918, -2.1151, -1.3330],
        [-1.6743, -1.5149, -2.5594, -1.7464, -1.2125, -0.9091],
        [-1.4637, -2.5347, -1.3833, -2.5269, -1.4221, -1.4218],
        [-2.8283, -1.6937, -2.5247, -2.6588, -1.6842, -1.8763],
        [-1.4528, -0.9056, -1.3730, -1.5468, -2.9020, -1.5503],
        [-7.9061, -3.6469, -1.5905, -1.4183, -2.3955, -2.1109],
        [-2.2175, -2.2966, -2.2340, -2.3059, -2.1144, -2.1710],
        [-1.3066, -1.1242, -1.4722, -0.9979, -1.1897, -1.1398],
        [-2.7503, -1.2060, -3.0666, -1.1319, -1.3395, -0.8430]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.8532, -14.1718, -14.6206,  -7.6140, -13.6312,  -9.6050, -13.3837,
        -18.4035, -11.8104,  -9.7655,  -6.7866,  -8.5211,  -9.6166, -10.7524,
        -13.2660,  -9.7306, -19.0681, -13.3394,  -7.2305, -10.3372],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.9676e-01, -1.0408e+00, -2.6846e-01, -8.2963e-01, -1.3411e+00,
         -1.6424e+00],
        [-2.3118e+00, -1.3173e+00, -3.0236e+00, -3.6603e+00, -3.4646e+00,
         -5.5659e-04],
        [-3.6110e+00, -2.9170e+00, -2.4322e+00, -1.8536e-01, -2.6714e+00,
         -2.4941e+00],
        [-7.5468e-01, -9.9119e-01, -1.0865e+00, -2.7024e-01, -2.1877e-02,
         -2.6366e-02],
        [-3.1220e+00, -1.6873e-02, -4.8388e+00, -3.2575e-01, -3.6920e+00,
         -3.3104e-01],
        [-1.1859e+00, -6.6538e-01, -2.2257e-01, -1.1848e+00, -2.4732e+00,
         -1.4310e+00],
        [-1.3265e+00, -2.6429e+00, -1.2844e+00, -4.1093e+00, -5.4266e-02,
         -3.5922e-01],
        [-6.3540e+00, -4.5945e+00, -2.8389e+00, -4.5186e+00, -1.0708e+00,
         -3.2853e-01],
        [-6.5097e-01, -2.7861e+00, -3.1324e+00, -1.0469e+00, -4.3271e-01,
         -1.2177e-01],
        [-2.4107e+00, -1.1692e+00, -2.4484e+00, -4.3359e-02, -1.1784e+00,
         -4.0637e-01],
        [-7.7859e-03, -8.3318e-02, -2.3138e-02, -3.8579e-01, -3.3183e-03,
         -1.8537e-02],
        [-1.4218e+00, -2.8741e-01, -2.8635e-01, -5.3309e-02, -2.0310e+00,
         -6.3905e-01],
        [-7.8826e-01, -1.1053e+00, -2.4113e+00, -1.1173e+00, -5.2411e-01,
         -5.7426e-02],
        [-1.5314e+00, -3.5880e+00, -8.4148e-01, -2.2744e+00, -9.0007e-01,
         -5.6273e-01],
        [-3.5709e+00, -1.0672e+00, -3.1968e+00, -2.9655e+00, -1.2918e+00,
         -1.3615e+00],
        [-9.8837e-01, -3.1050e-03, -7.3845e-01, -9.1071e-01, -3.1053e+00,
         -1.4425e+00],
        [-6.2105e+00, -3.1346e+00, -1.2504e+00, -6.7433e-01, -3.1059e+00,
         -1.8784e+00],
        [-1.5387e+00, -2.1433e+00, -2.8563e+00, -1.2743e+00, -2.5093e+00,
         -2.6870e+00],
        [-5.7661e-01, -1.1646e-01, -6.3304e-01, -3.4329e-01, -1.5850e-01,
         -4.3857e-01],
        [-3.3559e+00, -7.2242e-01, -3.0705e+00, -1.1256e-03, -4.9598e-01,
         -4.1008e-04]], device='cuda:0')
tensor([ -5.3191, -13.7781, -14.3110,  -3.1509, -12.3265,  -7.1629,  -9.7766,
        -19.7052,  -8.1709,  -7.6564,  -0.5219,  -4.7190,  -6.0037,  -9.6981,
        -13.4537,  -7.1884, -16.2542, -13.0090,  -2.2665,  -7.6464],
       device='cuda:0')
tensor([-3.5341, -0.3937, -0.3097, -4.4632, -1.3047, -2.4422, -3.6071,  1.3017,
        -3.6395, -2.1090, -6.2647, -3.8022, -3.6129, -1.0543,  0.1878, -2.5422,
        -2.8140, -0.3304, -4.9640, -2.6908], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.1989, -1.2985, -1.7603, -1.3051, -1.4322, -2.1970],
        [-4.6555, -1.4186, -1.5484, -1.3015, -1.6140, -1.2907],
        [-2.2551, -1.1170, -1.1895, -1.4659, -1.4894, -1.3485],
        [-1.3783, -1.3486, -1.4501, -1.1826, -2.0467, -1.9887],
        [-3.8956, -2.7683, -1.7245, -1.0548, -1.8406, -1.7035],
        [-2.4486, -3.9314, -2.2574, -1.8537, -2.4808, -1.6127],
        [-1.4181, -1.5536, -3.5695, -1.3277, -1.3637, -1.1084],
        [-1.4357, -1.3082, -2.4563, -1.2833, -1.2479, -0.8618],
        [-2.0390, -1.0701, -3.1997, -1.1092, -1.2312, -1.1997],
        [-2.6229, -1.1746, -3.1935, -1.0896, -1.5740, -1.9538],
        [-3.7283, -1.1221, -1.2728, -1.1155, -1.5822, -1.7330],
        [-3.5937, -2.3803, -1.6776, -1.1518, -2.1852, -1.3390],
        [-2.1740, -4.2011, -2.1044, -1.0758, -2.7019, -3.0962],
        [-1.3743, -1.3392, -1.2015, -1.1190, -1.5260, -1.2704],
        [-2.0578, -1.3440, -2.7244, -1.7819, -2.9825, -1.2578],
        [-3.4156, -1.1321, -2.8361, -1.9591, -2.6186, -1.9064],
        [-1.1945, -1.5739, -1.8100, -3.5050, -1.4261, -0.9976],
        [-3.8780, -2.2404, -1.9849, -2.8082, -2.3823, -1.9967],
        [-2.7208, -2.2057, -2.4023, -1.7542, -2.4502, -2.0142],
        [-1.3703, -1.2899, -2.8865, -1.1307, -0.9761, -1.0058]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.1920, -11.8286,  -8.8653,  -9.3951, -12.9873, -14.5845, -10.3410,
         -8.5932,  -9.8488, -11.6084, -10.5538, -12.3276, -15.3534,  -7.8304,
        -12.1483, -13.8679, -10.5071, -15.2905, -13.5474,  -8.6592],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.9567e+00, -1.1416e-01, -1.3201e+00, -1.5109e-01, -9.0383e-01,
         -2.2138e+00],
        [-4.6978e+00, -2.5391e-01, -1.0256e+00, -2.9537e-01, -1.5867e+00,
         -5.3883e-02],
        [-2.6182e+00, -1.0153e-01, -5.4839e-02, -7.1664e-01, -1.0943e+00,
         -7.3977e-01],
        [-1.3384e+00, -8.6933e-02, -1.0427e+00, -4.1061e-01, -2.1006e+00,
         -1.0695e+00],
        [-3.9474e+00, -3.5420e+00, -1.2745e+00, -9.9144e-02, -1.7891e+00,
         -1.2291e+00],
        [-1.8450e+00, -3.7825e+00, -2.3783e+00, -1.2442e+00, -2.9551e+00,
         -1.3019e+00],
        [-1.2765e+00, -1.4576e+00, -4.5075e+00, -5.8639e-01, -8.2895e-02,
         -2.3833e-01],
        [-1.3727e+00, -1.0403e-01, -3.1875e+00, -9.4681e-02, -7.6510e-02,
         -1.8788e-01],
        [-2.1603e+00, -6.5066e-02, -4.6409e+00, -4.1001e-01, -8.3924e-01,
         -4.6807e-01],
        [-3.6215e+00, -1.0614e+00, -3.2911e+00, -1.3088e-03, -1.1774e+00,
         -1.9692e+00],
        [-4.3894e+00, -1.2350e-01, -6.8737e-01, -1.0772e-02, -1.1609e+00,
         -1.4003e+00],
        [-3.9094e+00, -2.3122e+00, -6.0361e-01, -5.0160e-03, -3.1667e+00,
         -1.4552e-01],
        [-1.4099e+00, -4.2365e+00, -2.1374e+00, -1.1251e-01, -3.8684e+00,
         -3.7392e+00],
        [-1.1532e+00, -6.3348e-01, -2.2365e-01, -4.1303e-02, -5.4151e-01,
         -2.1144e-01],
        [-2.7635e+00, -2.9346e-02, -3.7612e+00, -1.3075e+00, -3.6297e+00,
         -3.8163e-01],
        [-3.9242e+00, -6.4833e-02, -3.5072e+00, -1.8908e+00, -3.9262e+00,
         -1.9630e+00],
        [-8.3230e-01, -5.7227e-01, -2.0867e+00, -3.9873e+00, -9.0513e-01,
         -4.4107e-05],
        [-3.8852e+00, -2.4037e+00, -2.5063e+00, -3.5853e+00, -3.1924e+00,
         -2.2762e+00],
        [-2.4408e+00, -2.0072e+00, -3.0903e+00, -7.0978e-01, -3.4645e+00,
         -2.3465e+00],
        [-2.2488e-02, -9.0279e-01, -3.2510e+00, -5.3749e-01, -6.1987e-02,
         -4.9545e-02]], device='cuda:0')
tensor([ -7.6597,  -7.9133,  -5.3254,  -6.0488, -11.8813, -13.5071,  -8.1492,
         -5.0233,  -8.5836, -11.1220,  -7.7723, -10.1425, -15.5038,  -2.8046,
        -11.8729, -15.2763,  -8.3838, -17.8490, -14.0590,  -4.8253],
       device='cuda:0')
tensor([-2.5324, -3.9153, -3.5399, -3.3463, -1.1060, -1.0774, -2.1918, -3.5699,
        -1.2653, -0.4864, -2.7815, -2.1851,  0.1505, -5.0258, -0.2754,  1.4084,
        -2.1233,  2.5585,  0.5117, -3.8339], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.6596, -1.5684, -1.1711, -3.1167, -2.0839, -1.9030],
        [-4.3109, -2.2202, -1.5421, -1.3573, -2.5484, -1.1343],
        [-2.7217, -4.2450, -2.5853, -1.5992, -2.2613, -1.3166],
        [-1.3477, -1.3579, -1.4390, -1.0744, -1.3711, -1.0998],
        [-1.6565, -1.2657, -3.0122, -1.3906, -1.8433, -5.2781],
        [-2.7492, -1.1781, -2.0394, -1.5675, -1.8718, -0.9996],
        [-2.6016, -1.5805, -2.0545, -2.7358, -2.0085, -1.3640],
        [-3.4741, -2.3848, -2.2766, -3.2399, -2.2107, -2.7837],
        [-4.8632, -2.5197, -2.7947, -1.0212, -1.6851, -1.3121],
        [-1.4145, -1.5726, -2.1618, -1.0317, -1.0233, -0.9314],
        [-1.8248, -1.4969, -2.3094, -1.6831, -1.3195, -1.0569],
        [-3.1625, -1.0804, -3.1285, -1.1147, -1.2437, -1.2223],
        [-2.4883, -1.0889, -1.2097, -2.3914, -1.5043, -1.4344],
        [-3.0430, -1.8420, -1.9344, -1.0760, -1.5527, -1.2000],
        [-2.7464, -2.5717, -1.8338, -1.6275, -2.0629, -1.3893],
        [-1.4151, -1.2305, -1.8027, -1.9472, -1.6049, -1.1250],
        [-1.7962, -1.1185, -3.4102, -1.2620, -1.4026, -0.9422],
        [-2.3645, -1.2027, -2.0630, -1.1622, -1.8415, -3.6673],
        [-2.6437, -1.5157, -1.7813, -2.0176, -1.7575, -1.2101],
        [-3.1310, -2.3070, -1.7407, -1.4899, -2.1283, -2.3598]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.5028, -13.1133, -14.7290,  -7.6899, -14.4464, -10.4056, -12.3450,
        -16.3698, -14.1960,  -8.1353,  -9.6906, -10.9522, -10.1171, -10.6481,
        -12.2315,  -9.1253,  -9.9317, -12.3012, -10.9258, -13.1566],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.6813e+00, -5.1398e-01, -8.5788e-02, -3.0425e+00, -2.2967e+00,
         -1.8734e+00],
        [-3.9262e+00, -2.1480e+00, -8.4658e-01, -8.5036e-01, -3.7281e+00,
         -3.1014e-02],
        [-2.3137e+00, -5.5823e+00, -4.1720e+00, -4.4003e-01, -2.9405e+00,
         -5.7896e-01],
        [-1.4834e-01, -7.8952e-01, -6.5526e-01, -9.5112e-02, -2.5730e-01,
         -2.2127e-01],
        [-1.9343e+00, -1.5508e-02, -3.7878e+00, -4.2639e-01, -1.8986e+00,
         -3.4603e+00],
        [-3.3881e+00, -4.4163e-01, -2.0441e+00, -1.1969e+00, -2.3720e+00,
         -5.7186e-03],
        [-2.7583e+00, -6.0907e-01, -2.3385e+00, -2.7428e+00, -2.4968e+00,
         -6.1955e-01],
        [-3.6743e+00, -2.8327e+00, -3.5170e+00, -4.2517e+00, -2.7938e+00,
         -3.8345e+00],
        [-4.4665e+00, -2.3765e+00, -2.6263e+00, -2.0577e-03, -9.0001e-01,
         -2.8968e-01],
        [-9.9531e-01, -1.6045e+00, -2.0861e+00, -3.8216e-01, -1.7173e-01,
         -1.8182e-02],
        [-2.3520e+00, -8.5181e-02, -2.9773e+00, -1.1008e+00, -3.9062e-01,
         -7.0391e-02],
        [-3.6429e+00, -1.7420e-02, -4.1207e+00, -1.3459e-01, -5.0161e-01,
         -3.3646e-01],
        [-2.9226e+00, -4.8876e-06, -1.0559e-01, -2.2943e+00, -1.2198e+00,
         -9.1554e-01],
        [-2.7894e+00, -1.2210e+00, -2.2103e+00, -1.0841e-01, -7.4912e-01,
         -2.4279e-01],
        [-2.1251e+00, -3.1380e+00, -1.4953e+00, -4.9576e-01, -2.5714e+00,
         -1.2857e-01],
        [-2.1662e-01, -4.2522e-04, -1.7032e+00, -1.1421e+00, -6.4306e-01,
         -2.3093e-02],
        [-2.0134e+00, -1.7106e-02, -3.9024e+00, -7.3468e-02, -6.1645e-01,
         -7.6233e-03],
        [-2.8305e+00, -2.0977e-01, -2.1116e+00, -7.6481e-02, -2.2424e+00,
         -3.5381e+00],
        [-3.6801e+00, -2.0799e-01, -1.9590e+00, -1.6957e+00, -1.8453e+00,
         -2.3847e-01],
        [-2.9622e+00, -2.4181e+00, -1.3125e+00, -1.3527e+00, -2.7862e+00,
         -2.9996e+00]], device='cuda:0')
tensor([ -9.4936, -11.5302, -16.0276,  -2.1668, -11.5229,  -9.4485, -11.5650,
        -20.9041, -10.6610,  -5.2579,  -6.9762,  -8.7537,  -7.4579,  -7.3209,
         -9.9542,  -3.7285,  -6.6304, -11.0087,  -9.6265, -13.8313],
       device='cuda:0')
tensor([-2.0092, -1.5830,  1.2986, -5.5231, -2.9235, -0.9571, -0.7800,  4.5344,
        -3.5350, -2.8773, -2.7144, -2.1985, -2.6592, -3.3272, -2.2773, -5.3968,
        -3.3013, -1.2925, -1.2992,  0.6747], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.1274, -1.0121, -1.2951, -1.9341, -2.2066, -1.1833],
        [-3.4091, -2.5652, -1.6946, -1.4541, -2.3983, -1.1766],
        [-2.3128, -2.2196, -1.8482, -1.6869, -1.7566, -1.4317],
        [-1.3464, -1.3668, -1.8863, -0.9925, -2.1088, -1.5186],
        [-2.7031, -1.2346, -4.1855, -1.1572, -2.1954, -1.6448],
        [-4.0438, -1.3433, -3.1729, -1.1833, -2.2468, -1.1618],
        [-2.8133, -1.0035, -1.3147, -1.3868, -1.9236, -1.4033],
        [-1.7908, -1.3783, -1.5918, -1.2606, -1.6325, -2.1410],
        [-3.1098, -2.4209, -1.9287, -1.0736, -1.8495, -1.2827],
        [-2.3397, -2.3402, -2.6935, -1.3214, -2.0808, -2.1452],
        [-1.2112, -1.7407, -1.9786, -1.3321, -1.3351, -1.1972],
        [-3.2628, -1.4151, -2.5974, -1.4709, -1.6793, -0.9063],
        [-3.4316, -1.0754, -1.4361, -1.1440, -1.3869, -1.2847],
        [-3.0106, -2.4304, -1.8669, -1.8100, -1.3286, -1.5451],
        [-3.4229, -2.3152, -1.9617, -1.2415, -1.9829, -1.5732],
        [-3.3180, -2.3970, -3.0466, -1.2231, -2.1511, -2.0026],
        [-1.3147, -2.4258, -1.2900, -2.4183, -1.4373, -1.2158],
        [-2.7660, -1.2963, -2.3791, -1.4519, -1.2231, -1.2911],
        [-3.5468, -1.2297, -2.4128, -1.2803, -1.5063, -1.2772],
        [-2.8509, -1.2289, -1.1551, -2.1136, -1.4477, -3.1012]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.7586, -12.6979, -11.2558,  -9.2193, -13.1206, -13.1519,  -9.8453,
         -9.7951, -11.6652, -12.9208,  -8.7949, -11.3318,  -9.7587, -11.9916,
        -12.4973, -14.1383, -10.1018, -10.4075, -11.2531, -11.8974],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.7403, -0.1225, -0.0757, -1.4810, -2.5963, -0.0654],
        [-3.2536, -2.7937, -1.4786, -1.0472, -2.9976, -0.2523],
        [-1.3366, -2.0923, -1.5155, -0.5656, -1.3683, -0.6586],
        [-0.9791, -0.8234, -1.7453, -0.1694, -2.5687, -1.1106],
        [-3.9562, -0.0094, -5.2068, -0.0455, -2.4063, -0.7290],
        [-4.0073, -0.0364, -3.9845, -0.1370, -3.1851, -0.4531],
        [-3.1220, -0.0117, -0.2459, -0.5958, -2.1480, -0.7516],
        [-1.9351, -0.3862, -1.4153, -0.4753, -1.2037, -1.9224],
        [-2.8255, -2.5726, -2.1512, -0.2373, -1.9493, -0.0495],
        [-1.6960, -2.1807, -3.4787, -0.2802, -2.5019, -2.4657],
        [-0.2089, -1.7960, -2.0397, -0.4462, -0.2097, -0.4751],
        [-4.3793, -0.1192, -3.3246, -0.5956, -1.6794, -0.0199],
        [-3.8596, -0.0055, -0.4115, -0.0906, -0.9486, -0.3436],
        [-3.7008, -1.9450, -2.0399, -1.3114, -0.7362, -0.9465],
        [-3.1330, -2.4414, -2.3448, -0.6530, -2.3442, -0.8623],
        [-2.5402, -1.9960, -3.9900, -0.1668, -2.3910, -2.0870],
        [-0.7965, -2.9904, -0.3942, -1.8741, -0.2755, -0.3555],
        [-3.2241, -0.4587, -2.6574, -0.6629, -0.0301, -0.7740],
        [-3.9100, -0.0137, -3.3103, -0.3113, -1.5321, -0.2885],
        [-3.2003, -0.2732, -0.1413, -1.7316, -1.0181, -3.9874]],
       device='cuda:0')
tensor([ -8.0812, -11.8230,  -7.5370,  -7.3966, -12.3533, -11.8034,  -6.8750,
         -7.3380,  -9.7853, -12.6031,  -5.1756, -10.1182,  -5.6594, -10.6798,
        -11.7788, -13.1710,  -6.6863,  -7.8072,  -9.3659, -10.3519],
       device='cuda:0')
tensor([-2.6774, -0.8749, -3.7188, -1.8227, -0.7674, -1.3485, -2.9704, -2.4571,
        -1.8799, -0.3177, -3.6193, -1.2136, -4.0993, -1.3118, -0.7185, -0.9672,
        -3.4155, -2.6003, -1.8872, -1.5455], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.3365, -1.1728, -1.6707, -0.9269, -1.4528, -1.5751],
        [-2.6928, -1.2257, -1.9030, -1.4149, -1.4453, -1.8624],
        [-2.7272, -1.3261, -2.9584, -1.6229, -2.1134, -1.2239],
        [-1.6879, -1.0049, -1.8596, -3.6544, -1.7949, -1.1680],
        [-4.3509, -2.0436, -2.0897, -1.5065, -2.0283, -1.3034],
        [-2.4329, -2.5865, -1.7246, -1.1446, -1.9777, -1.2569],
        [-1.3692, -1.6114, -2.3701, -1.0213, -1.2747, -1.0265],
        [-1.9779, -1.2321, -1.9483, -1.4789, -1.5354, -0.8235],
        [-3.0781, -1.1022, -1.7829, -1.0333, -2.3975, -1.1489],
        [-1.5613, -0.9412, -1.1421, -1.4555, -2.0928, -1.5125],
        [-3.2196, -1.4168, -1.5842, -1.1418, -2.2388, -1.8238],
        [-2.2070, -3.4006, -2.2637, -1.9757, -2.0433, -1.3535],
        [-1.6093, -1.2145, -1.3095, -0.9477, -1.4360, -1.1144],
        [-2.0421, -1.2210, -3.1976, -1.2089, -2.6526, -0.8381],
        [-4.2773, -1.3132, -2.7920, -1.1395, -1.6078, -1.1399],
        [-1.5366, -1.2662, -3.0140, -1.0898, -2.6217, -1.4384],
        [-3.3713, -1.1913, -1.9911, -1.1938, -2.2867, -3.5132],
        [-3.9189, -2.8258, -1.9084, -1.1693, -1.9604, -1.2096],
        [-3.2382, -3.1678, -2.4446, -1.3347, -1.9622, -1.4808],
        [-2.0040, -1.8631, -2.3861, -1.2541, -1.5242, -1.2762]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.1348, -10.5441, -11.9720, -11.1698, -13.3223, -11.1233,  -8.6731,
         -8.9962, -10.5430,  -8.7054, -11.4250, -13.2437,  -7.6313, -11.1604,
        -12.2697, -10.9666, -13.5474, -12.9924, -13.6283, -10.3077],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-8.4722e-01, -1.9178e-01, -1.3049e+00, -4.8120e-03, -8.2844e-01,
         -1.2212e+00],
        [-3.6729e+00, -3.5989e-02, -1.8693e+00, -4.4560e-01, -8.0626e-01,
         -1.1266e+00],
        [-3.3166e+00, -2.8522e-03, -4.0305e+00, -1.2936e+00, -3.0222e+00,
         -6.6237e-01],
        [-1.6390e+00, -5.0551e-03, -2.0066e+00, -3.8069e+00, -1.8665e+00,
         -1.2132e-03],
        [-4.7023e+00, -1.4375e+00, -2.9667e+00, -1.5326e+00, -2.3605e+00,
         -1.8769e-02],
        [-1.7642e+00, -2.5025e+00, -1.1540e+00, -1.1145e-01, -2.3262e+00,
         -4.2636e-01],
        [-9.3455e-01, -1.5193e+00, -2.5675e+00, -1.2610e-01, -8.2962e-02,
         -4.3473e-02],
        [-2.5490e+00, -1.3830e-01, -1.7877e+00, -6.6642e-01, -1.2265e+00,
         -1.9753e-03],
        [-3.5510e+00, -8.3321e-02, -1.4980e+00, -4.2979e-02, -3.3018e+00,
         -5.7459e-04],
        [-1.3542e+00, -1.9479e-01, -7.4185e-02, -7.4299e-01, -2.2259e+00,
         -1.1281e+00],
        [-2.7978e+00, -6.7868e-02, -1.3442e+00, -3.2602e-01, -2.7554e+00,
         -1.8605e+00],
        [-1.3595e+00, -4.6226e+00, -3.2526e+00, -6.7299e-01, -2.4768e+00,
         -2.7635e-01],
        [-1.9731e+00, -3.2421e-01, -3.3038e-01, -1.3512e-01, -5.3604e-01,
         -8.3934e-02],
        [-2.9289e+00, -7.8428e-04, -3.8762e+00, -3.3866e-02, -3.3783e+00,
         -1.2421e-02],
        [-4.3991e+00, -3.1915e-02, -3.5608e+00, -2.5269e-02, -1.7405e+00,
         -3.8439e-01],
        [-1.4539e+00, -9.7661e-01, -3.0453e+00, -2.4946e-02, -3.4015e+00,
         -8.0720e-01],
        [-4.1848e+00, -4.5962e-01, -2.3512e+00, -1.5972e-01, -2.6241e+00,
         -2.5780e+00],
        [-4.4301e+00, -3.2653e+00, -1.9361e+00, -2.2161e-02, -2.2637e+00,
         -1.8571e-01],
        [-3.7221e+00, -3.3160e+00, -2.9449e+00, -3.2598e-01, -2.5258e+00,
         -1.0503e+00],
        [-2.8420e+00, -2.2057e+00, -2.8283e+00, -1.3175e-01, -6.7191e-01,
         -2.0477e-01]], device='cuda:0')
tensor([ -4.3983,  -7.9566, -12.3280,  -9.3252, -13.0183,  -8.2847,  -5.2739,
         -6.3699,  -8.4777,  -5.7202,  -9.1517, -12.6607,  -3.3828, -10.2305,
        -10.1420,  -9.7094, -12.3574, -12.1031, -13.8852,  -8.8844],
       device='cuda:0')
tensor([-3.7365, -2.5875,  0.3560, -1.8446, -0.3040, -2.8386, -3.3992, -2.6263,
        -2.0652, -2.9853, -2.2733, -0.5829, -4.2485, -0.9299, -2.1277, -1.2572,
        -1.1899, -0.8893,  0.2569, -1.4233], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.5080, -1.7359, -2.1788, -1.6106, -2.5535, -2.5399],
        [-3.8144, -3.2001, -2.7463, -1.9264, -2.2923, -2.1818],
        [-1.3908, -1.1624, -1.2765, -1.7079, -1.7718, -1.1584],
        [-2.7950, -1.3817, -3.5225, -1.2608, -1.1679, -1.1524],
        [-1.7165, -1.7606, -2.0705, -2.2017, -1.4172, -1.3462],
        [-3.4166, -1.2330, -5.3130, -1.4093, -1.8455, -1.4668],
        [-2.8785, -1.3204, -1.1254, -1.4896, -2.5019, -1.0263],
        [-2.5125, -2.8365, -2.0500, -1.1455, -1.8997, -1.3111],
        [-3.4730, -4.4395, -2.1362, -1.2921, -2.3890, -1.4856],
        [-1.3669, -1.2004, -1.4392, -1.2878, -1.4039, -1.0935],
        [-3.2992, -1.2444, -2.4371, -1.2885, -2.7084, -1.3809],
        [-7.1295, -1.3246, -2.4798, -1.3842, -1.6455, -1.1851],
        [-2.0252, -1.9888, -2.2352, -3.4192, -1.8900, -1.6806],
        [-4.7112, -3.0457, -1.7815, -1.2826, -1.8683, -1.3825],
        [-2.5528, -2.9244, -2.3930, -1.2212, -2.0731, -1.5646],
        [-1.4468, -1.2735, -2.0658, -3.8232, -1.8655, -1.0569],
        [-1.9159, -1.3687, -2.1601, -1.4266, -1.4066, -0.9776],
        [-3.6290, -1.3394, -3.9427, -1.1683, -1.1787, -0.9876],
        [-2.2930, -1.1647, -2.3665, -1.2448, -1.7762, -1.1345],
        [-2.1579, -1.2833, -2.8353, -1.1693, -2.4273, -4.0225]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.1266, -16.1613,  -8.4677, -11.2803, -10.5127, -14.6842, -10.3422,
        -11.7553, -15.2153,  -7.7918, -12.3584, -15.1488, -13.2390, -14.0720,
        -12.7292, -11.5318,  -9.2554, -12.2457,  -9.9797, -13.8956],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.6563e+00, -7.4073e-01, -3.0094e+00, -1.1587e+00, -3.8514e+00,
         -2.5467e+00],
        [-4.5989e+00, -3.9826e+00, -4.2403e+00, -6.9634e-01, -3.3806e+00,
         -2.7484e+00],
        [-8.2124e-01, -6.8355e-04, -3.2107e-02, -1.6987e+00, -1.8295e+00,
         -1.3110e-01],
        [-3.8794e+00, -2.9637e-01, -4.2624e+00, -2.6307e-01, -2.2184e-02,
         -3.8305e-01],
        [-1.9846e+00, -1.0235e+00, -2.3113e+00, -2.0603e+00, -1.0410e+00,
         -5.3670e-01],
        [-3.5001e+00, -1.1003e-04, -6.1304e+00, -7.4294e-01, -2.2499e+00,
         -1.1593e+00],
        [-3.1475e+00, -1.9806e-01, -3.3972e-01, -1.1376e+00, -3.1360e+00,
         -9.9724e-02],
        [-2.1165e+00, -3.7296e+00, -2.6991e+00, -4.7824e-01, -2.0523e+00,
         -9.5133e-02],
        [-3.3236e+00, -4.5921e+00, -2.2555e+00, -3.9693e-01, -3.0728e+00,
         -1.0983e+00],
        [-9.0056e-01, -2.3658e-01, -7.5669e-01, -3.6904e-01, -4.0457e-01,
         -1.3623e-02],
        [-4.2756e+00, -3.2987e-01, -3.0615e+00, -1.5673e-01, -3.3054e+00,
         -6.0046e-01],
        [-6.4238e+00, -6.8582e-02, -3.0066e+00, -6.8204e-01, -1.9374e+00,
         -5.9035e-01],
        [-2.2761e+00, -1.0931e+00, -2.7551e+00, -3.5287e+00, -2.0222e+00,
         -1.5407e+00],
        [-4.6694e+00, -4.1658e+00, -1.4254e+00, -9.2415e-01, -1.8639e+00,
         -1.1437e-01],
        [-1.9607e+00, -3.0181e+00, -2.9861e+00, -9.8475e-02, -2.4626e+00,
         -1.3438e+00],
        [-7.2973e-02, -5.5955e-01, -2.2025e+00, -3.2289e+00, -1.5859e+00,
         -4.0770e-05],
        [-2.3475e+00, -8.9078e-01, -1.9502e+00, -6.2089e-01, -1.5522e-03,
         -3.9280e-01],
        [-4.2754e+00, -3.4661e-02, -6.2628e+00, -2.5427e-02, -3.7342e-01,
         -1.5306e-02],
        [-2.8326e+00, -8.3129e-01, -2.6314e+00, -3.2254e-01, -1.5545e+00,
         -1.0491e-01],
        [-2.7269e+00, -2.3297e-01, -3.1587e+00, -7.4163e-02, -2.9610e+00,
         -3.8129e+00]], device='cuda:0')
tensor([-12.9633, -19.6473,  -4.5133,  -9.1064,  -8.9573, -13.7826,  -8.0586,
        -11.1708, -14.7393,  -2.6811, -11.7295, -12.7088, -13.2158, -13.1630,
        -11.8698,  -7.6498,  -6.2038, -10.9870,  -8.2772, -12.9666],
       device='cuda:0')
tensor([-0.1634,  3.4859, -3.9544, -2.1739, -1.5553, -0.9016, -2.2836, -0.5845,
        -0.4760, -5.1107, -0.6289, -2.4400, -0.0232, -0.9090, -0.8593, -3.8819,
        -3.0517, -1.2587, -1.7024, -0.9290], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.6276, -1.4246, -3.9766, -1.8856, -1.2385, -1.2268],
        [-1.6470, -1.3772, -1.1911, -1.6964, -2.3854, -1.7726],
        [-6.1107, -2.4821, -2.0541, -1.0519, -2.1559, -1.3034],
        [-2.9365, -2.6537, -1.8490, -1.5770, -2.1035, -1.3111],
        [-1.3835, -1.3403, -2.1022, -1.5815, -1.5578, -1.0882],
        [-2.4417, -1.9588, -2.6886, -1.3283, -1.7758, -1.2961],
        [-2.4217, -1.0398, -2.5925, -2.6694, -2.6449, -1.5571],
        [-1.0652, -2.5896, -2.2257, -3.0190, -1.5547, -4.0552],
        [-3.3255, -2.3206, -2.5652, -1.2453, -1.9123, -1.3406],
        [-2.7728, -3.1282, -2.1615, -1.8532, -1.6440, -2.4646],
        [-1.8212, -1.9320, -2.7214, -4.2740, -1.5605, -1.0379],
        [-1.3059, -1.5774, -1.7943, -1.4716, -1.6844, -0.8003],
        [-0.9929, -1.3617, -1.7686, -1.5782, -1.2381, -1.0679],
        [-1.5183, -0.9035, -1.2682, -1.0149, -1.6793, -1.0574],
        [-3.0747, -3.7101, -4.4007, -4.3392, -3.5340, -1.2123],
        [-1.8929, -1.8076, -1.5926, -3.4148, -1.8776, -1.4822],
        [-3.3628, -1.3138, -2.8661, -3.4605, -2.1085, -1.6655],
        [-3.2535, -1.1162, -1.7853, -2.0558, -4.1796, -3.3186],
        [-3.3497, -1.9663, -2.1932, -1.4042, -2.1898, -1.3400],
        [-2.3243, -2.7119, -2.1875, -2.8012, -1.5415, -1.1786]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.3796, -10.0698, -15.1581, -12.4308,  -9.0534, -11.4893, -12.9255,
        -14.5094, -12.7095, -14.0243, -13.3471,  -8.6340,  -8.0075,  -7.4416,
        -20.2710, -12.0676, -14.7772, -15.7090, -12.4431, -12.7449],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.9833e+00, -2.7210e-01, -5.0671e+00, -1.6782e+00, -2.0424e-01,
         -4.2869e-01],
        [-1.7855e+00, -4.1845e-01, -4.0616e-02, -1.0881e+00, -2.4846e+00,
         -1.6109e+00],
        [-5.4729e+00, -2.8204e+00, -2.5193e+00, -2.1091e-01, -2.5281e+00,
         -4.4131e-01],
        [-2.5780e+00, -3.4966e+00, -1.3350e+00, -4.4512e-01, -2.5747e+00,
         -6.1160e-03],
        [-6.3652e-01, -7.9232e-01, -2.2873e+00, -7.9890e-01, -5.8958e-01,
         -4.1262e-02],
        [-3.0014e+00, -1.0606e+00, -3.8124e+00, -2.9193e-02, -2.0004e+00,
         -7.3738e-01],
        [-2.9435e+00, -8.9288e-05, -3.3916e+00, -2.4837e+00, -3.3191e+00,
         -1.4604e+00],
        [-5.5095e-01, -2.6035e+00, -3.0563e+00, -3.5674e+00, -1.1094e+00,
         -2.6527e+00],
        [-3.2065e+00, -2.4970e+00, -4.8408e+00, -7.7382e-01, -1.9599e+00,
         -1.0814e-02],
        [-2.2925e+00, -3.1461e+00, -2.3677e+00, -1.0288e+00, -1.3201e+00,
         -2.9216e+00],
        [-8.0978e-01, -2.3283e+00, -3.4538e+00, -3.4434e+00, -7.7237e-01,
         -2.8057e-02],
        [-1.3454e+00, -1.7399e+00, -1.1724e+00, -6.9277e-01, -1.7893e+00,
         -2.2084e-03],
        [-1.8559e-01, -2.5799e-02, -1.2367e+00, -1.1719e+00, -2.3306e-01,
         -2.5299e-01],
        [-1.3400e+00, -2.5110e-01, -5.7000e-01, -2.6438e-01, -1.0834e+00,
         -7.3372e-02],
        [-2.9175e+00, -5.6679e+00, -5.9059e+00, -3.9857e+00, -4.2313e+00,
         -2.3274e-01],
        [-3.0032e+00, -1.1643e+00, -1.1988e+00, -2.9776e+00, -2.3357e+00,
         -5.7949e-01],
        [-3.9994e+00, -5.1031e-01, -3.4521e+00, -3.6815e+00, -2.0567e+00,
         -1.5602e+00],
        [-4.8709e+00, -1.2999e-01, -1.4499e+00, -1.4128e+00, -4.5389e+00,
         -3.3630e+00],
        [-3.3950e+00, -1.1525e+00, -2.7015e+00, -5.1631e-01, -2.9300e+00,
         -1.1021e+00],
        [-1.5355e+00, -2.6624e+00, -2.0729e+00, -2.3243e+00, -1.7679e-01,
         -1.0235e-01]], device='cuda:0')
tensor([-10.6336,  -7.4281, -13.9929, -10.4355,  -5.1459, -10.6414, -13.5984,
        -13.5403, -13.2888, -13.0767, -10.8358,  -6.7420,  -3.1060,  -3.5823,
        -22.9411, -11.2592, -15.2603, -15.7655, -11.7973,  -8.8742],
       device='cuda:0')
tensor([-1.7460, -2.6416, -1.1652, -1.9954, -3.9075, -0.8480,  0.6729, -0.9691,
         0.5793, -0.9476, -2.5113, -1.8920, -4.9015, -3.8593,  2.6702, -0.8084,
         0.4831,  0.0565, -0.6458, -3.8707], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.1927, -1.3978, -2.3666, -1.2889, -1.8328, -0.9844],
        [-2.2585, -1.2354, -5.6165, -1.1303, -1.5408, -1.6228],
        [-1.8768, -1.2521, -1.5752, -1.1963, -1.8898, -0.8374],
        [-2.9938, -2.3570, -1.7652, -1.2750, -1.9429, -1.3169],
        [-2.7917, -2.4823, -2.2105, -1.5191, -1.6152, -1.2374],
        [-1.9885, -1.2274, -1.1490, -1.0132, -2.6001, -1.8477],
        [-3.1092, -1.1612, -1.9171, -1.1449, -1.8071, -0.9401],
        [-2.3990, -1.5125, -1.3413, -1.2484, -1.4141, -1.0447],
        [-4.6328, -1.2460, -2.5188, -1.1167, -1.5608, -1.2321],
        [-1.8214, -1.1287, -1.5286, -1.2664, -2.2075, -2.1815],
        [-3.2165, -2.2922, -2.5469, -1.1330, -2.2686, -1.1565],
        [-3.8041, -4.4327, -2.2947, -1.1470, -1.7077, -1.8806],
        [-1.4649, -1.1877, -1.4400, -1.2187, -1.4422, -1.2495],
        [-3.0025, -1.3965, -2.7812, -1.2898, -1.4554, -1.1760],
        [-4.4401, -1.2392, -2.8345, -1.4172, -1.3058, -1.0879],
        [-3.2036, -1.4937, -1.4129, -1.3863, -1.9809, -1.4959],
        [-1.5572, -1.2891, -2.0645, -1.0001, -1.9738, -1.8011],
        [-2.4023, -2.5346, -1.9821, -1.1806, -1.7247, -1.4601],
        [-3.4851, -3.3970, -2.1955, -2.3297, -2.5370, -1.4292],
        [-1.3517, -2.2585, -2.4664, -1.6793, -1.3759, -1.0643]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.0633, -13.4042,  -8.6277, -11.6508, -11.8561,  -9.8259, -10.0797,
         -8.9601, -12.3071, -10.1342, -12.6137, -15.2667,  -8.0030, -11.1014,
        -12.3247, -10.9734,  -9.6858, -11.2844, -15.3736, -10.1960],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.2526e+00, -2.9747e-01, -3.1286e+00, -6.0378e-01, -1.9414e+00,
         -1.1953e-01],
        [-2.6635e+00, -9.6805e-01, -5.1136e+00, -3.5763e-07, -1.2163e+00,
         -1.2389e+00],
        [-2.1973e+00, -2.8395e-01, -1.5043e+00, -4.9352e-02, -2.1414e+00,
         -8.6939e-03],
        [-3.2617e+00, -2.1523e+00, -1.1808e+00, -1.0610e-05, -2.0859e+00,
         -4.1431e-02],
        [-2.8177e+00, -2.2818e+00, -2.3833e+00, -6.4703e-01, -1.5704e+00,
         -2.3663e-01],
        [-2.9185e+00, -2.1956e-02, -1.3202e-01, -6.9507e-03, -3.5547e+00,
         -1.7406e+00],
        [-4.3816e+00, -2.5655e-03, -2.0607e+00, -1.8158e-01, -1.5152e+00,
         -4.2311e-02],
        [-2.6040e+00, -3.6581e-01, -2.0795e-02, -1.7036e-01, -8.8589e-01,
         -2.3645e-01],
        [-5.1146e+00, -5.0667e-01, -2.3928e+00, -6.6434e-02, -1.2566e+00,
         -2.9938e-01],
        [-1.8566e+00, -3.4520e-02, -1.5056e+00, -1.9204e-01, -2.7376e+00,
         -1.8890e+00],
        [-3.3255e+00, -1.9244e+00, -3.9231e+00, -6.7129e-03, -3.3579e+00,
         -2.3603e-05],
        [-4.3730e+00, -4.7456e+00, -2.7245e+00, -1.1275e-01, -1.7662e+00,
         -1.9935e+00],
        [-1.4540e+00, -1.0781e-01, -8.5784e-01, -3.6155e-01, -3.9602e-01,
         -4.2705e-01],
        [-4.1944e+00, -5.9248e-03, -4.0691e+00, -6.5378e-02, -7.8363e-01,
         -1.8352e-01],
        [-4.4610e+00, -1.3837e-02, -3.4509e+00, -8.0238e-01, -5.5573e-01,
         -1.2535e-02],
        [-3.4885e+00, -4.8178e-01, -5.5009e-01, -6.0031e-01, -2.3007e+00,
         -1.2291e+00],
        [-1.6784e+00, -1.3375e-01, -2.1622e+00, -1.5330e-01, -1.9687e+00,
         -1.4619e+00],
        [-1.8452e+00, -3.3390e+00, -2.1126e+00, -5.3628e-01, -1.3033e+00,
         -3.5417e-01],
        [-3.1567e+00, -3.3040e+00, -2.2872e+00, -1.8622e+00, -3.0982e+00,
         -9.4263e-01],
        [-1.5983e-02, -2.9420e+00, -2.9651e+00, -8.6242e-01, -8.9300e-02,
         -1.7165e-02]], device='cuda:0')
tensor([-10.3434, -11.2004,  -6.1849,  -8.7221,  -9.9368,  -8.3747,  -8.1839,
         -4.2833,  -9.6365,  -8.2154, -12.5376, -15.7155,  -3.6043,  -9.3019,
         -9.2964,  -8.6505,  -7.5582,  -9.4906, -14.6509,  -6.8920],
       device='cuda:0')
tensor([-0.7198, -2.2038, -2.4428, -2.9287, -1.9194, -1.4512, -1.8958, -4.6768,
        -2.6706, -1.9189, -0.0761,  0.4488, -4.3987, -1.7995, -3.0283, -2.3229,
        -2.1276, -1.7939, -0.7227, -3.3041], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.1117, -1.3421, -2.0285, -1.0801, -1.3123, -1.2817],
        [-1.9022, -1.2091, -2.1108, -1.0620, -1.6838, -0.8965],
        [-2.0269, -1.1327, -1.0778, -1.2418, -1.8784, -1.1853],
        [-3.1640, -1.4418, -1.1995, -1.3297, -1.1675, -2.0633],
        [-1.1959, -1.4672, -1.6983, -2.3292, -1.6123, -1.7149],
        [-2.5040, -1.5469, -1.9916, -1.0703, -1.7358, -1.0330],
        [-2.0355, -1.0857, -2.9832, -1.0047, -2.3672, -1.7562],
        [-1.5662, -1.6067, -1.0641, -0.9743, -1.1876, -1.1955],
        [-1.1233, -1.1810, -1.4725, -0.9947, -1.2872, -1.1991],
        [-1.0572, -1.0494, -1.2661, -1.0067, -1.1211, -1.4321],
        [-3.8328, -2.9280, -2.3939, -1.0624, -1.0566, -1.4978],
        [-1.7002, -1.8259, -3.1337, -2.0554, -1.2214, -0.8224],
        [-2.1608, -1.5523, -3.2049, -1.2257, -1.1271, -1.3186],
        [-1.8093, -0.8964, -1.1542, -0.9118, -2.0477, -1.8610],
        [-0.9955, -1.2257, -1.3340, -1.3848, -1.4441, -0.9983],
        [-3.3258, -2.0991, -1.5587, -1.2544, -2.7088, -1.4111],
        [-4.3050, -1.7660, -3.1937, -1.0924, -2.2510, -2.9211],
        [-1.4516, -1.0998, -1.8716, -1.0369, -1.2215, -1.9466],
        [-3.0118, -1.1387, -1.9268, -1.0648, -1.9305, -1.8017],
        [-2.9413, -1.8049, -1.5273, -2.2780, -1.4714, -0.9307]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.1563,  -8.8644,  -8.5428, -10.3658, -10.0178,  -9.8817, -11.2326,
         -7.5945,  -7.2579,  -6.9327, -12.7715, -10.7591, -10.5894,  -8.6803,
         -7.3824, -12.3579, -15.5291,  -8.6281, -10.8743, -10.9535],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.7129e+00, -7.8014e-01, -2.0183e+00, -1.5208e-01, -6.4015e-05,
         -3.5558e-01],
        [-2.0385e+00, -2.2832e-01, -1.9548e+00, -3.2558e-02, -1.4395e+00,
         -2.0704e-02],
        [-1.8293e+00, -1.3172e-01, -9.3445e-02, -3.9280e-01, -1.8761e+00,
         -3.2567e-01],
        [-2.9746e+00, -6.4794e-01, -1.0840e-01, -5.6072e-01, -2.6652e-01,
         -2.2169e+00],
        [-7.4386e-02, -7.5997e-01, -1.5413e+00, -1.9683e+00, -1.0660e+00,
         -1.0667e+00],
        [-2.1987e+00, -7.0920e-01, -2.2368e+00, -2.6384e-01, -1.2673e+00,
         -3.0008e-02],
        [-1.3172e+00, -3.9885e-02, -3.9036e+00, -2.9384e-03, -2.9004e+00,
         -1.8090e+00],
        [-1.4212e+00, -1.4783e+00, -8.0102e-02, -8.7702e-04, -2.2403e-01,
         -5.1333e-01],
        [-2.7108e-01, -6.3056e-03, -7.2537e-01, -1.9372e-04, -2.3033e-01,
         -3.4051e-01],
        [-1.0949e-01, -1.2050e-02, -2.8237e-01, -2.7299e-05, -2.9313e-01,
         -8.9704e-01],
        [-3.2279e+00, -3.0404e+00, -2.5853e+00, -1.4139e-01, -3.1808e-02,
         -7.7246e-01],
        [-2.2252e+00, -2.1393e+00, -3.3549e+00, -1.6874e+00, -8.2580e-01,
         -1.1454e-01],
        [-2.0656e+00, -9.5784e-01, -3.3856e+00, -6.7196e-01, -1.4881e-01,
         -7.1757e-01],
        [-1.9144e+00, -6.8018e-02, -2.4765e-01, -1.3630e-01, -1.7007e+00,
         -1.8855e+00],
        [-1.5487e-01, -2.5670e-01, -8.8192e-01, -6.3215e-01, -8.7572e-01,
         -9.8533e-02],
        [-3.4258e+00, -1.6189e+00, -8.6869e-01, -4.4837e-01, -4.0149e+00,
         -7.1030e-01],
        [-3.5428e+00, -1.1872e+00, -2.9778e+00, -1.2287e-01, -2.5910e+00,
         -3.1786e+00],
        [-1.4813e+00, -5.5059e-03, -1.7203e+00, -4.2486e-01, -3.1965e-01,
         -1.8333e+00],
        [-3.7654e+00, -1.6454e-01, -1.8801e+00, -7.2380e-03, -1.8229e+00,
         -1.1631e+00],
        [-3.2416e+00, -1.0709e+00, -5.6481e-01, -2.6046e+00, -4.0098e-01,
         -4.2951e-02]], device='cuda:0')
tensor([ -5.0191,  -5.7144,  -4.6490,  -6.7751,  -6.4767,  -6.7058,  -9.9732,
         -3.7179,  -1.5738,  -1.5941,  -9.7992, -10.3471,  -7.9474,  -5.9525,
         -2.8999, -11.0870, -13.6002,  -5.7849,  -8.8033,  -7.9259],
       device='cuda:0')
tensor([-4.1373, -3.1500, -3.8938, -3.5907, -3.5411, -3.1759, -1.2595, -3.8766,
        -5.6841, -5.3386, -2.9722, -0.4119, -2.6420, -2.7278, -4.4825, -1.2709,
        -1.9289, -2.8432, -2.0711, -3.0276], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.8291, -2.0313, -2.2109, -1.1315, -1.9688, -2.0475],
        [-1.4480, -1.2462, -1.3978, -1.0624, -1.6784, -1.1914],
        [-3.2640, -1.3052, -2.1646, -1.2324, -1.2555, -1.1517],
        [-3.6749, -1.2856, -2.5892, -1.1760, -2.1414, -1.2282],
        [-2.9524, -1.1283, -1.4727, -1.6837, -2.1486, -1.8583],
        [-1.9713, -1.1173, -1.4826, -1.2235, -2.4935, -1.8748],
        [-3.3121, -2.4621, -2.8364, -1.0711, -2.0730, -1.3128],
        [-2.3192, -2.3485, -2.2485, -1.1718, -1.7189, -1.2080],
        [-1.3912, -1.3742, -1.8901, -1.0949, -1.2996, -1.2521],
        [-1.5018, -1.4141, -2.7856, -1.4140, -1.5330, -0.8900],
        [-1.3546, -1.0871, -2.9595, -1.1598, -1.4987, -1.9010],
        [-1.7051, -1.0698, -1.7010, -1.9780, -2.1056, -3.2080],
        [-4.9064, -2.3807, -1.9785, -1.1847, -1.8257, -1.0221],
        [-2.7279, -2.8828, -1.8060, -1.9321, -1.9648, -1.4697],
        [-1.3332, -1.4165, -1.4152, -1.3080, -1.6283, -1.1943],
        [-3.2012, -1.6296, -2.9812, -1.3908, -2.8269, -1.5401],
        [-3.1945, -1.1878, -2.6412, -1.1364, -1.7432, -1.2097],
        [-2.3364, -1.3952, -1.4932, -2.4571, -1.6344, -1.3925],
        [-4.4533, -2.1337, -3.5743, -2.3505, -2.7185, -2.2445],
        [-3.4765, -3.1557, -4.1430, -0.9645, -2.2715, -1.4936]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.2191,  -8.0242, -10.3734, -12.0953, -11.2439, -10.1632, -13.0675,
        -11.0149,  -8.3021,  -9.5386,  -9.9606, -11.7674, -13.2981, -12.7833,
         -8.2956, -13.5696, -11.1127, -10.7088, -17.4748, -15.5050],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.2844e+00, -1.6327e+00, -2.7812e+00, -6.2455e-02, -2.3101e+00,
         -2.6086e+00],
        [-1.4701e+00, -3.1656e-01, -7.2073e-01, -1.0512e-03, -1.2992e+00,
         -1.0300e-01],
        [-4.3857e+00, -7.0525e-03, -2.5118e+00, -5.3004e-02, -2.6755e-01,
         -2.5870e-01],
        [-3.9726e+00, -2.5010e-04, -3.2476e+00, -5.1684e-02, -2.9358e+00,
         -6.7503e-01],
        [-3.4480e+00, -3.2783e-04, -5.1097e-01, -1.2029e+00, -2.7544e+00,
         -1.8786e+00],
        [-2.1931e+00, -2.3818e-02, -1.0463e+00, -4.1530e-01, -3.1744e+00,
         -1.8265e+00],
        [-3.2335e+00, -2.5999e+00, -4.9893e+00, -6.2845e-02, -2.7113e+00,
         -4.8043e-01],
        [-1.7489e+00, -2.2231e+00, -2.7028e+00, -3.6541e-02, -1.5846e+00,
         -3.6078e-01],
        [-1.1666e+00, -9.0207e-01, -1.7524e+00, -2.5006e-01, -3.0062e-01,
         -5.6479e-01],
        [-1.4761e+00, -5.9180e-02, -3.9962e+00, -4.0899e-01, -1.1579e+00,
         -1.6362e-01],
        [-1.0362e+00, -1.1563e-04, -3.5473e+00, -1.3277e-01, -1.2372e+00,
         -2.2577e+00],
        [-2.0940e+00, -3.1800e-01, -1.0195e+00, -1.7749e+00, -2.3116e+00,
         -3.2022e+00],
        [-4.4981e+00, -2.2943e+00, -2.2651e+00, -2.9214e-01, -1.6982e+00,
         -1.0954e-01],
        [-2.1885e+00, -3.9255e+00, -1.0429e+00, -6.5939e-01, -2.2585e+00,
         -5.7889e-01],
        [-5.3338e-01, -9.0119e-01, -5.3601e-01, -1.9897e-01, -1.0298e+00,
         -4.0046e-02],
        [-4.7747e+00, -6.0633e-01, -3.9322e+00, -4.3290e-01, -3.5175e+00,
         -5.2910e-01],
        [-3.6694e+00, -7.7096e-01, -2.9464e+00, -9.4152e-02, -2.0669e+00,
         -5.2897e-01],
        [-2.6783e+00, -7.7857e-02, -1.1710e+00, -2.4094e+00, -1.7600e+00,
         -1.0725e+00],
        [-5.2031e+00, -1.9786e+00, -6.3336e+00, -3.2217e+00, -3.9303e+00,
         -2.6820e+00],
        [-2.8426e+00, -3.1678e+00, -3.8717e+00, -1.5145e-02, -3.0925e+00,
         -8.1374e-01]], device='cuda:0')
tensor([-11.6795,  -3.9107,  -7.4838, -10.8831,  -9.7953,  -8.6794, -14.0773,
         -8.6567,  -4.9366,  -7.2620,  -8.2113, -10.7201, -11.1573, -10.6537,
         -3.2394, -13.7928, -10.0768,  -9.1691, -23.3494, -13.8034],
       device='cuda:0')
tensor([-0.5396, -4.1136, -2.8897, -1.2122, -1.4487, -1.4838,  1.0098, -2.3582,
        -3.3656, -2.2766, -1.7493, -1.0473, -2.1407, -2.1297, -5.0562,  0.2232,
        -1.0360, -1.5397,  5.8745, -1.7016], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.6997, -2.4580, -1.9578, -1.2101, -2.1507, -1.1449],
        [-3.3275, -3.0847, -2.0139, -1.8935, -2.5388, -1.8108],
        [-1.5928, -1.6673, -1.9745, -1.6929, -1.4537, -1.2182],
        [-1.5269, -1.4328, -2.3685, -1.4341, -1.3475, -1.0458],
        [-3.4013, -1.1775, -1.9569, -1.3938, -1.4780, -1.2719],
        [-2.2882, -1.2078, -1.1374, -1.6317, -1.3766, -2.3362],
        [-3.0091, -2.9498, -2.0932, -1.1112, -2.1093, -1.7639],
        [-2.4523, -2.9077, -1.9537, -1.8039, -2.4678, -1.3228],
        [-1.4135, -1.7180, -2.0325, -1.6654, -1.4439, -1.1210],
        [-2.5994, -2.1277, -2.9619, -1.2736, -2.1547, -0.9714],
        [-4.2763, -0.9703, -2.3941, -1.0424, -1.5142, -1.4649],
        [-2.0291, -1.3056, -1.2636, -1.4754, -1.4102, -1.2218],
        [-4.1153, -2.1733, -2.3135, -1.2485, -2.6660, -1.7330],
        [-3.0581, -2.7199, -2.4411, -1.4462, -2.2355, -1.5061],
        [-1.2836, -1.1885, -1.5798, -1.4017, -1.3866, -1.2116],
        [-3.2738, -1.3139, -2.3832, -1.6186, -1.2208, -0.9552],
        [-1.7527, -1.4347, -2.3803, -1.1663, -1.3572, -1.8463],
        [-2.4604, -1.3885, -1.6552, -1.5121, -1.4928, -1.4639],
        [-1.3981, -1.3291, -2.0341, -1.4784, -2.6358, -1.8714],
        [-2.7015, -2.7089, -2.0774, -1.1087, -1.9636, -2.1836]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.6211, -14.6691,  -9.5994,  -9.1555, -10.6794,  -9.9779, -13.0365,
        -12.9082,  -9.3943, -12.0886, -11.6623,  -8.7057, -14.2495, -13.4070,
         -8.0518, -10.7656,  -9.9376,  -9.9730, -10.7468, -12.7436],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.9552e+00, -2.8513e+00, -1.5619e+00, -2.0128e-01, -2.7782e+00,
         -3.7185e-01],
        [-3.5878e+00, -3.5930e+00, -1.8734e+00, -9.5421e-01, -4.1142e+00,
         -1.8752e+00],
        [-1.8858e+00, -1.6620e+00, -2.1078e+00, -9.1068e-01, -4.0837e-01,
         -9.8962e-02],
        [-1.9989e+00, -4.9385e-03, -3.1228e+00, -3.3734e-01, -1.8474e-01,
         -9.6512e-01],
        [-4.0160e+00, -3.5219e-02, -2.1691e+00, -5.3667e-01, -1.3390e+00,
         -3.2915e-02],
        [-2.5518e+00, -1.8216e-01, -8.4284e-02, -1.1398e+00, -8.2367e-01,
         -2.8863e+00],
        [-2.7125e+00, -3.3190e+00, -2.6730e+00, -3.5714e-01, -2.4967e+00,
         -9.2401e-01],
        [-1.7172e+00, -4.1135e+00, -1.9651e+00, -6.9328e-01, -3.5840e+00,
         -3.3779e-02],
        [-7.2756e-01, -1.7966e+00, -2.1987e+00, -7.8109e-01, -1.1229e-01,
         -1.0912e-01],
        [-3.7929e+00, -1.5606e+00, -4.1761e+00, -4.7537e-02, -2.4637e+00,
         -9.2550e-02],
        [-4.6539e+00, -2.5477e-01, -2.7180e+00, -2.4009e-04, -1.4801e+00,
         -1.1806e+00],
        [-2.6009e+00, -3.2586e-01, -1.3192e-02, -7.6162e-01, -8.6167e-01,
         -2.7326e-01],
        [-4.0745e+00, -1.9739e+00, -3.2241e+00, -4.5641e-01, -3.7140e+00,
         -1.6847e+00],
        [-2.9004e+00, -3.2289e+00, -3.1388e+00, -2.7469e-01, -3.2934e+00,
         -7.5694e-01],
        [-4.5713e-01, -1.5196e-01, -1.0701e+00, -5.8132e-01, -2.6879e-01,
         -1.4255e-01],
        [-4.4577e+00, -8.1742e-04, -2.9224e+00, -9.8323e-01, -6.6787e-02,
         -4.8230e-01],
        [-1.7617e+00, -2.4336e-01, -2.9859e+00, -1.9407e-02, -7.9959e-01,
         -2.0788e+00],
        [-3.3264e+00, -3.1245e-01, -1.0155e+00, -7.9513e-01, -1.0210e+00,
         -8.7943e-01],
        [-1.1588e+00, -2.1355e-01, -2.0947e+00, -1.0730e+00, -3.3469e+00,
         -1.7362e+00],
        [-2.4048e+00, -3.4533e+00, -2.8179e+00, -4.4360e-02, -2.2668e+00,
         -2.1426e+00]], device='cuda:0')
tensor([-12.7197, -15.9978,  -7.0737,  -6.6138,  -8.1288,  -7.6679, -12.4822,
        -12.1068,  -5.7253, -12.1333, -10.2876,  -4.8365, -15.1275, -13.5931,
         -2.6719,  -8.9133,  -7.8887,  -7.3498,  -9.6231, -13.1297],
       device='cuda:0')
tensor([-0.9015,  1.3287, -2.5257, -2.5417, -2.5506, -2.3100, -0.5542, -0.8014,
        -3.6690,  0.0447, -1.3747, -3.8692,  0.8780,  0.1861, -5.3800, -1.8523,
        -2.0489, -2.6232, -1.1237,  0.3860], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.7772, -3.2123, -2.3980, -1.5807, -1.5857, -1.7271],
        [-1.3852, -1.9964, -3.2100, -4.0987, -1.3976, -1.0621],
        [-1.3458, -1.5956, -1.4938, -1.2018, -1.7165, -0.8679],
        [-3.5768, -1.3115, -1.9207, -1.3328, -1.5132, -1.0280],
        [-3.3933, -1.8348, -1.8512, -0.9717, -1.8352, -3.3556],
        [-3.2828, -1.1998, -1.1664, -1.1473, -2.5806, -1.6550],
        [-2.8907, -2.4027, -1.8253, -1.1153, -1.8088, -1.7968],
        [-2.3550, -3.2169, -2.3054, -1.0952, -2.2352, -2.3108],
        [-1.3791, -1.2087, -1.8072, -1.4149, -1.4769, -1.1634],
        [-1.5698, -1.2896, -2.4346, -1.3010, -2.2040, -0.8414],
        [-2.4498, -1.0483, -2.5656, -1.1342, -1.5134, -1.2466],
        [-1.7246, -1.1488, -1.2081, -2.5105, -2.5606, -1.0520],
        [-4.6305, -3.0924, -1.8705, -2.5426, -2.1412, -1.9634],
        [-3.2841, -2.4938, -2.9852, -1.1292, -2.2139, -1.4705],
        [-1.2656, -1.2022, -1.5643, -1.0237, -1.4989, -1.5913],
        [-2.2578, -1.2615, -2.9561, -1.3031, -2.3145, -2.2233],
        [-2.5655, -1.4309, -1.8897, -1.2197, -1.5807, -1.2950],
        [-1.9914, -1.0921, -3.6058, -1.1473, -2.4720, -1.4910],
        [-2.5799, -1.1231, -1.0791, -1.1595, -3.6951, -1.1799],
        [-4.4265, -2.3369, -1.6293, -1.1466, -1.7816, -1.4095]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-15.2811, -13.1499,  -8.2214, -10.6832, -13.2418, -11.0319, -11.8395,
        -13.5185,  -8.4502,  -9.6404,  -9.9579, -10.2045, -16.2406, -13.5767,
         -8.1461, -12.3163,  -9.9815, -11.7996, -10.8166, -12.7305],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.4009e+00, -2.9953e+00, -2.6442e+00, -9.4492e-01, -1.1287e+00,
         -1.5724e+00],
        [-1.1097e+00, -2.4128e+00, -4.0279e+00, -3.5657e+00, -1.7531e-01,
         -1.2879e-02],
        [-1.1058e+00, -1.7155e+00, -5.1288e-01, -5.4836e-02, -1.8703e+00,
         -3.9661e-04],
        [-3.8687e+00, -3.7332e-02, -1.6827e+00, -6.3264e-01, -3.4364e-01,
         -2.7011e-02],
        [-5.4390e+00, -1.9076e+00, -1.9285e+00, -2.1482e-01, -1.2907e+00,
         -3.1255e+00],
        [-4.0324e+00, -8.2807e-02, -4.8344e-01, -4.8207e-02, -3.0344e+00,
         -1.2237e+00],
        [-2.8224e+00, -2.3825e+00, -1.7216e+00, -1.0808e-02, -1.5698e+00,
         -1.8259e+00],
        [-1.5224e+00, -3.0108e+00, -2.3464e+00, -2.5244e-01, -2.6599e+00,
         -2.4242e+00],
        [-3.7287e-01, -2.0918e-01, -1.6286e+00, -4.7336e-01, -4.0292e-01,
         -6.4535e-02],
        [-1.7154e+00, -1.4269e-02, -3.0569e+00, -2.3271e-01, -2.3395e+00,
         -1.7700e-02],
        [-2.7841e+00, -4.1942e-02, -2.9641e+00, -3.6507e-02, -1.3509e+00,
         -5.3350e-01],
        [-1.8955e+00, -2.8538e-02, -1.1836e-01, -2.4432e+00, -3.2463e+00,
         -3.7170e-02],
        [-4.1287e+00, -3.8678e+00, -2.2061e+00, -2.7673e+00, -2.7095e+00,
         -2.0505e+00],
        [-3.4537e+00, -2.6557e+00, -4.4812e+00, -2.7680e-02, -3.1129e+00,
         -8.3215e-01],
        [-5.7596e-01, -1.3848e-01, -1.0660e+00, -1.0054e-01, -7.6188e-01,
         -1.2486e+00],
        [-3.1576e+00, -1.1944e-02, -3.9557e+00, -2.3308e-01, -2.4541e+00,
         -1.3744e+00],
        [-2.9603e+00, -9.6232e-02, -1.5425e+00, -3.6011e-02, -1.5447e+00,
         -7.7872e-01],
        [-2.3623e+00, -2.8744e-01, -3.7014e+00, -1.5626e-01, -3.1477e+00,
         -9.0665e-01],
        [-2.8808e+00, -1.9145e-02, -8.7509e-02, -1.0640e-02, -4.3288e+00,
         -4.0709e-01],
        [-4.5891e+00, -2.3626e+00, -9.6578e-01, -3.2724e-01, -1.6540e+00,
         -7.4994e-01]], device='cuda:0')
tensor([-13.6864, -11.3044,  -5.2597,  -6.5920, -13.9062,  -8.9049, -10.3331,
        -12.2162,  -3.1515,  -7.3765,  -7.7111,  -7.7690, -17.7298, -14.5633,
         -3.8914, -11.1869,  -6.9585, -10.5617,  -7.7339, -10.6487],
       device='cuda:0')
tensor([-1.5947, -1.8455, -2.9617, -4.0912,  0.6644, -2.1270, -1.5064, -1.3023,
        -5.2987, -2.2639, -2.2469, -2.4355,  1.4892,  0.9866, -4.2547, -1.1293,
        -3.0231, -1.2379, -3.0827, -2.0818], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.0714, -1.7586, -1.2016, -4.0224, -1.4322, -1.3227],
        [-6.7236, -2.8622, -1.5358, -4.7371, -2.8086, -1.3558],
        [-3.8363, -1.0235, -1.6721, -1.1048, -1.3777, -1.1747],
        [-1.1542, -1.3357, -1.4985, -1.2449, -1.2025, -1.0238],
        [-1.5748, -1.3654, -3.0350, -1.3915, -1.1151, -1.1766],
        [-1.0388, -0.9488, -1.5576, -1.2748, -1.2485, -1.1907],
        [-2.7571, -2.0162, -1.4886, -1.6505, -1.2598, -1.1793],
        [-1.2878, -2.8733, -3.0910, -1.9975, -1.9287, -1.6143],
        [-1.6194, -3.0783, -1.7031, -2.4887, -1.4362, -2.0661],
        [-1.7375, -1.0294, -2.0542, -1.6785, -2.2283, -1.8131],
        [-4.2862, -3.3673, -1.6946, -1.6311, -1.8515, -1.9268],
        [-2.5521, -2.7058, -1.7412, -1.2607, -1.6872, -1.7563],
        [-1.2801, -1.2853, -2.1685, -1.0442, -1.1956, -1.3067],
        [-4.2870, -1.1070, -2.9913, -1.1319, -1.6301, -0.9378],
        [-3.3077, -1.2943, -1.3676, -1.5794, -1.3203, -1.1528],
        [-1.1293, -2.4131, -1.3293, -3.2596, -1.1739, -1.2605],
        [-3.2803, -2.2600, -3.1363, -1.5149, -1.5329, -1.0686],
        [-2.7852, -2.2255, -2.2454, -1.0746, -2.3712, -1.5100],
        [-1.2243, -1.5862, -1.6542, -1.0081, -1.2705, -1.1538],
        [-3.8518, -1.2366, -4.1390, -1.5356, -1.9589, -2.9255]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.8088, -20.0230, -10.1891,  -7.4596,  -9.6585,  -7.2593, -10.3517,
        -12.7927, -12.3918, -10.5408, -14.7575, -11.7033,  -8.2805, -12.0852,
        -10.0222, -10.5658, -12.7929, -12.2119,  -7.8971, -15.6475],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-9.5555e-02, -1.0043e+00, -3.6907e-01, -3.9548e+00, -9.6101e-01,
         -5.9844e-01],
        [-5.6203e+00, -3.2857e+00, -1.2134e+00, -5.0393e+00, -3.1350e+00,
         -7.8439e-01],
        [-3.3182e+00, -5.1732e-03, -1.2701e+00, -3.1824e-01, -8.7726e-02,
         -2.9062e-01],
        [-4.0948e-01, -8.1387e-01, -1.0529e+00, -4.5572e-01, -1.5775e-02,
         -1.4732e-02],
        [-1.3126e+00, -1.4929e-01, -4.0991e+00, -5.8228e-01, -8.5698e-02,
         -6.3347e-01],
        [-1.1638e-03, -8.3199e-02, -1.0730e+00, -3.4632e-01, -5.5785e-01,
         -4.1812e-02],
        [-1.8700e+00, -2.3192e+00, -1.4397e+00, -1.2104e+00, -1.9036e-02,
         -5.9619e-01],
        [-1.0167e+00, -3.7550e+00, -2.8884e+00, -1.5856e+00, -1.8196e+00,
         -7.0252e-01],
        [-1.4205e+00, -3.0803e+00, -1.2742e+00, -2.5751e+00, -8.5017e-01,
         -1.8809e+00],
        [-1.5886e+00, -1.2703e-01, -2.0548e+00, -1.0822e+00, -2.2489e+00,
         -1.9423e+00],
        [-3.8448e+00, -2.7467e+00, -1.3821e+00, -1.2306e+00, -1.7802e+00,
         -1.7519e+00],
        [-1.7454e+00, -2.4255e+00, -1.3988e+00, -1.6627e-01, -1.2988e+00,
         -1.4260e+00],
        [-1.7184e-01, -6.4980e-01, -2.3634e+00, -4.4366e-01, -9.5473e-02,
         -8.6030e-01],
        [-5.0743e+00, -5.7076e-02, -3.2180e+00, -1.1778e-03, -1.3352e+00,
         -7.0136e-02],
        [-3.2097e+00, -5.0753e-02, -2.2267e-01, -1.1766e+00, -6.2316e-01,
         -3.9039e-01],
        [-6.7817e-02, -2.1066e+00, -2.4218e-01, -3.2146e+00, -4.6406e-01,
         -5.1273e-01],
        [-2.8831e+00, -2.2540e+00, -3.6544e+00, -1.2337e+00, -7.8485e-01,
         -1.4822e-02],
        [-2.5005e+00, -2.1485e+00, -3.0833e+00, -4.8244e-04, -3.1041e+00,
         -1.2847e+00],
        [-1.9765e-04, -1.3604e+00, -1.3542e+00, -2.8942e-02, -1.1634e-01,
         -4.3596e-01],
        [-4.7383e+00, -4.5199e-02, -5.2200e+00, -8.0192e-01, -2.1605e+00,
         -2.0643e+00]], device='cuda:0')
tensor([ -6.9831, -19.0781,  -5.2901,  -2.7625,  -6.8625,  -2.1034,  -7.4545,
        -11.7678, -11.0811,  -9.0438, -12.7363,  -8.4609,  -4.5844,  -9.7559,
         -5.6732,  -6.6080, -10.8248, -12.1216,  -3.2960, -15.0303],
       device='cuda:0')
tensor([-3.8257, -0.9450, -4.8991, -4.6971, -2.7960, -5.1559, -2.8971, -1.0249,
        -1.3108, -1.4970, -2.0211, -3.2424, -3.6960, -2.3293, -4.3490, -3.9578,
        -1.9681, -0.0903, -4.6011, -0.6172], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.7123, -1.1908, -1.9810, -1.2515, -1.3152, -1.4956],
        [-3.9946, -1.4386, -2.1009, -1.1810, -1.5006, -1.2972],
        [-2.5626, -0.9707, -1.2826, -1.6734, -2.4894, -1.3277],
        [-3.4036, -1.6577, -1.6979, -2.2386, -2.0813, -1.7041],
        [-2.9875, -2.9318, -1.8380, -2.2903, -1.9811, -1.3599],
        [-1.5032, -1.1645, -2.2058, -0.9986, -1.6171, -1.3154],
        [-4.0306, -1.1967, -3.5080, -1.1988, -1.7666, -2.4883],
        [-2.3829, -1.3864, -2.9949, -1.1823, -1.5401, -2.1823],
        [-2.0540, -1.1677, -1.2555, -3.0670, -1.7806, -1.4779],
        [-2.5980, -1.9364, -1.6673, -1.1903, -2.2403, -1.1243],
        [-2.2669, -2.3119, -2.2237, -1.4074, -2.1875, -1.4596],
        [-1.3411, -1.2050, -1.4049, -1.0068, -1.8967, -1.7962],
        [-2.0041, -1.2646, -2.7221, -1.1880, -1.8345, -1.8237],
        [-2.8773, -1.4824, -2.1598, -1.5730, -1.5506, -1.2743],
        [-1.7913, -1.5835, -1.4666, -3.8377, -1.4155, -1.5237],
        [-5.4542, -2.0116, -1.9414, -1.5307, -2.5277, -1.2927],
        [-2.6387, -3.5684, -2.2823, -1.3220, -2.4396, -1.1944],
        [-1.3257, -1.1810, -2.1711, -1.3746, -1.2498, -1.2359],
        [-2.1196, -1.2130, -3.0305, -1.6047, -1.3781, -1.3748],
        [-1.5951, -1.2954, -2.1088, -1.2512, -1.6454, -1.7104]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.9463, -11.5129, -10.3064, -12.7832, -13.3887,  -8.8046, -14.1890,
        -11.6689, -10.8026, -10.7568, -11.8570,  -8.6506, -10.8371, -10.9174,
        -11.6183, -14.7583, -13.4454,  -8.5380, -10.7207,  -9.6062],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.7538e+00, -1.7663e-02, -2.0630e+00, -1.2366e-01, -4.4353e-01,
         -7.5584e-01],
        [-4.6498e+00, -1.3187e-01, -2.1520e+00, -9.3104e-02, -1.2920e+00,
         -6.9684e-01],
        [-3.0967e+00, -5.4641e-02, -2.2469e-01, -1.1068e+00, -3.0640e+00,
         -5.5383e-01],
        [-3.1099e+00, -9.3850e-01, -1.6647e+00, -2.4236e+00, -2.2758e+00,
         -1.3689e+00],
        [-3.2831e+00, -3.7552e+00, -1.1855e+00, -1.0636e+00, -2.2263e+00,
         -1.3657e-01],
        [-1.6347e+00, -1.5912e-01, -2.3399e+00, -2.1383e-03, -1.3830e+00,
         -6.0431e-01],
        [-5.3756e+00, -2.6493e-02, -4.6081e+00, -5.9588e-03, -1.5322e+00,
         -1.7456e+00],
        [-2.7355e+00, -3.7153e-03, -3.9265e+00, -1.2440e-02, -1.4670e+00,
         -1.9418e+00],
        [-2.3500e+00, -8.7594e-03, -6.6160e-02, -3.0739e+00, -1.9055e+00,
         -9.3240e-01],
        [-2.2165e+00, -1.4583e+00, -1.4206e+00, -3.0044e-01, -2.8104e+00,
         -3.2509e-02],
        [-1.3486e+00, -2.4223e+00, -3.1145e+00, -1.4911e-01, -2.9556e+00,
         -7.2830e-01],
        [-9.0319e-01, -8.4849e-02, -6.7189e-01, -1.4894e-02, -2.0231e+00,
         -1.6040e+00],
        [-2.7215e+00, -3.3338e-02, -3.4335e+00, -8.8955e-03, -1.6393e+00,
         -1.0692e+00],
        [-3.3389e+00, -2.4230e-01, -2.1997e+00, -1.2045e+00, -1.5122e+00,
         -7.3481e-01],
        [-1.7744e+00, -5.8020e-01, -1.1299e+00, -4.2085e+00, -1.0051e+00,
         -1.1740e+00],
        [-5.0655e+00, -1.5467e+00, -2.4038e+00, -1.3600e+00, -3.6065e+00,
         -3.9048e-01],
        [-2.1226e+00, -3.8654e+00, -2.6858e+00, -1.5900e-01, -3.3275e+00,
         -2.3385e-02],
        [-2.9519e-01, -3.4821e-01, -2.2473e+00, -6.5548e-01, -1.0692e-01,
         -5.6229e-01],
        [-2.6722e+00, -8.7777e-02, -3.8111e+00, -1.0192e+00, -7.2377e-01,
         -5.6016e-01],
        [-1.7457e+00, -3.4774e-01, -2.1943e+00, -1.2758e-01, -1.7607e+00,
         -1.4516e+00]], device='cuda:0')
tensor([ -7.1575,  -9.0156,  -8.1006, -11.7814, -11.6502,  -6.1232, -13.2939,
        -10.0871,  -8.3368,  -8.2388, -10.7183,  -5.3019,  -8.9057,  -9.2324,
         -9.8722, -14.3730, -12.1837,  -4.2154,  -8.8742,  -7.6277],
       device='cuda:0')
tensor([-2.7888, -2.4973, -2.2059, -1.0018, -1.7385, -2.6813, -0.8951, -1.5818,
        -2.4658, -2.5180, -1.1387, -3.3488, -1.9313, -1.6851, -1.7461, -0.3853,
        -1.2617, -4.3226, -1.8465, -1.9786], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.5145, -1.2988, -1.8588, -2.0734, -3.0811, -3.8307],
        [-3.0580, -1.2535, -2.8082, -1.3525, -2.1803, -1.0624],
        [-2.2208, -1.1128, -1.0515, -2.5101, -1.8430, -1.2115],
        [-4.8080, -2.3149, -2.1717, -3.7348, -2.7539, -2.4201],
        [-4.6222, -3.1149, -4.2261, -0.9699, -2.4557, -1.3043],
        [-1.3271, -1.4239, -2.0951, -1.1784, -1.2237, -1.0752],
        [-2.1737, -1.5928, -2.9140, -1.7216, -1.2188, -0.8827],
        [-3.5935, -1.0263, -2.6418, -1.0480, -1.2549, -1.6026],
        [-2.1705, -1.2050, -1.6192, -1.2283, -1.8741, -1.8550],
        [-1.8009, -1.1092, -1.0998, -1.1543, -2.6518, -1.6890],
        [-5.2322, -2.0838, -1.9773, -1.0978, -2.2630, -1.5684],
        [-3.6158, -3.0281, -2.0086, -2.0386, -1.8905, -1.8880],
        [-1.3727, -1.1788, -2.1161, -0.9362, -1.5455, -1.0088],
        [-1.3040, -1.2594, -2.6952, -1.2263, -2.9621, -0.9020],
        [-6.2714, -1.4440, -4.2518, -1.1304, -2.1274, -1.0649],
        [-1.4190, -1.1004, -1.1011, -1.5669, -1.7029, -1.2906],
        [-6.4634, -2.0191, -2.8445, -1.2949, -2.9518, -0.9864],
        [-2.6951, -2.2656, -1.8269, -1.6906, -2.2630, -1.7192],
        [-1.4310, -1.2077, -1.2372, -1.0545, -1.5638, -1.1631],
        [-2.9786, -1.2969, -3.6967, -1.3006, -1.6930, -0.9242]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-15.6573, -11.7148,  -9.9496, -18.2033, -16.6931,  -8.3233, -10.5037,
        -11.1670,  -9.9521,  -9.5049, -14.2225, -14.4696,  -8.1581, -10.3490,
        -16.2898,  -8.1809, -16.5600, -12.4603,  -7.6573, -11.8900],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-5.0134e+00, -9.1951e-03, -1.7547e+00, -1.8524e+00, -3.8172e+00,
         -2.5388e+00],
        [-3.6536e+00, -4.8763e-02, -3.9574e+00, -6.7759e-01, -3.0765e+00,
         -9.9185e-02],
        [-2.3017e+00, -8.5393e-03, -1.2926e-01, -2.7680e+00, -2.1055e+00,
         -2.0165e-01],
        [-4.3681e+00, -2.4893e+00, -3.0960e+00, -4.2092e+00, -3.9729e+00,
         -2.8807e+00],
        [-4.3628e+00, -2.9845e+00, -3.7621e+00, -8.5690e-03, -3.7358e+00,
         -5.5014e-02],
        [-4.0546e-01, -1.2464e+00, -2.0518e+00, -4.0128e-01, -8.7843e-02,
         -1.2361e-01],
        [-2.7166e+00, -4.8815e-01, -4.0126e+00, -1.2468e+00, -2.4935e-03,
         -5.2437e-03],
        [-3.9589e+00, -1.3363e-01, -3.5061e+00, -6.7560e-02, -6.5472e-01,
         -1.6902e+00],
        [-2.5146e+00, -4.6192e-01, -8.8529e-01, -2.3952e-01, -2.1429e+00,
         -1.3644e+00],
        [-1.9733e+00, -4.8198e-03, -1.8327e-01, -1.1016e-01, -3.3093e+00,
         -1.3342e+00],
        [-5.4965e+00, -1.7458e+00, -2.2964e+00, -3.4683e-01, -3.0454e+00,
         -1.0011e+00],
        [-4.1360e+00, -3.2908e+00, -2.0018e+00, -1.0787e+00, -2.0981e+00,
         -2.0589e+00],
        [-1.1299e+00, -4.5085e-02, -2.2610e+00, -5.9545e-02, -1.1485e+00,
         -1.5449e-02],
        [-1.1442e+00, -4.3336e-02, -3.5215e+00, -7.3612e-04, -3.7233e+00,
         -1.8113e-02],
        [-5.7052e+00, -2.0566e-01, -5.3428e+00, -1.1253e-04, -3.0961e+00,
         -1.9073e-06],
        [-9.7270e-01, -6.4557e-01, -9.8922e-03, -1.0922e+00, -1.5489e+00,
         -5.3194e-01],
        [-6.1644e+00, -1.4076e+00, -4.2703e+00, -4.3183e-01, -4.3215e+00,
         -6.3848e-01],
        [-2.2308e+00, -2.2286e+00, -1.2924e+00, -4.7944e-01, -3.3152e+00,
         -1.6767e+00],
        [-1.2377e+00, -6.2120e-02, -8.7917e-04, -1.4805e-03, -8.4555e-01,
         -7.4400e-03],
        [-4.2988e+00, -3.7361e-02, -4.7303e+00, -3.1487e-01, -1.4999e+00,
         -1.0139e-01]], device='cuda:0')
tensor([-14.9857, -11.5131,  -7.5146, -21.0163, -14.9087,  -4.3164,  -8.4718,
        -10.0112,  -7.6087,  -6.9151, -13.9320, -14.6644,  -4.6596,  -8.4512,
        -14.3498,  -4.8012, -17.2342, -11.2231,  -2.1552, -10.9827],
       device='cuda:0')
tensor([-0.6716, -0.2017, -2.4351,  2.8130, -1.7844, -4.0069, -2.0318, -1.1559,
        -2.3433, -2.5898, -0.2904,  0.1948, -3.4985, -1.8978, -1.9400, -3.3797,
         0.6741, -1.2372, -5.5021, -0.9074], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.5816, -1.5163, -2.5685, -1.2442, -2.5459, -1.8087],
        [-2.8032, -0.9067, -1.3624, -1.8209, -2.3689, -1.8082],
        [-4.7431, -2.8011, -2.4108, -1.3584, -2.6824, -1.1251],
        [-2.8119, -2.3633, -2.4565, -2.2082, -2.0869, -1.4979],
        [-1.2893, -2.0435, -2.8072, -0.9706, -1.2460, -1.1020],
        [-2.1563, -1.1190, -2.3748, -1.2767, -1.5433, -1.5985],
        [-1.7870, -1.4967, -2.0133, -1.5294, -1.3992, -1.1737],
        [-1.6683, -1.2578, -2.8548, -2.7319, -2.0293, -1.2796],
        [-3.9618, -1.7877, -2.0061, -2.6757, -2.5200, -1.2489],
        [-2.4130, -3.6309, -2.3360, -1.9838, -1.9380, -1.2204],
        [-1.2638, -1.2001, -2.5339, -1.2126, -1.1629, -1.1205],
        [-2.4795, -1.2910, -2.8013, -1.1983, -1.8553, -0.9851],
        [-4.3531, -1.4129, -1.9248, -1.3689, -1.3800, -0.9763],
        [-1.6697, -1.0834, -1.5909, -1.4646, -2.0773, -1.8616],
        [-3.6338, -2.0000, -2.0910, -1.1671, -1.9163, -1.5035],
        [-2.6672, -2.0012, -1.3456, -1.2144, -1.5967, -1.2621],
        [-1.2642, -1.9674, -1.2943, -1.8236, -1.3459, -1.0045],
        [-4.0246, -1.5220, -1.8472, -1.2982, -1.5472, -1.0109],
        [-1.8433, -1.0582, -2.7243, -1.1525, -1.3581, -1.2050],
        [-1.6710, -1.1478, -1.2021, -1.4402, -1.6820, -1.2255]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.2652, -11.0702, -15.1208, -13.4247,  -9.4586, -10.0687,  -9.3992,
        -11.8217, -14.2002, -13.5223,  -8.4937, -10.6105, -11.4161,  -9.7474,
        -12.3117, -10.0871,  -8.6999, -11.2501,  -9.3413,  -8.3686],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.2494e+00, -7.4819e-01, -3.1276e+00, -6.8910e-01, -2.6141e+00,
         -1.3233e+00],
        [-3.5994e+00, -1.5969e-01, -6.3738e-01, -1.2552e+00, -2.5824e+00,
         -2.0062e+00],
        [-4.3473e+00, -2.3089e+00, -3.0384e+00, -4.1890e-01, -3.8993e+00,
         -6.5551e-01],
        [-2.6869e+00, -2.4230e+00, -3.5346e+00, -1.1175e+00, -2.6099e+00,
         -1.1037e+00],
        [-2.9542e-02, -2.2596e+00, -3.3982e+00, -2.0741e-01, -1.3553e-02,
         -1.9211e-03],
        [-2.9156e+00, -3.5642e-01, -2.1290e+00, -5.4749e-03, -9.1769e-01,
         -8.3404e-01],
        [-1.8918e+00, -2.4632e-01, -1.8574e+00, -1.0249e+00, -9.0542e-01,
         -4.1829e-01],
        [-1.3857e+00, -3.0763e-01, -3.4067e+00, -2.4222e+00, -2.2009e+00,
         -7.8982e-01],
        [-3.9118e+00, -8.2824e-01, -2.6268e+00, -3.2252e+00, -3.8186e+00,
         -3.5609e-01],
        [-1.6924e+00, -3.8233e+00, -2.8282e+00, -1.0711e+00, -2.0063e+00,
         -2.5386e-01],
        [-3.0032e-01, -4.6749e-01, -2.8610e+00, -3.8171e-01, -2.8681e-02,
         -4.4886e-01],
        [-2.9257e+00, -7.0285e-01, -3.1382e+00, -7.1444e-02, -2.1478e+00,
         -1.5747e-01],
        [-4.3562e+00, -8.6846e-02, -1.5554e+00, -6.3985e-01, -3.8028e-04,
         -1.1075e-01],
        [-1.7499e+00, -2.7875e-02, -1.0240e+00, -6.8303e-01, -1.8539e+00,
         -1.5033e+00],
        [-3.4015e+00, -1.5552e+00, -2.9005e+00, -3.3815e-01, -2.0734e+00,
         -7.5521e-01],
        [-1.7913e+00, -1.5763e+00, -1.7525e-01, -2.3140e-01, -1.1321e+00,
         -2.5596e-01],
        [-7.2765e-01, -2.0773e+00, -5.3736e-01, -1.2623e+00, -7.0988e-02,
         -2.5151e-02],
        [-4.9436e+00, -2.8943e-01, -1.7589e+00, -2.0567e-01, -1.1125e+00,
         -3.5763e-07],
        [-1.8777e+00, -1.8322e-01, -3.2363e+00, -7.9898e-02, -3.6156e-01,
         -8.3274e-03],
        [-1.7349e+00, -9.4030e-02, -5.6653e-02, -7.1113e-01, -1.7340e+00,
         -4.2365e-01]], device='cuda:0')
tensor([-11.7516, -10.2403, -14.6683, -13.4755,  -5.9102,  -7.1581,  -6.3441,
        -10.5129, -14.7667, -11.6751,  -4.4881,  -9.1434,  -6.7494,  -6.8421,
        -11.0240,  -5.1623,  -4.7007,  -8.3101,  -5.7470,  -4.7543],
       device='cuda:0')
tensor([-0.5135, -0.8299, -0.4525,  0.0508, -3.5483, -2.9105, -3.0551, -1.3088,
         0.5666, -1.8471, -4.0056, -1.4671, -4.6667, -2.9053, -1.2877, -4.9248,
        -3.9992, -2.9400, -3.5943, -3.6142], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.8932, -1.2232, -1.8727, -1.1498, -2.3532, -1.2404],
        [-4.0803, -1.1620, -1.3094, -1.1681, -2.6879, -1.6072],
        [-3.3674, -2.1660, -1.7231, -1.1250, -1.9824, -1.3964],
        [-2.9008, -4.9626, -1.7610, -1.1172, -1.7571, -1.4990],
        [-1.2534, -1.3721, -1.9249, -1.2444, -1.5448, -1.2120],
        [-3.1100, -1.3901, -3.3508, -1.5266, -2.9182, -2.5583],
        [-0.9790, -1.1555, -4.1343, -1.2778, -2.7097, -1.1126],
        [-1.6819, -1.1773, -1.1659, -1.9859, -2.0861, -1.3802],
        [-2.8129, -2.1804, -1.8014, -1.7753, -1.9799, -1.7437],
        [-2.4410, -2.5397, -2.2575, -1.3665, -2.3873, -1.6179],
        [-2.0247, -1.0994, -1.2139, -1.1461, -1.8870, -0.9940],
        [-2.0698, -1.1783, -1.5671, -1.1776, -2.5360, -1.1467],
        [-5.3247, -1.4376, -3.0916, -1.2569, -1.8543, -1.8166],
        [-1.9891, -1.2560, -2.6178, -1.0705, -2.1649, -2.2479],
        [-3.3111, -1.1453, -1.8646, -1.1822, -3.1408, -1.3638],
        [-3.4350, -1.7748, -2.1297, -1.1584, -1.9181, -1.3913],
        [-2.8472, -1.9790, -2.2010, -1.5225, -2.0319, -1.2488],
        [-1.2677, -1.2522, -2.2435, -1.2187, -1.3243, -1.5796],
        [-1.9690, -1.2825, -2.9212, -1.3411, -1.1877, -0.8537],
        [-2.8368, -1.4333, -1.4528, -1.1469, -1.2856, -1.5028]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.7325, -12.0149, -11.7603, -13.9976,  -8.5516, -14.8541, -11.3690,
         -9.4773, -12.2937, -12.6100,  -8.3652,  -9.6755, -14.7818, -11.3463,
        -12.0079, -11.8072, -11.8304,  -8.8860,  -9.5550,  -9.6582],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.0455e+00, -1.0029e+00, -1.4918e+00, -1.4167e-01, -2.7792e+00,
         -2.7772e-01],
        [-4.9853e+00, -3.6049e-01, -8.9707e-01, -3.3692e-03, -3.2667e+00,
         -9.1864e-01],
        [-3.3771e+00, -1.9213e+00, -3.9504e-01, -1.5832e-01, -2.3763e+00,
         -2.4923e-01],
        [-2.8652e+00, -4.9595e+00, -1.2179e+00, -1.9982e-01, -2.0079e+00,
         -1.0198e+00],
        [-6.0949e-01, -7.6786e-01, -1.9963e+00, -1.8623e-01, -6.6623e-01,
         -5.9170e-03],
        [-4.1548e+00, -8.4188e-02, -5.0645e+00, -7.7067e-01, -3.5318e+00,
         -2.0167e+00],
        [-4.1898e-01, -1.6724e-01, -5.1708e+00, -5.4170e-01, -4.0715e+00,
         -3.0963e-01],
        [-1.8375e+00, -3.9618e-02, -5.0203e-01, -1.8769e+00, -2.7047e+00,
         -6.4934e-01],
        [-2.7920e+00, -2.1338e+00, -1.6551e+00, -2.1468e+00, -2.1642e+00,
         -1.5213e+00],
        [-1.8626e+00, -2.6641e+00, -2.9121e+00, -2.0984e-01, -3.4105e+00,
         -1.5005e+00],
        [-2.9914e+00, -5.4944e-02, -2.2738e-01, -4.0856e-01, -1.9759e+00,
         -1.3470e-02],
        [-2.8057e+00, -1.6775e-01, -9.1796e-01, -1.3830e-01, -2.9447e+00,
         -3.8102e-01],
        [-5.3567e+00, -1.3694e-01, -4.3216e+00, -6.7454e-02, -2.3350e+00,
         -1.5856e+00],
        [-2.3024e+00, -8.1861e-01, -2.4305e+00, -5.6178e-02, -2.6042e+00,
         -2.3456e+00],
        [-4.0070e+00, -1.6550e-01, -1.9250e+00, -5.9425e-03, -3.5392e+00,
         -6.3609e-01],
        [-3.6455e+00, -6.2518e-01, -2.9765e+00, -1.1776e-01, -2.1174e+00,
         -6.4564e-01],
        [-2.8230e+00, -1.6064e+00, -2.4015e+00, -4.9870e-01, -2.5951e+00,
         -1.4899e-01],
        [-6.8174e-01, -5.7245e-01, -2.3793e+00, -3.7653e-01, -4.9385e-01,
         -1.2588e+00],
        [-2.4915e+00, -1.8920e-01, -4.0273e+00, -2.8275e-01, -6.5100e-04,
         -1.9860e-02],
        [-2.5822e+00, -1.2287e+00, -5.0201e-01, -3.1971e-02, -5.3959e-01,
         -9.5905e-01]], device='cuda:0')
tensor([ -7.7388, -10.4316,  -8.4772, -12.2701,  -4.2320, -15.6226, -10.6799,
         -7.6101, -12.4131, -12.5597,  -5.6717,  -7.3554, -13.8033, -10.5575,
        -10.2787, -10.1281, -10.0737,  -5.7627,  -7.0112,  -5.8435],
       device='cuda:0')
tensor([-1.9938, -1.5833, -3.2831, -1.7275, -4.3196,  0.7686, -0.6891, -1.8672,
         0.1194, -0.0502, -2.6935, -2.3200, -0.9785, -0.7888, -1.7292, -1.6792,
        -1.7566, -3.1234, -2.5438, -3.8147], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.5552, -2.5366, -1.9311, -2.7795, -1.9342, -1.3386],
        [-3.5171, -4.6274, -1.9995, -1.5344, -2.0798, -1.7070],
        [-1.2752, -2.8743, -2.9736, -0.9649, -1.4062, -1.0577],
        [-1.4674, -1.2658, -2.2246, -1.2981, -1.6437, -0.8051],
        [-5.6148, -1.2570, -1.6454, -1.2308, -1.4829, -1.2410],
        [-2.9240, -1.0021, -1.3084, -1.9997, -1.7278, -1.3736],
        [-2.5200, -2.3778, -1.7094, -1.1191, -2.1739, -1.3049],
        [-2.2371, -2.5762, -1.8816, -1.1211, -2.0319, -2.4678],
        [-1.4702, -1.1953, -1.9261, -0.9731, -1.3747, -1.3879],
        [-2.7406, -1.3634, -1.8558, -1.1991, -1.5046, -0.8685],
        [-2.9035, -1.3176, -2.6945, -1.0908, -1.6013, -1.2214],
        [-2.6502, -1.3438, -1.1928, -1.9575, -1.8693, -1.1870],
        [-3.1425, -2.2037, -1.5131, -1.2331, -1.7098, -1.1328],
        [-3.7517, -3.0991, -2.8133, -1.6024, -2.6915, -1.3758],
        [-1.4293, -1.1599, -1.6089, -1.1725, -1.4938, -1.3451],
        [-2.0345, -1.2181, -2.7480, -1.1909, -2.3123, -0.8735],
        [-2.8064, -1.3044, -2.7488, -1.1346, -1.3359, -1.1180],
        [-2.2805, -1.6096, -1.3606, -1.6161, -1.5295, -2.4717],
        [-1.1441, -1.3558, -1.3711, -1.1300, -1.5735, -2.3894],
        [-4.0365, -2.7915, -1.6324, -1.0914, -1.9023, -1.2690]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.0751, -15.4652, -10.5519,  -8.7047, -12.4720, -10.3355, -11.2051,
        -12.3156,  -8.3274,  -9.5320, -10.8292, -10.2007, -10.9351, -15.3340,
         -8.2095, -10.3773, -10.4480, -10.8681,  -8.9640, -12.7230],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.9368e+00, -3.3151e+00, -2.1505e+00, -3.4614e+00, -2.0562e+00,
         -5.3894e-01],
        [-3.7000e+00, -5.0160e+00, -2.1212e+00, -4.2354e-01, -2.4835e+00,
         -1.7237e+00],
        [-4.7875e-02, -3.8294e+00, -3.7387e+00, -1.0023e-03, -4.0426e-01,
         -1.0489e-01],
        [-1.8103e+00, -8.9049e-01, -2.0419e+00, -9.9566e-02, -1.3994e+00,
         -4.1950e-02],
        [-5.8874e+00, -4.0370e-01, -7.1205e-01, -4.2938e-01, -1.5946e+00,
         -5.6554e-01],
        [-3.8219e+00, -1.7976e-02, -4.1410e-01, -1.5234e+00, -1.4874e+00,
         -8.5127e-01],
        [-2.1221e+00, -2.4841e+00, -1.4255e+00, -4.0742e-01, -2.7948e+00,
         -1.6079e-01],
        [-1.2241e+00, -2.7165e+00, -1.6899e+00, -6.1704e-02, -2.4149e+00,
         -3.3694e+00],
        [-1.4835e+00, -1.4406e-01, -1.9521e+00, -5.2522e-03, -1.5143e-01,
         -8.2020e-01],
        [-3.8083e+00, -1.5067e-01, -1.7654e+00, -1.1060e-01, -9.9309e-01,
         -3.0994e-06],
        [-3.0531e+00, -1.2647e-01, -3.1395e+00, -4.2290e-02, -1.7057e+00,
         -4.8308e-01],
        [-3.1763e+00, -2.1635e-01, -1.7182e-01, -1.6943e+00, -2.0277e+00,
         -1.4126e-01],
        [-3.0105e+00, -1.9125e+00, -7.2832e-01, -3.9400e-01, -1.3389e+00,
         -1.1597e-02],
        [-4.0186e+00, -3.5660e+00, -4.2853e+00, -4.5546e-01, -4.1467e+00,
         -7.2346e-01],
        [-1.2034e+00, -1.0055e-01, -1.1194e+00, -2.3538e-01, -8.8574e-01,
         -7.5641e-01],
        [-2.6112e+00, -5.2180e-02, -3.5117e+00, -1.8170e-02, -2.5667e+00,
         -1.1090e-01],
        [-2.9362e+00, -7.5985e-02, -3.1204e+00, -7.3664e-02, -3.9177e-01,
         -8.5899e-03],
        [-2.7118e+00, -6.2301e-01, -3.5228e-01, -9.4268e-01, -1.2524e+00,
         -2.9281e+00],
        [-2.0496e-01, -2.7017e-01, -7.8353e-01, -3.6884e-01, -1.0133e+00,
         -2.0835e+00],
        [-4.1392e+00, -3.8375e+00, -3.3992e-01, -1.0805e-02, -1.8119e+00,
         -3.3581e-04]], device='cuda:0')
tensor([-13.4590, -15.4680,  -8.1261,  -6.2836,  -9.5927,  -8.1160,  -9.3948,
        -11.4765,  -4.5565,  -6.8280,  -8.5501,  -7.4277,  -7.3959, -17.1956,
         -4.3009,  -8.8708,  -6.6066,  -8.8103,  -4.7243, -10.1396],
       device='cuda:0')
tensor([ 3.8387e-01,  2.7866e-03, -2.4258e+00, -2.4211e+00, -2.8793e+00,
        -2.2195e+00, -1.8103e+00, -8.3918e-01, -3.7709e+00, -2.7040e+00,
        -2.2792e+00, -2.7729e+00, -3.5393e+00,  1.8616e+00, -3.9087e+00,
        -1.5064e+00, -3.8414e+00, -2.0578e+00, -4.2396e+00, -2.5834e+00],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([[-2.7052, -1.3665, -2.3341, -1.5383, -1.2814, -1.7218],
        [-2.4831, -1.4211, -1.0713, -1.5533, -1.3332, -1.0512],
        [-3.0104, -2.9686, -3.8583, -3.7495, -1.7232, -0.9253],
        [-4.7276, -1.5479, -2.6823, -4.2730, -1.9286, -1.7919],
        [-2.4425, -0.9105, -1.3039, -0.9859, -2.1023, -1.1466],
        [-1.3965, -1.6342, -1.8842, -1.2459, -1.3764, -0.9556],
        [-2.2013, -2.3370, -1.7847, -0.9916, -1.9279, -1.7586],
        [-1.4760, -1.9724, -1.2609, -1.0941, -1.5084, -1.1530],
        [-1.7389, -1.8875, -1.7152, -1.0555, -2.3789, -1.6249],
        [-2.5176, -1.1700, -1.3169, -1.3078, -5.0651, -6.2579],
        [-3.0297, -1.2549, -1.9460, -1.1145, -2.4475, -2.0988],
        [-3.2880, -1.0409, -1.8279, -1.2470, -2.5194, -1.3404],
        [-3.9589, -2.1174, -2.2153, -2.1759, -2.0377, -2.6367],
        [-1.4769, -2.0230, -1.1486, -0.9630, -1.5305, -1.6623],
        [-1.2947, -1.8946, -1.8491, -1.1805, -1.1434, -1.2666],
        [-1.8765, -1.2316, -1.6266, -1.1499, -1.2452, -2.1243],
        [-1.3105, -2.0434, -2.8252, -1.0318, -1.1606, -1.0213],
        [-4.1983, -1.3874, -1.1906, -1.0281, -1.1218, -2.8479],
        [-2.3722, -1.8693, -1.9425, -2.4567, -1.3949, -1.9745],
        [-1.9343, -1.5060, -2.3178, -1.8451, -1.9268, -1.6414]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.9472,  -8.9132, -16.2353, -16.9513,  -8.8917,  -8.4929, -11.0010,
         -8.4647, -10.4009, -17.6353, -11.8914, -11.2636, -15.1419,  -8.8043,
         -8.6289,  -9.2541,  -9.3928, -11.7741, -12.0101, -11.1715],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.4081e+00, -1.4534e-01, -2.7929e+00, -8.4325e-01, -1.0314e-02,
         -1.1517e+00],
        [-1.7572e+00, -9.9309e-01, -2.1694e-01, -8.5783e-01, -5.5983e-02,
         -2.7491e-01],
        [-4.1014e+00, -3.8218e+00, -3.6213e+00, -3.4038e+00, -1.3508e+00,
         -1.3065e-04],
        [-5.1235e+00, -8.1342e-01, -3.2026e+00, -4.5724e+00, -1.8253e+00,
         -1.5261e+00],
        [-3.2846e+00, -1.7474e-01, -1.3419e-01, -2.4844e-02, -2.1806e+00,
         -4.9889e-01],
        [-2.0846e-01, -5.9146e-01, -1.9288e+00, -4.3446e-02, -1.7353e-01,
         -3.0388e-01],
        [-1.5908e+00, -2.5759e+00, -1.5326e+00, -3.4179e-01, -1.9432e+00,
         -1.4030e+00],
        [-1.3516e-01, -1.6455e+00, -4.4215e-01, -2.7202e-01, -6.0346e-01,
         -4.1167e-02],
        [-2.1434e+00, -2.2944e+00, -1.3924e+00, -1.8315e-02, -2.0622e+00,
         -6.1488e-01],
        [-3.5816e+00, -9.2242e-02, -1.5024e-01, -6.8800e-02, -6.7553e+00,
         -4.4896e+00],
        [-3.4747e+00, -1.1082e-01, -1.8978e+00, -3.9346e-02, -3.1530e+00,
         -2.0406e+00],
        [-3.5157e+00, -1.1665e-01, -1.8868e+00, -3.8464e-01, -3.1588e+00,
         -5.2712e-01],
        [-4.8630e+00, -1.8917e+00, -3.3050e+00, -2.9598e+00, -2.3967e+00,
         -3.4777e+00],
        [-8.4068e-02, -1.3461e+00, -2.6005e-01, -5.5573e-02, -3.2195e-01,
         -1.2325e+00],
        [-6.6598e-01, -1.2173e+00, -1.6309e+00, -4.6460e-01, -3.3401e-03,
         -8.6418e-01],
        [-2.0256e+00, -3.8632e-01, -1.2853e+00, -1.7365e-03, -2.2007e-02,
         -2.1859e+00],
        [-7.2700e-01, -1.3468e+00, -4.1279e+00, -5.6671e-02, -8.7732e-03,
         -2.5822e-01],
        [-4.7341e+00, -1.1924e+00, -3.6214e-01, -1.0556e-03, -2.7143e-02,
         -3.1966e+00],
        [-4.1109e+00, -2.2385e+00, -1.9121e+00, -2.3416e+00, -3.6595e-01,
         -8.8664e-01],
        [-2.5924e+00, -5.2971e-01, -2.7420e+00, -1.7650e+00, -1.5515e+00,
         -6.9044e-01]], device='cuda:0')
tensor([ -7.3516,  -4.1560, -16.2993, -17.0633,  -6.2979,  -3.2496,  -9.3873,
         -3.1395,  -8.5255, -15.1378, -10.7162,  -9.5897, -18.8937,  -3.3002,
         -4.8462,  -5.9068,  -6.5254,  -9.5135, -11.8558,  -9.8711],
       device='cuda:0')
tensor([-3.5956, -4.7572,  0.0640,  0.1120, -2.5938, -5.2433, -1.6137, -5.3253,
        -1.8754, -2.4975, -1.1752, -1.6739,  3.7519, -5.5041, -3.7826, -3.3473,
        -2.8674, -2.2605, -0.1543, -1.3004], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.5322, -1.7524, -1.0808, -1.6958, -2.2116, -2.1123],
        [-3.9501, -1.9011, -1.6335, -1.0738, -2.1053, -1.1861],
        [-2.0259, -3.0278, -1.8084, -1.5807, -1.8676, -1.2279],
        [-1.4129, -1.4710, -2.2639, -1.1593, -1.5248, -1.1386],
        [-2.2745, -1.4860, -1.7494, -1.2697, -1.5356, -1.5849],
        [-3.5077, -1.0694, -3.3656, -1.0810, -1.3007, -1.2743],
        [-3.5304, -1.2374, -1.2148, -1.3892, -1.8834, -1.7753],
        [-1.2415, -1.2707, -1.8115, -1.0917, -1.4443, -1.4542],
        [-3.5355, -2.2664, -1.8815, -1.0202, -2.1117, -1.3194],
        [-2.4921, -4.0828, -2.1116, -1.6936, -2.1893, -1.9569],
        [-1.2661, -1.2114, -1.2409, -0.9866, -1.5642, -1.1108],
        [-3.0594, -1.2848, -3.2641, -1.3157, -1.2119, -5.3907],
        [-2.9934, -1.3508, -1.6720, -1.4280, -1.6469, -0.8580],
        [-1.4078, -1.2296, -1.5463, -1.8200, -1.7548, -1.3419],
        [-3.8597, -2.1656, -2.0762, -2.8397, -2.5919, -2.3742],
        [-2.7528, -4.7283, -2.6717, -1.1188, -2.0004, -1.8790],
        [-1.4845, -1.2206, -2.1853, -0.9528, -1.6554, -1.6330],
        [-2.3726, -1.2827, -2.3906, -1.2297, -1.6106, -0.9851],
        [-2.9566, -1.2872, -2.0221, -2.1445, -1.2453, -1.1080],
        [-2.2207, -1.8855, -1.3817, -2.5193, -1.4405, -1.4478]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.3851, -11.8499, -11.5382,  -8.9705,  -9.9002, -11.5987, -11.0305,
         -8.3140, -12.1347, -14.5262,  -7.3800, -15.5266,  -9.9491,  -9.1004,
        -15.9073, -15.1510,  -9.1317,  -9.8713, -10.7638, -10.8955],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.8551e+00, -8.0273e-01, -1.0138e-01, -1.2319e+00, -2.7382e+00,
         -2.4335e+00],
        [-4.1996e+00, -1.2922e+00, -1.1314e+00, -1.4128e-01, -2.6298e+00,
         -4.5347e-04],
        [-8.1726e-01, -3.7164e+00, -1.2232e+00, -3.6811e-01, -1.7699e+00,
         -1.3346e-01],
        [-1.0861e+00, -1.1197e+00, -2.4341e+00, -3.0866e-01, -1.0630e+00,
         -2.1681e-01],
        [-3.2096e+00, -1.9702e-01, -1.3086e+00, -2.2239e-02, -1.0759e+00,
         -6.4757e-01],
        [-3.8181e+00, -2.2798e-01, -4.3258e+00, -9.0912e-02, -7.0541e-01,
         -5.9921e-01],
        [-3.9293e+00, -2.5891e-03, -8.1704e-02, -6.0130e-01, -2.2372e+00,
         -1.3926e+00],
        [-7.6377e-01, -3.6175e-02, -1.8268e+00, -4.2790e-03, -4.4872e-01,
         -9.0893e-01],
        [-3.3490e+00, -2.2795e+00, -1.9164e+00, -7.7035e-02, -2.6500e+00,
         -1.1726e-02],
        [-1.9578e+00, -4.1899e+00, -2.1503e+00, -9.3517e-01, -2.7037e+00,
         -1.9808e+00],
        [-4.2816e-01, -1.2077e-01, -1.8923e-01, -3.2665e-03, -9.7123e-01,
         -1.5611e-01],
        [-4.3749e+00, -2.0175e-02, -4.2850e+00, -2.7172e-01, -1.3294e-01,
         -3.4313e+00],
        [-3.8930e+00, -1.7958e-03, -8.4985e-01, -7.9309e-01, -1.7439e+00,
         -1.7154e-04],
        [-1.0572e+00, -1.1656e-01, -1.4217e+00, -1.6134e+00, -1.9255e+00,
         -8.1468e-01],
        [-3.3663e+00, -2.0817e+00, -2.8382e+00, -3.3114e+00, -3.6808e+00,
         -2.7166e+00],
        [-2.4990e+00, -5.3180e+00, -3.6527e+00, -1.1539e-04, -2.3205e+00,
         -2.0619e+00],
        [-1.5075e+00, -3.9833e-01, -2.3935e+00, -2.7945e-02, -1.5103e+00,
         -1.3749e+00],
        [-3.2803e+00, -6.6117e-03, -3.0989e+00, -7.7432e-03, -1.1881e+00,
         -7.1875e-02],
        [-3.3215e+00, -3.8293e-02, -2.0341e+00, -1.9410e+00, -5.7290e-01,
         -1.2222e-01],
        [-2.5071e+00, -9.7270e-01, -9.6241e-01, -2.6460e+00, -1.0264e+00,
         -1.0705e+00]], device='cuda:0')
tensor([-11.1628,  -9.3947,  -8.0284,  -6.2283,  -6.4610,  -9.7674,  -8.2447,
         -3.9887, -10.2837, -13.9176,  -1.8688, -12.5161,  -7.2818,  -6.9490,
        -17.9950, -15.8523,  -7.2125,  -7.6536,  -8.0300,  -9.1851],
       device='cuda:0')
tensor([-1.2222, -2.4552, -3.5098, -2.7422, -3.4392, -1.8313, -2.7858, -4.3253,
        -1.8510, -0.6086, -5.5112, -3.0105, -2.6673, -2.1514,  2.0877,  0.7013,
        -1.9192, -2.2177, -2.7337, -1.7104], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.6542, -1.9308, -1.8995, -1.1403, -1.7076, -1.3783],
        [-2.7237, -2.9209, -2.0674, -1.2758, -1.6679, -1.4631],
        [-2.1491, -1.1959, -2.4265, -1.1625, -1.5349, -1.1658],
        [-2.7725, -1.6585, -2.5762, -1.3353, -1.2902, -1.3874],
        [-4.7830, -1.5256, -2.4168, -1.1973, -1.4704, -1.4364],
        [-3.6781, -1.1432, -2.6298, -1.2943, -2.0593, -1.7114],
        [-1.7597, -1.1537, -1.1319, -1.2133, -3.1706, -1.8488],
        [-2.6943, -2.1309, -2.5227, -1.1558, -2.0686, -1.3048],
        [-3.4372, -3.3973, -2.6881, -1.9028, -2.0409, -1.2843],
        [-1.2950, -2.4669, -1.9879, -1.7036, -1.3668, -1.1591],
        [-2.5804, -1.3154, -2.6370, -1.4104, -1.6708, -1.1607],
        [-1.8010, -1.1989, -2.9394, -1.1307, -1.7762, -1.3151],
        [-2.3429, -1.2365, -2.2075, -1.1069, -1.9808, -2.5719],
        [-3.2685, -2.3866, -2.1017, -1.2562, -2.0974, -1.2258],
        [-2.4746, -2.7812, -1.9605, -1.3807, -2.7432, -1.2412],
        [-1.1640, -1.2447, -1.6582, -2.9647, -1.6284, -1.7338],
        [-2.9363, -1.4208, -2.4888, -1.3104, -1.2474, -0.8551],
        [-4.7846, -1.2750, -2.7526, -1.2962, -1.1702, -2.0867],
        [-1.4672, -2.0111, -2.6748, -2.1336, -1.1062, -1.2505],
        [-4.8886, -1.3676, -2.2711, -2.0094, -1.8244, -1.6526]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.7107, -12.1188,  -9.6345, -11.0202, -12.8295, -12.5162, -10.2781,
        -11.8771, -14.7506,  -9.9794, -10.7746, -10.1613, -11.4465, -12.3362,
        -12.5815, -10.3938, -10.2587, -13.3652, -10.6434, -14.0138],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.8477e+00, -9.1802e-01, -1.4412e+00, -2.0326e-01, -1.2883e+00,
         -6.8230e-01],
        [-2.3737e+00, -3.3092e+00, -2.1041e+00, -1.8609e-01, -1.2478e+00,
         -9.1975e-01],
        [-3.2893e+00, -2.8289e-02, -2.7402e+00, -1.8966e-01, -7.2998e-01,
         -3.8614e-02],
        [-4.2016e+00, -5.9132e-01, -3.7118e+00, -1.4222e-01, -2.6788e-01,
         -1.4043e+00],
        [-5.1038e+00, -4.6031e-01, -2.9767e+00, -9.9987e-02, -1.1804e+00,
         -1.0891e+00],
        [-4.1240e+00, -4.3716e-01, -2.5746e+00, -3.7315e-01, -2.4771e+00,
         -1.3955e+00],
        [-2.1821e+00, -6.4087e-02, -2.7901e-01, -5.1140e-02, -3.9540e+00,
         -9.4710e-01],
        [-2.2900e+00, -1.9467e+00, -4.3034e+00, -4.4662e-01, -2.5769e+00,
         -3.1032e-01],
        [-3.9012e+00, -3.6856e+00, -3.3935e+00, -1.0234e+00, -2.6612e+00,
         -3.8679e-01],
        [-8.4728e-01, -3.4113e+00, -2.0785e+00, -1.0074e+00, -1.7749e-01,
         -3.2322e-03],
        [-3.3932e+00, -5.7948e-03, -3.2144e+00, -5.0143e-01, -1.6965e+00,
         -3.2118e-01],
        [-1.9615e+00, -3.8009e-01, -4.0961e+00, -1.1553e-01, -2.2025e+00,
         -9.0939e-01],
        [-2.4378e+00, -8.7879e-01, -2.4537e+00, -1.7194e-02, -2.3069e+00,
         -2.8450e+00],
        [-2.9582e+00, -2.6751e+00, -2.9228e+00, -5.7705e-01, -2.5186e+00,
         -2.3103e-03],
        [-2.0211e+00, -2.8582e+00, -1.7556e+00, -4.3532e-01, -4.4274e+00,
         -4.6647e-04],
        [-1.4492e-01, -4.4104e-01, -1.3251e+00, -3.0052e+00, -1.1037e+00,
         -1.3975e+00],
        [-4.1352e+00, -8.1034e-01, -3.2891e+00, -1.3804e-01, -5.2754e-03,
         -1.8136e-01],
        [-4.7731e+00, -1.1323e+00, -3.4979e+00, -8.5563e-01, -2.7764e-04,
         -1.7314e+00],
        [-1.2344e+00, -2.5573e+00, -3.3684e+00, -1.5669e+00, -3.6145e-01,
         -3.3908e-01],
        [-6.3628e+00, -4.6941e-01, -2.3871e+00, -2.0077e+00, -1.4726e+00,
         -1.2296e+00]], device='cuda:0')
tensor([ -9.3807, -10.1407,  -7.0160, -10.3192, -10.9103, -11.3815,  -7.4774,
        -11.8739, -15.0517,  -7.5251,  -9.1325,  -9.6650, -10.9394, -11.6542,
        -11.4980,  -7.4173,  -8.5593, -11.9906,  -9.4276, -13.9292],
       device='cuda:0')
tensor([-3.3300e+00, -1.9781e+00, -2.6185e+00, -7.0097e-01, -1.9192e+00,
        -1.1347e+00, -2.8007e+00, -3.2272e-03,  3.0115e-01, -2.4542e+00,
        -1.6421e+00, -4.9632e-01, -5.0710e-01, -6.8204e-01, -1.0834e+00,
        -2.9765e+00, -1.6994e+00, -1.3746e+00, -1.2158e+00, -8.4601e-02],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([[-3.0936, -2.4077, -2.0727, -2.4049, -1.8761, -1.5429],
        [-2.8661, -3.3341, -2.2916, -2.2051, -1.9054, -1.2176],
        [-1.2784, -2.2197, -2.2174, -1.2241, -1.2875, -1.0603],
        [-2.1072, -1.2340, -2.5792, -1.3516, -1.3286, -2.5716],
        [-2.6460, -1.4204, -2.9835, -1.3910, -1.3277, -1.0646],
        [-2.0868, -1.5472, -2.1711, -2.5477, -2.4012, -2.5182],
        [-2.4128, -3.2112, -1.9313, -1.1662, -1.9202, -1.5465],
        [-2.5537, -2.9573, -2.0367, -1.4008, -2.5041, -1.5285],
        [-1.3920, -1.3331, -1.8644, -1.7384, -1.4369, -1.2215],
        [-1.8135, -1.1799, -2.3918, -1.3182, -1.4359, -1.9226],
        [-3.9581, -1.3755, -2.0114, -1.1056, -1.9727, -1.1303],
        [-6.7756, -1.2804, -1.1958, -1.0985, -2.0218, -1.3636],
        [-5.5514, -2.2448, -1.8581, -1.1371, -2.6723, -1.3526],
        [-2.5644, -2.9391, -3.0855, -1.0256, -2.2165, -1.6715],
        [-1.1815, -1.2236, -1.1505, -1.6618, -1.4865, -1.6244],
        [-4.1022, -1.7688, -1.5686, -1.1834, -1.4183, -1.8569],
        [-2.2644, -1.9902, -1.5946, -1.2839, -1.7125, -0.9765],
        [-1.6231, -2.1936, -2.2632, -1.3260, -1.2950, -1.2588],
        [-3.2427, -4.7537, -1.7883, -1.8667, -2.0545, -0.9762],
        [-5.4011, -1.0622, -1.9730, -4.3683, -1.2414, -1.1910]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.3979, -13.8199,  -9.2873, -11.1723, -10.8332, -13.2723, -12.1883,
        -12.9812,  -8.9864, -10.0619, -11.5536, -13.7358, -14.8163, -13.5027,
         -8.3283, -11.8982,  -9.8221,  -9.9596, -14.6821, -15.2371],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.5927e+00, -2.9389e+00, -2.9055e+00, -2.9288e+00, -1.7577e+00,
         -1.2178e+00],
        [-2.6366e+00, -3.6204e+00, -2.8149e+00, -1.2920e+00, -1.9174e+00,
         -1.7878e-01],
        [-1.9004e-01, -2.7681e+00, -2.3599e+00, -3.7412e-01, -1.3367e-01,
         -2.6964e-01],
        [-2.6136e+00, -2.7941e-01, -2.9852e+00, -3.1196e-01, -1.2769e-01,
         -1.8594e+00],
        [-3.6452e+00, -1.8504e-01, -4.6853e+00, -7.1628e-01, -1.1163e+00,
         -1.9860e-01],
        [-2.3237e+00, -6.7123e-01, -2.1519e+00, -2.4292e+00, -3.0009e+00,
         -3.0575e+00],
        [-1.8665e+00, -4.8126e+00, -1.9830e+00, -3.9692e-01, -2.0214e+00,
         -7.4629e-01],
        [-1.6718e+00, -2.6716e+00, -1.9989e+00, -5.3447e-01, -2.9663e+00,
         -1.2274e+00],
        [-9.6109e-01, -4.8538e-01, -1.8714e+00, -8.0491e-01, -8.5825e-02,
         -4.2475e-01],
        [-2.0787e+00, -1.3020e-01, -2.6837e+00, -2.7163e-01, -8.5572e-01,
         -1.4493e+00],
        [-4.8721e+00, -1.0413e+00, -1.9986e+00, -1.1415e-02, -2.6632e+00,
         -3.7462e-01],
        [-6.1881e+00, -9.8793e-01, -4.7213e-01, -4.5672e-02, -2.4871e+00,
         -8.5276e-01],
        [-5.3231e+00, -2.2271e+00, -1.9503e+00, -2.2533e-01, -3.9768e+00,
         -7.5541e-01],
        [-2.0289e+00, -3.3637e+00, -4.2970e+00, -1.2206e-02, -3.3362e+00,
         -1.4661e+00],
        [-1.4226e-01, -3.3379e-06, -1.0148e-01, -1.0762e+00, -3.1914e-01,
         -1.2499e+00],
        [-5.1563e+00, -2.1273e+00, -8.8446e-01, -7.3029e-02, -8.6715e-01,
         -1.8186e+00],
        [-1.6535e+00, -1.3605e+00, -1.2477e+00, -1.9192e-01, -8.3591e-01,
         -1.6594e-02],
        [-4.8714e-01, -2.9217e+00, -2.7042e+00, -7.6917e-01, -2.1115e-01,
         -3.1963e-01],
        [-4.3144e+00, -4.8950e+00, -1.4383e+00, -1.4794e+00, -1.9870e+00,
         -3.6031e-01],
        [-5.5120e+00, -2.0121e-03, -2.2280e+00, -5.0023e+00, -3.9823e-01,
         -1.4908e-01]], device='cuda:0')
tensor([-14.3415, -12.4600,  -6.0955,  -8.1773, -10.5466, -13.6344, -11.8267,
        -11.0704,  -4.6333,  -7.4692, -10.9612, -11.0337, -14.4581, -14.5041,
         -2.8890, -10.9268,  -5.3061,  -7.4129, -14.4744, -13.2915],
       device='cuda:0')
tensor([ 0.9436, -1.3599, -3.1919, -2.9950, -0.2866,  0.3621, -0.3616, -1.9108,
        -4.3530, -2.5927, -0.5924, -2.7021, -0.3582,  1.0014, -5.4393, -0.9715,
        -4.5160, -2.5467, -0.2076, -1.9455], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.4817, -2.4539, -1.9025, -1.1446, -1.7843, -1.3029],
        [-3.6361, -2.6631, -1.7696, -1.5530, -2.3903, -1.2894],
        [-1.4402, -1.0996, -1.2775, -0.9817, -1.4810, -1.4837],
        [-2.6761, -1.6166, -2.8438, -1.2047, -1.5847, -1.3116],
        [-3.1002, -1.1444, -2.9796, -1.1491, -1.1752, -1.0585],
        [-3.5826, -1.8740, -1.2408, -1.7124, -1.7089, -1.3014],
        [-3.7934, -2.4110, -2.0887, -1.6651, -2.5035, -1.5848],
        [-2.5004, -3.6631, -2.4157, -2.3008, -1.9706, -1.3246],
        [-1.5761, -1.2514, -1.3144, -0.9591, -1.3352, -1.0733],
        [-1.9361, -1.0517, -1.8752, -1.3086, -1.8731, -1.7484],
        [-2.1174, -1.5427, -1.7031, -1.3177, -2.4241, -2.6111],
        [-1.8544, -1.1873, -2.0379, -2.5267, -1.3364, -1.4278],
        [-2.5410, -2.1514, -2.3413, -1.3447, -2.4472, -2.5806],
        [-3.9553, -2.4051, -2.9131, -0.9712, -1.7578, -1.2792],
        [-1.4936, -1.3191, -1.9489, -0.9370, -1.1049, -0.9650],
        [-2.6713, -1.3753, -1.8341, -1.3874, -1.3645, -1.0112],
        [-1.0183, -1.5747, -1.7141, -2.3287, -1.5016, -0.9448],
        [-1.6410, -1.1260, -1.7451, -1.3266, -2.0100, -1.3402],
        [-1.6134, -1.6565, -1.9546, -1.0004, -1.9306, -2.7385],
        [-3.3825, -2.3129, -2.0006, -1.1641, -1.7382, -2.2325]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.0698, -13.3014,  -7.7637, -11.2375, -10.6071, -11.4201, -14.0465,
        -14.1752,  -7.5096,  -9.7931, -11.7163, -10.3705, -13.4063, -13.2817,
         -7.7686,  -9.6437,  -9.0822,  -9.1889, -10.8941, -12.8309],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.5921e+00, -2.4078e+00, -2.0234e+00, -6.0399e-02, -1.5861e+00,
         -1.7724e-01],
        [-4.0649e+00, -2.9274e+00, -1.1969e+00, -3.4379e-01, -3.4933e+00,
         -8.0053e-02],
        [-1.1349e+00, -8.8699e-02, -2.9310e-01, -1.0308e-01, -1.1545e+00,
         -1.0611e+00],
        [-3.8022e+00, -7.2287e-01, -3.8202e+00, -1.9124e-01, -1.0797e+00,
         -5.1929e-01],
        [-3.5558e+00, -7.4212e-02, -3.9793e+00, -1.9641e-01, -2.0338e-01,
         -3.7419e-02],
        [-3.9256e+00, -1.2641e+00, -7.3601e-02, -1.3112e+00, -1.8679e+00,
         -5.1057e-01],
        [-3.5859e+00, -2.4073e+00, -2.6236e+00, -1.5154e+00, -3.2845e+00,
         -1.0053e+00],
        [-1.9869e+00, -4.5821e+00, -3.3885e+00, -1.1501e+00, -2.1344e+00,
         -2.2657e-01],
        [-1.9426e+00, -5.9269e-02, -1.1460e-01, -2.1127e-02, -1.7325e-02,
         -4.5885e-02],
        [-2.3421e+00, -6.3488e-01, -1.7072e+00, -4.6020e-01, -2.2231e+00,
         -1.6889e+00],
        [-2.9418e+00, -7.1714e-03, -6.9584e-01, -2.8720e-01, -3.7308e+00,
         -1.5503e+00],
        [-1.6645e+00, -1.6116e-01, -2.3263e+00, -2.5395e+00, -4.9196e-01,
         -6.5238e-01],
        [-2.6259e+00, -1.9459e+00, -3.7940e+00, -1.2116e+00, -3.4184e+00,
         -3.4712e+00],
        [-3.4538e+00, -2.1911e+00, -2.8557e+00, -1.8136e-01, -1.6070e+00,
         -1.0604e-01],
        [-1.4073e+00, -9.5497e-02, -1.8155e+00, -4.7905e-02, -2.6819e-01,
         -9.8228e-05],
        [-3.1550e+00, -8.0215e-01, -1.7635e+00, -4.8026e-01, -5.1362e-01,
         -6.8998e-04],
        [-1.1616e-01, -3.5533e-01, -1.2645e+00, -2.5105e+00, -1.3953e+00,
         -1.1139e-02],
        [-1.6546e+00, -8.6501e-02, -1.2024e+00, -4.3956e-01, -2.2603e+00,
         -6.9057e-01],
        [-1.8760e+00, -8.8342e-01, -2.1076e+00, -1.0522e-01, -1.7062e+00,
         -2.2963e+00],
        [-2.9536e+00, -2.5777e+00, -2.0653e+00, -3.7432e-01, -1.5379e+00,
         -2.0913e+00]], device='cuda:0')
tensor([ -9.8469, -12.1063,  -3.8354, -10.1355,  -8.0465,  -8.9530, -14.4220,
        -13.4686,  -2.2008,  -9.0564,  -9.2130,  -7.8358, -16.4671, -10.3950,
         -3.6345,  -6.7152,  -5.6528,  -6.3339,  -8.9748, -11.6001],
       device='cuda:0')
tensor([-2.2228, -1.1951, -3.9282, -1.1020, -2.5606, -2.4671,  0.3755, -0.7066,
        -5.3088, -0.7367, -2.5032, -2.5347,  3.0608, -2.8867, -4.1340, -2.9285,
        -3.4294, -2.8550, -1.9193, -1.2307], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.5044, -1.2995, -1.7094, -1.3772, -1.3778, -0.8447],
        [-1.9818, -1.1314, -1.9466, -1.1269, -1.3293, -1.5595],
        [-4.7838, -1.0279, -1.7095, -1.1026, -1.4939, -1.6279],
        [-1.7405, -1.1643, -1.0887, -1.4801, -1.6236, -3.0470],
        [-2.6168, -2.6602, -2.0569, -1.1520, -1.6478, -1.3548],
        [-2.6268, -2.4151, -1.9640, -1.1123, -2.8490, -2.0231],
        [-1.4047, -1.2263, -1.2648, -1.1360, -1.5827, -1.1490],
        [-3.3638, -1.3120, -2.2843, -1.2466, -2.0195, -3.3191],
        [-3.2924, -1.3478, -2.6507, -1.3073, -1.9198, -1.2859],
        [-2.5941, -1.0122, -1.3232, -2.3384, -2.3500, -1.2434],
        [-2.4956, -1.9570, -2.0451, -2.0930, -3.6263, -1.8565],
        [-2.3178, -2.8180, -4.4974, -0.9383, -1.6084, -1.3504],
        [-1.4933, -1.1529, -1.9637, -0.9411, -1.4220, -1.1245],
        [-2.7191, -1.3162, -2.5116, -1.1853, -1.2623, -1.7464],
        [-1.1710, -1.4387, -1.4158, -1.6518, -1.2645, -0.9403],
        [-4.6754, -2.2914, -2.4189, -2.2411, -2.0565, -1.1108],
        [-1.4403, -1.6822, -1.3706, -2.7683, -1.3709, -1.8298],
        [-2.2375, -1.8215, -3.0043, -4.6130, -4.0238, -3.2631],
        [-2.5900, -0.9653, -2.8493, -1.9316, -3.2061, -2.0922],
        [-4.1440, -1.9344, -1.7745, -1.4125, -2.1307, -1.2349]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.1130,  -9.0755, -11.7457, -10.1442, -11.4886, -12.9904,  -7.7635,
        -13.5453, -11.8039, -10.8612, -14.0734, -13.5303,  -8.0975, -10.7410,
         -7.8822, -14.7941, -10.4622, -18.9631, -13.6346, -12.6308],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-5.6232e+00, -4.1495e-02, -1.1321e+00, -3.7381e-01, -7.7275e-01,
         -2.3098e-01],
        [-2.0125e+00, -5.6757e-02, -1.9833e+00, -2.7423e-03, -2.6420e-01,
         -6.4614e-01],
        [-5.4894e+00, -2.4903e-01, -1.2647e+00, -1.7935e-01, -1.1321e+00,
         -1.3782e+00],
        [-2.3749e+00, -1.5974e-05, -1.1462e-01, -7.9606e-01, -1.1551e+00,
         -1.7726e+00],
        [-2.3994e+00, -2.3394e+00, -2.2898e+00, -1.4132e-02, -8.9128e-01,
         -4.9931e-01],
        [-2.0428e+00, -2.3220e+00, -1.8504e+00, -1.4079e-01, -4.0721e+00,
         -2.2006e+00],
        [-9.8268e-01, -6.9525e-02, -1.2635e-01, -9.5523e-02, -8.4341e-01,
         -1.0744e-01],
        [-4.7146e+00, -6.2075e-01, -2.7281e+00, -1.7651e-01, -2.0589e+00,
         -2.2887e+00],
        [-4.0252e+00, -2.2908e-01, -3.5511e+00, -5.0393e-01, -2.6675e+00,
         -7.3641e-01],
        [-2.8247e+00, -9.9953e-02, -8.4044e-01, -2.3339e+00, -3.1319e+00,
         -1.9771e-02],
        [-2.3439e+00, -1.2826e+00, -2.7953e+00, -2.7008e+00, -5.7877e+00,
         -1.8363e+00],
        [-1.2557e+00, -2.6877e+00, -4.2177e+00, -3.7323e-03, -7.0855e-01,
         -3.8913e-01],
        [-1.6596e+00, -3.1809e-03, -1.8488e+00, -9.1240e-03, -9.5870e-01,
         -2.1395e-01],
        [-3.4826e+00, -3.0994e-06, -3.3556e+00, -2.2213e-02, -2.3482e-01,
         -1.0799e+00],
        [-1.2440e-01, -1.2844e-01, -6.7796e-02, -1.3032e+00, -1.8389e-01,
         -7.2605e-02],
        [-3.3980e+00, -3.0004e+00, -2.8803e+00, -2.2326e+00, -2.4503e+00,
         -5.0519e-03],
        [-7.4607e-01, -1.0202e+00, -7.8970e-01, -2.5781e+00, -1.2178e+00,
         -8.9843e-01],
        [-2.7167e+00, -1.3655e+00, -3.5701e+00, -4.5932e+00, -4.1546e+00,
         -2.9926e+00],
        [-3.9649e+00, -7.4973e-02, -3.2064e+00, -1.1439e+00, -3.4856e+00,
         -2.1039e+00],
        [-4.1146e+00, -8.8749e-01, -1.4207e+00, -3.3613e-01, -2.8985e+00,
         -1.1118e+00]], device='cuda:0')
tensor([ -8.1743,  -4.9657,  -9.6929,  -6.2134,  -8.4333, -12.6287,  -2.2249,
        -12.5875, -11.7132,  -9.2508, -16.7467,  -9.2625,  -4.6934,  -8.1751,
         -1.8803, -13.9667,  -7.2503, -19.3927, -13.9797, -10.7693],
       device='cuda:0')
tensor([-2.9387, -4.1099, -2.0528, -3.9308, -3.0553, -0.3617, -5.5385, -0.9578,
        -0.0907, -1.6105,  2.6733, -4.2677, -3.4041, -2.5658, -6.0019, -0.8274,
        -3.2119,  0.4297,  0.3452, -1.8616], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.4381, -1.1781, -1.2763, -1.3231, -2.3280, -2.2935],
        [-3.5212, -2.9346, -2.1056, -1.2326, -2.2927, -2.3377],
        [-3.9274, -2.8806, -2.1489, -1.3116, -1.9187, -1.4567],
        [-1.2343, -1.3502, -1.7652, -1.0600, -1.2762, -1.1267],
        [-2.1605, -1.2735, -2.9540, -1.3498, -1.5002, -1.2181],
        [-3.2110, -1.2534, -2.4050, -1.2398, -2.0596, -1.4804],
        [-2.0402, -1.3555, -2.0199, -2.5195, -2.5617, -1.2128],
        [-5.8004, -2.9043, -1.9150, -1.4158, -2.4364, -1.3215],
        [-1.9511, -2.6200, -1.5827, -1.1707, -1.9057, -2.1991],
        [-1.2811, -1.7015, -1.1877, -1.1580, -1.3146, -1.1835],
        [-2.1881, -1.3244, -2.4429, -1.2893, -1.4067, -0.8058],
        [-3.1119, -1.1992, -2.0261, -1.8514, -2.2460, -1.1151],
        [-1.5294, -2.8995, -1.3569, -3.4041, -1.3107, -1.6723],
        [-4.6315, -1.8824, -1.6648, -1.3784, -2.1506, -1.3097],
        [-2.2101, -2.5410, -2.1743, -1.3271, -1.5845, -1.2515],
        [-1.3585, -1.9973, -2.6462, -1.6974, -1.7003, -1.4425],
        [-1.7927, -1.2881, -3.2469, -1.4882, -1.3786, -1.0068],
        [-3.2072, -1.1713, -1.6772, -1.0045, -1.6901, -1.1091],
        [-1.4986, -1.2678, -1.2989, -1.0301, -2.2561, -1.3353],
        [-1.8369, -1.3247, -1.8141, -1.1521, -1.5843, -1.6550]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.8370, -14.4244, -13.6439,  -7.8127, -10.4562, -11.6492, -11.7095,
        -15.7933, -11.4294,  -7.8264,  -9.4573, -11.5497, -12.1730, -13.0173,
        -11.0885, -10.8422, -10.2013,  -9.8595,  -8.6868,  -9.3671],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.1105e+00, -7.5732e-02, -4.8332e-01, -3.9725e-01, -2.6510e+00,
         -1.5101e+00],
        [-3.6838e+00, -3.0877e+00, -2.7639e+00, -2.7828e-01, -3.2666e+00,
         -2.7892e+00],
        [-4.4451e+00, -3.2524e+00, -2.4097e+00, -1.5753e-01, -2.1932e+00,
         -1.0081e+00],
        [-3.9086e-01, -8.0987e-01, -1.5088e+00, -5.0285e-02, -4.2138e-02,
         -3.2491e-02],
        [-2.9269e+00, -3.3969e-03, -3.5404e+00, -3.7212e-01, -1.0157e+00,
         -3.0019e-01],
        [-3.5755e+00, -2.4785e-01, -2.7871e+00, -3.6011e-01, -2.6621e+00,
         -1.0945e+00],
        [-2.4472e+00, -3.0058e-01, -2.3262e+00, -2.1852e+00, -3.1498e+00,
         -4.2894e-01],
        [-5.4973e+00, -3.7554e+00, -2.1984e+00, -1.1244e+00, -3.5261e+00,
         -1.2958e-01],
        [-7.5434e-01, -2.7520e+00, -6.4868e-01, -7.0608e-02, -2.0817e+00,
         -2.7023e+00],
        [-8.5402e-01, -1.6827e+00, -2.2241e-02, -5.6363e-02, -1.8068e-02,
         -1.4151e-01],
        [-2.9299e+00, -5.4886e-02, -2.8444e+00, -2.0534e-01, -8.9184e-01,
         -1.5777e-03],
        [-3.2701e+00, -2.5332e-02, -2.0890e+00, -1.5388e+00, -2.9779e+00,
         -4.3526e-02],
        [-1.3206e+00, -2.3201e+00, -6.6426e-01, -3.4066e+00, -7.9887e-01,
         -1.4936e+00],
        [-4.3264e+00, -1.2284e+00, -9.1491e-01, -1.0643e+00, -2.6758e+00,
         -1.3442e-01],
        [-1.2582e+00, -2.6449e+00, -2.6788e+00, -1.9596e-01, -7.3603e-01,
         -5.2753e-01],
        [-1.0069e+00, -2.2374e+00, -3.1435e+00, -9.8029e-01, -1.1884e+00,
         -8.1234e-01],
        [-2.0106e+00, -6.4783e-03, -4.0387e+00, -7.7387e-01, -6.7985e-01,
         -2.0529e-01],
        [-3.6215e+00, -6.5820e-01, -1.0854e+00, -2.2072e-03, -1.9316e+00,
         -1.4042e-01],
        [-1.1589e+00, -7.8170e-01, -1.7423e-01, -8.9979e-04, -2.7639e+00,
         -6.6483e-01],
        [-2.2114e+00, -2.1412e-03, -1.9235e+00, -2.4182e-01, -1.2490e+00,
         -1.0482e+00]], device='cuda:0')
tensor([ -8.2279, -15.8695, -13.4660,  -2.8345,  -8.1587, -10.7272, -10.8380,
        -16.2312,  -9.0096,  -2.7749,  -6.9279,  -9.9448, -10.0040, -10.3442,
         -8.0414,  -9.3689,  -7.7148,  -7.4393,  -5.5445,  -6.6760],
       device='cuda:0')
tensor([-2.6091,  1.4451, -0.1778, -4.9782, -2.2975, -0.9220, -0.8716,  0.4379,
        -2.4198, -5.0514, -2.5294, -1.6049, -2.1690, -2.6731, -3.0471, -1.4733,
        -2.4866, -2.4201, -3.1422, -2.6910], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.2565, -2.3396, -1.9634, -1.2191, -2.1606, -1.4315],
        [-3.3742, -3.2962, -3.7016, -2.8912, -1.6311, -1.2096],
        [-1.3275, -1.0819, -1.9751, -0.8185, -1.1370, -1.1682],
        [-2.3213, -1.2537, -2.4355, -1.3249, -1.9167, -2.0649],
        [-1.3714, -1.1995, -2.4994, -1.4674, -2.0379, -1.1278],
        [-1.4339, -2.6354, -2.3270, -2.7076, -1.9685, -1.2663],
        [-5.1411, -2.5008, -1.9297, -1.8960, -2.2264, -1.3949],
        [-2.3385, -3.4207, -2.0484, -2.3584, -1.6795, -1.4787],
        [-1.2864, -2.5203, -2.0184, -0.9803, -1.2845, -1.0058],
        [-2.9836, -1.2709, -2.3958, -1.5063, -1.8915, -1.4830],
        [-1.8562, -1.3270, -2.1878, -1.0516, -2.3722, -1.1877],
        [-3.5339, -1.1131, -1.0986, -1.1328, -1.8724, -1.8689],
        [-1.8930, -2.1437, -2.2700, -1.0719, -2.3040, -1.2478],
        [-3.1124, -3.1767, -2.2788, -1.1541, -1.9497, -1.3501],
        [-1.3555, -1.1664, -2.2493, -1.3568, -1.4097, -1.2483],
        [-2.1961, -1.2682, -3.1119, -1.2038, -1.4564, -1.3145],
        [-2.9487, -1.7421, -1.7187, -1.2168, -1.3719, -0.9207],
        [-1.7463, -2.3141, -1.7295, -2.2244, -1.4203, -1.8967],
        [-1.5098, -1.4312, -2.1139, -3.1028, -1.1923, -1.4171],
        [-2.1054, -1.4354, -2.0457, -2.7283, -1.7750, -2.0521]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.3707, -16.1039,  -7.5082, -11.3171,  -9.7035, -12.3388, -15.0890,
        -13.3242,  -9.0957, -11.5309,  -9.9824, -10.6198, -10.9304, -13.0219,
         -8.7860, -10.5509,  -9.9189, -11.3315, -10.7670, -12.1419],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.3069e+00, -2.4491e+00, -2.3661e+00, -5.8479e-01, -2.9155e+00,
         -9.7491e-01],
        [-3.0868e+00, -3.5871e+00, -4.1065e+00, -2.3927e+00, -1.1519e+00,
         -2.3117e-02],
        [-8.4623e-01, -1.0831e-01, -1.6690e+00, -4.1039e-03, -4.6256e-01,
         -3.5628e-01],
        [-3.2473e+00, -3.0986e-02, -2.9264e+00, -2.7387e-01, -2.0918e+00,
         -1.2030e+00],
        [-1.1181e+00, -1.2703e-02, -2.9704e+00, -8.4662e-01, -2.8308e+00,
         -3.9347e-02],
        [-1.0806e+00, -1.8544e+00, -2.5217e+00, -2.6800e+00, -2.2252e+00,
         -6.4585e-01],
        [-4.9543e+00, -3.0211e+00, -2.0268e+00, -2.1243e+00, -2.9734e+00,
         -4.5718e-01],
        [-1.5705e+00, -3.5800e+00, -2.1711e+00, -1.5787e+00, -1.2584e+00,
         -1.1705e+00],
        [-2.6671e-02, -3.3048e+00, -2.0849e+00, -1.1599e-02, -1.1737e-01,
         -9.9499e-03],
        [-4.2338e+00, -2.7212e-01, -2.6186e+00, -6.9201e-01, -2.2928e+00,
         -4.4125e-01],
        [-2.0308e+00, -1.1738e+00, -2.6346e+00, -9.8435e-02, -3.2868e+00,
         -5.1084e-01],
        [-4.0616e+00, -1.5673e-02, -9.8275e-02, -1.2837e-01, -2.0498e+00,
         -1.8416e+00],
        [-1.1133e+00, -2.0305e+00, -3.3935e+00, -1.9664e-01, -2.9808e+00,
         -6.7658e-03],
        [-3.0391e+00, -3.8612e+00, -2.8787e+00, -6.3139e-02, -2.2751e+00,
         -1.1365e-01],
        [-1.4751e+00, -8.8151e-03, -2.1004e+00, -5.5078e-01, -1.0773e-01,
         -1.2580e-01],
        [-2.5423e+00, -5.6234e-01, -4.3313e+00, -4.8987e-02, -8.8701e-01,
         -5.3964e-01],
        [-2.1704e+00, -7.3368e-01, -1.7648e+00, -1.3625e-02, -1.6751e-03,
         -1.9436e-01],
        [-7.8013e-01, -3.3490e+00, -1.8490e+00, -1.6560e+00, -5.4464e-01,
         -1.8186e+00],
        [-1.3721e+00, -8.8985e-01, -1.9300e+00, -3.0663e+00, -6.0782e-01,
         -7.0279e-01],
        [-2.5153e+00, -4.7421e-01, -2.3162e+00, -3.0908e+00, -1.3828e+00,
         -1.7746e+00]], device='cuda:0')
tensor([-13.5972, -14.3482,  -3.4465,  -9.7733,  -7.8179, -11.0076, -15.5571,
        -11.3292,  -5.5553, -10.5505,  -9.7352,  -8.1953,  -9.7215, -12.2308,
         -4.3687,  -8.9116,  -4.8785,  -9.9973,  -8.5688, -11.5538],
       device='cuda:0')
tensor([ 0.2266, -1.7557, -4.0617, -1.5438, -1.8856, -1.3312,  0.4681, -1.9950,
        -3.5405, -0.9804, -0.2471, -2.4245, -1.2089, -0.7911, -4.4173, -1.6393,
        -5.0404, -1.3341, -2.1982, -0.5881], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.3372, -1.2481, -2.4907, -1.2703, -1.6513, -1.0600],
        [-1.6553, -1.1559, -7.0281, -1.1423, -1.5321, -1.9798],
        [-1.1343, -1.1486, -1.6390, -1.1535, -1.8454, -3.3048],
        [-1.4252, -1.3816, -1.3557, -1.1029, -1.3091, -1.3061],
        [-3.5813, -2.4294, -2.1512, -1.1274, -2.0479, -1.3253],
        [-3.1195, -3.7143, -1.9267, -1.4981, -2.6113, -1.5092],
        [-2.4363, -1.1681, -1.6876, -1.1488, -3.0562, -1.0773],
        [-2.1925, -1.3676, -2.3148, -0.9446, -1.7582, -0.8265],
        [-3.3115, -1.1442, -2.6779, -1.0891, -1.4036, -1.4278],
        [-1.8409, -1.2024, -1.9854, -1.2633, -1.9846, -1.7931],
        [-2.4738, -1.1527, -1.6244, -1.1415, -1.7408, -1.9114],
        [-2.4784, -1.9580, -1.7364, -1.0754, -2.1242, -1.3505],
        [-2.9405, -3.7400, -1.9942, -1.6245, -1.9758, -1.4541],
        [-1.4464, -1.2172, -1.4330, -1.0405, -1.7815, -1.1766],
        [-2.0223, -1.1958, -2.3915, -1.1940, -1.4996, -0.9023],
        [-2.4791, -1.3310, -2.1881, -1.1772, -1.5428, -1.3368],
        [-1.5417, -0.9724, -1.3458, -1.5340, -1.3096, -1.4632],
        [-2.2289, -1.6867, -1.8072, -1.3550, -1.6126, -1.0724],
        [-2.7808, -2.2762, -1.9989, -1.4182, -2.1276, -1.4619],
        [-1.3943, -1.3327, -1.8312, -1.1493, -1.4229, -1.2759]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.0574, -14.4935, -10.2257,  -7.8808, -12.6625, -14.3791, -10.5742,
         -9.4043, -11.0540, -10.0697, -10.0446, -10.7230, -13.7292,  -8.0952,
         -9.2055, -10.0550,  -8.1666,  -9.7628, -12.0635,  -8.4063],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.6042e+00, -8.5681e-02, -3.0489e+00, -4.0389e-01, -1.6737e+00,
         -2.2761e-02],
        [-1.7506e+00, -4.1887e-02, -6.5423e+00, -4.9020e-02, -1.2760e+00,
         -1.7618e+00],
        [-6.1180e-01, -2.2213e-01, -8.1381e-01, -1.4152e-01, -1.7289e+00,
         -2.2880e+00],
        [-4.3975e-01, -2.7813e-01, -8.0694e-01, -2.7129e-02, -1.1123e-01,
         -6.2176e-01],
        [-3.4047e+00, -2.5343e+00, -2.4651e+00, -1.7729e-01, -2.5546e+00,
         -2.8249e-03],
        [-3.1647e+00, -3.7786e+00, -1.7787e+00, -6.9176e-01, -3.5118e+00,
         -1.0996e+00],
        [-3.5645e+00, -7.7271e-02, -1.4398e+00, -4.8393e-02, -3.5326e+00,
         -7.3308e-03],
        [-3.2419e+00, -2.4127e-01, -3.0150e+00, -1.6871e-02, -1.6635e+00,
         -3.1348e-02],
        [-3.3802e+00, -1.1610e-02, -3.0029e+00, -3.6800e-02, -8.6457e-01,
         -1.0623e+00],
        [-2.1781e+00, -2.2472e-01, -1.3975e+00, -3.8213e-01, -2.3590e+00,
         -1.1916e+00],
        [-2.8534e+00, -8.0709e-02, -1.4538e+00, -3.9204e-03, -1.5689e+00,
         -1.7050e+00],
        [-2.0068e+00, -1.4118e+00, -1.2717e+00, -1.2250e-01, -2.7869e+00,
         -1.6596e-01],
        [-2.6385e+00, -3.6952e+00, -1.9048e+00, -8.6339e-01, -2.2649e+00,
         -1.0119e+00],
        [-1.3691e+00, -1.3167e-01, -8.0330e-01, -1.0324e-02, -1.6570e+00,
         -1.6230e-01],
        [-2.6726e+00, -1.6489e-01, -2.9474e+00, -2.7308e-03, -8.5684e-01,
         -1.0836e-04],
        [-2.7095e+00, -1.4398e-02, -2.2864e+00, -4.5730e-02, -1.4927e+00,
         -8.7441e-01],
        [-1.4447e+00, -1.4250e-01, -2.0488e-01, -8.4372e-01, -5.9019e-01,
         -8.4274e-01],
        [-1.6496e+00, -9.8985e-01, -1.8692e+00, -7.2495e-01, -1.1307e+00,
         -7.4579e-02],
        [-2.4703e+00, -2.3574e+00, -2.0942e+00, -1.8774e-01, -2.8000e+00,
         -4.1205e-01],
        [-1.2196e+00, -7.3252e-01, -1.5770e+00, -2.0271e-01, -4.0096e-01,
         -2.6882e-01]], device='cuda:0')
tensor([ -8.8391, -11.4216,  -5.8061,  -2.2849, -11.1388, -14.0251,  -8.6699,
         -8.2099,  -8.3584,  -7.7331,  -7.6658,  -7.7657, -12.3787,  -4.1337,
         -6.6445,  -7.4232,  -4.0687,  -6.4388, -10.3217,  -4.4017],
       device='cuda:0')
tensor([-2.2184, -3.0719, -4.4196, -5.5958, -1.5237, -0.3540, -1.9043, -1.1944,
        -2.6956, -2.3366, -2.3788, -2.9573, -1.3505, -3.9615, -2.5610, -2.6318,
        -4.0979, -3.3240, -1.7418, -4.0046], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.5652, -1.8664, -4.4877, -1.4023, -1.4213, -1.0291],
        [-0.9171, -1.6009, -2.5812, -1.7231, -1.0869, -1.0263],
        [-1.0508, -2.9446, -2.3568, -0.9925, -1.5923, -1.3674],
        [-2.6742, -1.1543, -2.1322, -5.5898, -1.2975, -1.0144],
        [-2.9184, -1.1631, -1.6481, -2.7157, -1.4170, -0.9564],
        [-1.1569, -3.5682, -1.4949, -1.1117, -1.3292, -1.2987],
        [-2.1146, -1.2733, -1.3267, -1.0632, -1.7720, -1.3293],
        [-1.3292, -1.0308, -1.1405, -1.2168, -1.1265, -1.2269],
        [-2.7154, -1.6085, -2.0037, -1.1039, -1.5202, -1.0072],
        [-1.2075, -3.3884, -2.5744, -1.5510, -1.2653, -1.2071],
        [-3.4781, -1.1982, -2.0435, -1.2605, -1.1389, -1.5662],
        [-1.2701, -1.5708, -1.3534, -0.8972, -1.4759, -1.6518],
        [-4.2588, -2.9721, -1.4409, -2.6822, -1.2940, -1.1085],
        [-4.0403, -2.5967, -1.2168, -1.2241, -2.2293, -1.3989],
        [-1.3425, -1.1349, -1.2836, -1.0942, -1.2534, -1.0546],
        [-4.3080, -1.1576, -7.0016, -1.2485, -1.2036, -0.9699],
        [-2.3545, -1.2277, -1.2265, -2.2022, -2.0458, -1.7215],
        [-2.6129, -1.6193, -1.3767, -2.5270, -1.2967, -0.8688],
        [-2.5619, -3.1145, -2.0678, -2.2065, -1.9821, -1.5294],
        [-2.5147, -1.3128, -1.8635, -1.1583, -1.2825, -1.2973]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.7720,  -8.9355, -10.3044, -13.8624, -10.8187,  -9.9595,  -8.8790,
         -7.0707,  -9.9588, -11.1937, -10.6854,  -8.2191, -13.7566, -12.7060,
         -7.1632, -15.8892, -10.7782, -10.3013, -13.4623,  -9.4291],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.8013e+00, -1.9026e+00, -4.0435e+00, -8.0523e-01, -9.9447e-01,
         -7.0806e-02],
        [-2.9492e-02, -1.3754e+00, -2.7340e+00, -1.4432e+00, -3.2848e-02,
         -2.4291e-02],
        [-1.8367e-01, -3.0987e+00, -2.2624e+00, -3.2871e-02, -1.2173e+00,
         -6.7980e-01],
        [-2.3255e+00, -1.1215e-01, -2.2392e+00, -4.5291e+00, -1.9896e-02,
         -4.8983e-03],
        [-2.2147e+00, -2.3224e-02, -1.5135e+00, -3.3472e+00, -2.4774e-01,
         -6.6578e-02],
        [-6.4422e-01, -3.7437e+00, -9.6770e-01, -1.2388e-01, -5.3685e-01,
         -4.7620e-01],
        [-2.2381e+00, -2.6062e-01, -2.3975e-01, -5.7007e-02, -1.6755e+00,
         -5.3301e-01],
        [-6.5173e-01, -1.3645e-01, -1.0523e-03, -3.9468e-01, -9.3107e-02,
         -2.0036e-01],
        [-2.1037e+00, -1.3440e+00, -1.9937e+00, -1.8889e-01, -1.1063e+00,
         -6.9431e-02],
        [-9.2187e-02, -4.7023e+00, -2.8912e+00, -7.7213e-01, -5.7083e-01,
         -7.8465e-01],
        [-3.2512e+00, -2.7461e-01, -1.7005e+00, -7.3551e-01, -2.5764e-01,
         -1.1903e+00],
        [-6.2341e-01, -9.9638e-01, -5.7508e-01, -3.9089e-04, -7.4791e-01,
         -1.2597e+00],
        [-3.6877e+00, -2.7668e+00, -9.2753e-01, -2.7044e+00, -4.2521e-01,
         -3.2694e-01],
        [-3.4187e+00, -2.3871e+00, -2.9731e-01, -4.7931e-01, -2.5153e+00,
         -7.4413e-01],
        [-1.3589e-01, -1.6810e-01, -4.2787e-01, -6.9877e-02, -6.9541e-02,
         -7.9870e-06],
        [-4.4168e+00, -1.6573e-02, -6.0189e+00, -6.5570e-01, -4.2921e-01,
         -7.4057e-02],
        [-2.3995e+00, -2.1275e-01, -6.0502e-02, -2.5496e+00, -2.5042e+00,
         -1.4822e+00],
        [-1.7031e+00, -7.6466e-01, -1.2151e-01, -2.6661e+00, -2.0211e-01,
         -6.4364e-02],
        [-1.9082e+00, -2.9101e+00, -1.7702e+00, -2.3288e+00, -1.5679e+00,
         -1.2650e+00],
        [-1.8229e+00, -8.1190e-01, -1.7646e+00, -4.9220e-01, -1.8123e-03,
         -5.3138e-01]], device='cuda:0')
tensor([ -9.6180,  -5.6391,  -7.4748,  -9.2308,  -7.4130,  -6.4925,  -5.0039,
         -1.4774,  -6.8061,  -9.8134,  -7.4098,  -4.2029, -10.8386,  -9.8419,
         -0.8713, -11.6112,  -9.2088,  -5.5219, -11.7501,  -5.4248],
       device='cuda:0')
tensor([-3.1540, -3.2963, -2.8296, -4.6316, -3.4057, -3.4670, -3.8751, -5.5933,
        -3.1527, -1.3804, -3.2756, -4.0162, -2.9179, -2.8640, -6.2919, -4.2781,
        -1.5694, -4.7795, -1.7122, -4.0043], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.2561, -3.2571, -1.7895, -2.8471, -2.6218, -1.4763],
        [-2.6396, -1.4396, -1.2142, -0.8154, -1.0763, -1.6587],
        [-2.3806, -1.6350, -3.5800, -1.1452, -1.6399, -2.4008],
        [-2.2273, -1.2541, -1.9067, -1.7188, -2.6604, -1.2309],
        [-3.1590, -1.3103, -1.6567, -2.4542, -1.4112, -0.9612],
        [-3.6545, -2.1762, -2.2134, -1.5133, -2.3595, -2.4192],
        [-3.4683, -2.0955, -2.7235, -0.9615, -1.7795, -1.2462],
        [-1.3399, -1.1586, -1.8400, -0.8623, -1.0695, -0.9572],
        [-2.7675, -1.4394, -4.0228, -1.8008, -1.7069, -1.3488],
        [-3.5261, -1.2265, -3.0891, -1.2586, -1.5084, -1.3680],
        [-2.2849, -3.1711, -1.5539, -1.9810, -1.2309, -2.6549],
        [-3.7986, -2.0166, -1.8107, -1.1889, -1.8473, -1.1020],
        [-2.2782, -3.8250, -2.7645, -2.9813, -1.4546, -1.2616],
        [-1.3138, -1.1399, -1.5660, -1.2043, -1.5045, -1.1081],
        [-2.4270, -1.2574, -2.8079, -1.2082, -1.3294, -1.0668],
        [-3.1441, -1.8296, -2.2955, -4.1059, -2.1110, -1.1077],
        [-1.1015, -6.7229, -3.7992, -4.6891, -1.2691, -1.0737],
        [-3.9684, -2.1084, -2.2507, -2.1710, -2.4087, -1.4153],
        [-2.4098, -1.7897, -5.8696, -4.3581, -1.4372, -1.2338],
        [-3.4001, -4.0892, -1.3902, -0.9998, -1.6335, -1.4132]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-14.2478,  -8.8437, -12.7815, -10.9983, -10.9524, -14.3361, -12.2746,
         -7.2275, -13.0862, -11.9767, -12.8767, -11.7642, -14.5652,  -7.8367,
        -10.0966, -14.5937, -18.6555, -14.3225, -17.0983, -12.9260],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.5367e+00, -3.5779e+00, -1.2947e+00, -2.0831e+00, -3.9930e+00,
         -1.0394e+00],
        [-3.7989e+00, -1.2498e+00, -4.2176e-01, -1.5851e-02, -1.0449e-02,
         -1.3221e+00],
        [-2.7776e+00, -5.0411e-01, -4.7885e+00, -8.8935e-03, -1.8711e+00,
         -2.4496e+00],
        [-2.5466e+00, -6.0346e-01, -1.6531e+00, -1.4162e+00, -3.5342e+00,
         -5.3338e-01],
        [-3.7137e+00, -8.6814e-02, -1.3527e+00, -2.4094e+00, -1.0332e+00,
         -2.7961e-03],
        [-3.7578e+00, -2.0411e+00, -3.2949e+00, -1.4903e+00, -3.4522e+00,
         -3.2862e+00],
        [-2.8368e+00, -1.7870e+00, -2.6713e+00, -1.7681e-01, -1.5346e+00,
         -1.7318e-02],
        [-9.1645e-02, -5.2663e-01, -1.4559e+00, -6.5956e-02, -2.7986e-01,
         -1.4423e-01],
        [-3.5379e+00, -6.8417e-03, -6.1233e+00, -1.5122e+00, -1.7791e+00,
         -1.1182e+00],
        [-4.1890e+00, -7.7934e-02, -4.0344e+00, -2.0044e-01, -1.0459e+00,
         -4.0901e-03],
        [-2.9940e+00, -2.1682e+00, -1.1889e+00, -1.4671e+00, -4.5866e-01,
         -3.1347e+00],
        [-3.5860e+00, -1.3426e+00, -8.4773e-01, -5.0258e-01, -1.9061e+00,
         -3.7600e-02],
        [-1.4449e+00, -4.1402e+00, -3.7296e+00, -2.2768e+00, -1.2289e-01,
         -1.2158e-01],
        [-4.3426e-01, -1.9140e-03, -1.0977e+00, -3.9595e-01, -9.4291e-01,
         -2.3983e-01],
        [-3.0443e+00, -4.2597e-03, -3.6932e+00, -5.5997e-02, -4.3984e-01,
         -1.8029e-01],
        [-3.6752e+00, -9.0495e-01, -2.5985e+00, -4.4283e+00, -2.9214e+00,
         -3.8005e-01],
        [-4.5418e-01, -5.3937e+00, -4.8848e+00, -4.9459e+00, -5.1051e-02,
         -3.3213e-03],
        [-4.0089e+00, -1.3891e+00, -3.5945e+00, -2.8183e+00, -3.4708e+00,
         -6.4485e-01],
        [-1.5426e+00, -1.8287e+00, -5.4101e+00, -4.3668e+00, -3.6549e-01,
         -2.2779e-03],
        [-3.6128e+00, -5.7069e+00, -7.9858e-01, -1.9624e-02, -1.4085e+00,
         -7.9325e-01]], device='cuda:0')
tensor([-13.5248,  -6.8189, -12.3999, -10.2869,  -8.5986, -17.3225,  -9.0238,
         -2.5642, -14.0776,  -9.5517, -11.4115,  -8.2226, -11.8359,  -3.1126,
         -7.4179, -14.9084, -15.7330, -15.9264, -13.5159, -12.3396],
       device='cuda:0')
tensor([-0.7230, -2.0248, -0.3815, -0.7114, -2.3539,  2.9865, -3.2508, -4.6634,
         0.9914, -2.4250, -1.4652, -3.5416, -2.7292, -4.7241, -2.6787,  0.3147,
        -2.9225,  1.6038, -3.5824, -0.5864], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.0874, -1.3406, -2.2608, -1.2554, -1.8379, -3.0269],
        [-4.2033, -1.2633, -2.3091, -1.1923, -1.8423, -1.0131],
        [-2.9370, -1.0846, -1.0997, -1.5017, -1.4197, -1.2859],
        [-3.8368, -2.5371, -1.6817, -1.1664, -2.2601, -2.2843],
        [-3.9202, -3.6979, -2.1301, -1.3456, -1.9228, -1.6095],
        [-1.3051, -1.3094, -1.6318, -1.3912, -1.3960, -1.2257],
        [-1.5700, -1.5217, -3.2727, -1.9243, -1.2275, -1.1348],
        [-1.3914, -1.2786, -2.1095, -1.2122, -1.2410, -0.8955],
        [-2.1532, -1.1147, -1.6501, -1.1408, -1.4180, -1.2422],
        [-1.8974, -1.4563, -1.1467, -1.1062, -2.0825, -1.4476],
        [-3.8577, -2.4074, -1.5444, -1.0495, -2.0311, -1.5585],
        [-2.8115, -3.2103, -2.5777, -1.6512, -2.3901, -1.0807],
        [-1.7820, -1.2165, -1.4909, -2.9049, -1.3253, -1.1901],
        [-2.3267, -1.0855, -1.6428, -1.3072, -1.8102, -1.1960],
        [-1.6569, -1.4531, -1.5277, -1.4253, -2.2395, -1.4271],
        [-1.9995, -1.6205, -2.2841, -4.7299, -1.6269, -0.9424],
        [-4.1453, -1.9851, -2.4566, -3.3467, -1.9397, -1.6214],
        [-5.0350, -3.6630, -2.9828, -1.0136, -1.5711, -1.3371],
        [-1.3193, -1.3502, -2.3587, -3.9502, -1.2271, -1.4323],
        [-2.5030, -1.4448, -2.4454, -1.2264, -1.3959, -0.8343]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-11.8090, -11.8234,  -9.3285, -13.7664, -14.6262,  -8.2592, -10.6510,
         -8.1282,  -8.7189,  -9.1367, -12.4487, -13.7214,  -9.9097,  -9.3685,
         -9.7296, -13.2034, -15.4947, -15.6027, -11.6377,  -9.8499],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.7005e+00, -3.7452e-02, -2.7519e+00, -7.2025e-02, -1.7610e+00,
         -2.1966e+00],
        [-4.3748e+00, -4.7721e-02, -2.4875e+00, -3.9163e-02, -2.4066e+00,
         -2.0282e-02],
        [-3.3341e+00, -1.9344e-01, -1.0709e-01, -8.7536e-01, -9.0099e-01,
         -4.8921e-01],
        [-3.7876e+00, -2.7420e+00, -1.3162e+00, -2.2656e-01, -3.0561e+00,
         -2.7801e+00],
        [-4.5666e+00, -4.6236e+00, -2.3684e+00, -1.8222e-01, -2.0793e+00,
         -1.3093e+00],
        [-8.1775e-01, -6.1643e-01, -1.2699e+00, -5.3813e-01, -5.2941e-04,
         -1.3935e-01],
        [-1.7915e+00, -3.5777e-01, -4.6788e+00, -1.6550e+00, -6.7282e-04,
         -2.0482e-01],
        [-1.2126e+00, -2.4211e-02, -2.4243e+00, -2.1137e-01, -5.5664e-01,
         -4.3683e-02],
        [-2.0879e+00, -2.5936e-01, -9.7486e-01, -7.3344e-03, -9.4784e-01,
         -6.0742e-01],
        [-2.4282e+00, -4.4252e-01, -2.6620e-01, -5.2881e-02, -2.2869e+00,
         -5.6645e-01],
        [-3.9178e+00, -2.5278e+00, -8.8970e-01, -1.5042e-01, -2.2626e+00,
         -1.0337e+00],
        [-2.2976e+00, -3.0325e+00, -3.0367e+00, -1.0267e+00, -2.9257e+00,
         -6.8772e-02],
        [-8.1019e-01, -1.1376e-01, -8.3187e-01, -2.2562e+00, -1.1605e-01,
         -4.0917e-01],
        [-3.0037e+00, -1.0200e-01, -1.0046e+00, -4.2477e-02, -1.9120e+00,
         -5.3654e-01],
        [-1.8598e+00, -1.1074e+00, -4.4092e-01, -7.9541e-01, -3.1291e+00,
         -9.3091e-01],
        [-2.5303e+00, -6.2251e-01, -3.2155e+00, -4.8952e+00, -1.5622e+00,
         -3.3930e-02],
        [-4.0697e+00, -1.7083e+00, -3.5751e+00, -4.0449e+00, -1.9879e+00,
         -1.2128e+00],
        [-4.0600e+00, -3.6373e+00, -2.5044e+00, -1.1371e-03, -5.4390e-01,
         -2.2668e-01],
        [-1.8362e-01, -1.0039e+00, -2.6331e+00, -3.0231e+00, -7.8816e-02,
         -1.1478e+00],
        [-3.1469e+00, -4.0685e-01, -3.1603e+00, -5.6730e-02, -9.8193e-01,
         -8.4788e-02]], device='cuda:0')
tensor([ -9.5195,  -9.3760,  -5.9002, -13.9085, -15.1294,  -3.3821,  -8.6886,
         -4.4728,  -4.8847,  -6.0432, -10.7819, -12.3881,  -4.5372,  -6.6014,
         -8.2635, -12.8597, -16.5986, -10.9735,  -8.0704,  -7.8375],
       device='cuda:0')
tensor([-2.2895, -2.4474, -3.4284,  0.1421,  0.5032, -4.8771, -1.9624, -3.6554,
        -3.8342, -3.0935, -1.6668, -1.3334, -5.3725, -2.7671, -1.4661, -0.3437,
         1.1039, -4.6292, -3.5673, -2.0124], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.0312, -2.4115, -2.3870, -2.6208, -3.1927, -2.1719],
        [-2.2898, -3.1829, -3.2422, -1.6547, -1.7723, -1.3505],
        [-1.3815, -2.0489, -3.0928, -1.6432, -1.3999, -1.3484],
        [-2.3148, -1.3747, -2.8948, -1.4387, -1.5867, -0.7851],
        [-4.7386, -0.9967, -2.0676, -1.1206, -1.3803, -1.3187],
        [-1.7844, -1.5267, -2.3338, -1.5776, -3.0224, -2.7351],
        [-2.2867, -2.2566, -2.0599, -1.0264, -1.9970, -1.3715],
        [-3.3656, -3.6744, -2.1897, -2.6361, -1.4764, -1.4321],
        [-1.8914, -1.3132, -1.3417, -1.4151, -1.5422, -1.1314],
        [-1.9216, -1.5964, -1.9670, -1.2647, -1.6742, -2.1324],
        [-2.4949, -1.0669, -3.0536, -1.0979, -1.4489, -1.8090],
        [-7.1521, -1.2111, -1.0885, -1.8554, -1.6087, -1.1288],
        [-2.9735, -3.7845, -1.6019, -1.1707, -2.0437, -1.2839],
        [-2.8191, -4.5066, -2.0041, -1.3412, -2.1801, -1.5797],
        [-1.3213, -1.2965, -1.4594, -1.3832, -1.5317, -1.6035],
        [-1.4194, -1.6128, -2.2666, -1.3483, -1.2008, -0.8955],
        [-2.9703, -1.3309, -2.2824, -1.0842, -1.3289, -1.1228],
        [-4.8766, -1.1544, -2.0394, -1.1842, -1.9641, -2.5344],
        [-2.2903, -1.2624, -1.3285, -1.1977, -1.8574, -1.4837],
        [-2.8072, -2.7990, -1.6879, -1.1050, -1.6098, -1.5632]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-16.8151, -13.4923, -10.9148, -10.3949, -11.6225, -12.9799, -10.9981,
        -14.7743,  -8.6350, -10.5564, -10.9712, -14.0446, -12.8582, -14.4307,
         -8.5955,  -8.7433, -10.1194, -13.7530,  -9.4199, -11.5723],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.0039e+00, -2.8586e+00, -3.9292e+00, -3.3589e+00, -5.0865e+00,
         -2.7628e+00],
        [-1.2783e+00, -3.3405e+00, -3.1615e+00, -8.0984e-01, -1.3722e+00,
         -5.2874e-01],
        [-1.9698e-01, -2.3664e+00, -3.3734e+00, -1.2562e+00, -3.7543e-01,
         -5.0679e-01],
        [-2.9596e+00, -6.3801e-04, -3.4961e+00, -6.2673e-01, -1.4103e+00,
         -3.5968e-02],
        [-5.5564e+00, -9.1049e-02, -2.4131e+00, -5.7374e-02, -1.0125e+00,
         -1.5630e-01],
        [-1.8792e+00, -6.3034e-01, -2.6315e+00, -8.4323e-01, -3.2526e+00,
         -2.7319e+00],
        [-1.5662e+00, -2.1404e+00, -2.6852e+00, -1.1333e-01, -2.3963e+00,
         -2.2760e-01],
        [-2.9887e+00, -3.7930e+00, -2.3881e+00, -1.9182e+00, -7.7368e-01,
         -8.9924e-01],
        [-2.7590e+00, -7.1216e-01, -3.9391e-01, -4.0135e-01, -7.2592e-01,
         -9.0133e-02],
        [-2.7716e+00, -3.7428e-01, -2.0155e+00, -1.8099e-01, -1.4360e+00,
         -1.0794e+00],
        [-2.6149e+00, -4.6379e-01, -3.9978e+00, -3.2201e-02, -1.2646e+00,
         -2.1325e+00],
        [-5.8020e+00, -3.4752e-03, -2.6923e-01, -1.5834e+00, -1.6078e+00,
         -7.0645e-02],
        [-3.0351e+00, -4.4166e+00, -6.7914e-01, -3.5529e-02, -2.4590e+00,
         -2.9160e-01],
        [-2.4581e+00, -5.4018e+00, -1.9656e+00, -2.2710e-01, -3.0156e+00,
         -1.3374e+00],
        [-1.0876e+00, -5.7832e-01, -8.9914e-01, -5.4092e-01, -7.4241e-01,
         -1.2131e+00],
        [-1.2926e+00, -5.6919e-01, -2.8645e+00, -2.5818e-01, -2.0883e-03,
         -5.0060e-02],
        [-3.4737e+00, -1.7476e-01, -2.8038e+00, -3.1083e-03, -9.5720e-01,
         -4.4017e-01],
        [-5.1982e+00, -5.3109e-01, -1.5840e+00, -4.2249e-02, -2.2444e+00,
         -2.6411e+00],
        [-2.8351e+00, -8.6214e-02, -8.7338e-01, -6.1607e-02, -1.8811e+00,
         -7.8503e-01],
        [-2.5336e+00, -3.4252e+00, -1.0644e+00, -2.3427e-01, -9.8038e-01,
         -8.7804e-01]], device='cuda:0')
tensor([-21.9999, -10.4911,  -8.0752,  -8.5293,  -9.2867, -11.9688,  -9.1290,
        -12.7608,  -5.0824,  -7.8577, -10.5057,  -9.3365, -10.9169, -14.4056,
         -5.0616,  -5.0366,  -7.8527, -12.2411,  -6.5224,  -9.1160],
       device='cuda:0')
tensor([ 5.1848, -3.0012, -2.8396, -1.8656, -2.3358, -1.0111, -1.8691, -2.0135,
        -3.5525, -2.6987, -0.4656, -4.7080, -1.9413, -0.0252, -3.5339, -3.7067,
        -2.2667, -1.5119, -2.8974, -2.4563], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.5073, -1.2554, -1.5189, -1.1149, -2.0488, -1.8670],
        [-2.4980, -1.9555, -1.9510, -1.2006, -1.5940, -1.9098],
        [-3.0079, -1.9582, -2.6060, -2.0365, -1.6451, -1.1056],
        [-1.2332, -1.4654, -2.6067, -4.9019, -1.3295, -1.1863],
        [-2.1532, -1.2670, -2.6663, -1.3352, -2.6121, -2.5181],
        [-2.6117, -1.0344, -2.6065, -1.1163, -2.2065, -1.4323],
        [-4.5110, -0.9756, -1.7086, -2.0277, -1.4345, -1.5575],
        [-3.7915, -2.2522, -2.0761, -1.6809, -2.6911, -3.7779],
        [-4.4189, -2.9825, -3.1912, -0.9457, -2.2246, -1.3389],
        [-1.6555, -1.4989, -1.4974, -2.3774, -1.4020, -1.0107],
        [-1.3219, -1.3232, -1.6985, -2.0767, -1.2022, -1.9642],
        [-1.4427, -2.1174, -1.3663, -2.1949, -1.3339, -3.0902],
        [-1.8666, -1.3510, -1.0278, -1.4574, -1.3972, -1.8399],
        [-1.4174, -1.7158, -1.3549, -2.3542, -1.0826, -0.8972],
        [-1.3506, -3.5572, -2.7105, -3.2876, -1.7680, -1.9008],
        [-2.2819, -1.8718, -4.0081, -1.6932, -5.0546, -3.4085],
        [-3.9690, -2.5827, -2.2900, -1.4179, -2.3962, -3.7966],
        [-2.0344, -3.8029, -2.2713, -2.5997, -3.0262, -2.8690],
        [-1.3497, -1.2358, -1.5629, -1.3326, -1.3385, -1.0972],
        [-3.2019, -1.1418, -2.6452, -1.6625, -1.8673, -1.0976]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.3124, -11.1089, -12.3593, -12.7230, -12.5519, -11.0077, -12.2148,
        -16.2698, -15.1018,  -9.4419,  -9.5867, -11.5456,  -8.9398,  -8.8222,
        -14.5747, -18.3182, -16.4525, -16.6035,  -7.9166, -11.6163],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.4185e+00, -3.4800e-01, -1.1982e+00, -9.6317e-02, -2.1091e+00,
         -1.2200e+00],
        [-2.0773e+00, -1.2442e+00, -2.1649e+00, -6.7589e-02, -3.4379e-01,
         -1.8411e+00],
        [-2.7070e+00, -1.4718e+00, -2.6466e+00, -1.6142e+00, -1.6711e+00,
         -1.3403e-02],
        [-1.4869e-01, -1.2010e+00, -2.8939e+00, -4.0278e+00, -5.6347e-01,
         -3.8806e-01],
        [-2.8814e+00, -8.7607e-02, -3.7615e+00, -2.6300e-01, -3.1212e+00,
         -2.3659e+00],
        [-3.1746e+00, -1.7114e-01, -3.4480e+00, -2.0437e-01, -2.8797e+00,
         -1.2616e+00],
        [-5.3710e+00, -1.3760e-02, -1.7329e+00, -1.8570e+00, -7.9537e-01,
         -8.6754e-01],
        [-4.2369e+00, -2.2431e+00, -2.8158e+00, -1.9921e+00, -4.3555e+00,
         -5.6213e+00],
        [-3.7345e+00, -2.6973e+00, -2.7739e+00, -5.3573e-02, -2.9370e+00,
         -2.7372e-01],
        [-2.2814e+00, -4.9403e-01, -4.2148e-01, -2.3316e+00, -1.1468e+00,
         -3.5911e-01],
        [-8.8960e-01, -3.5732e-01, -1.7325e+00, -1.8205e+00, -7.2311e-02,
         -1.6623e+00],
        [-6.3967e-01, -1.2910e+00, -4.4590e-01, -1.9086e+00, -1.0540e+00,
         -3.0533e+00],
        [-8.6350e-01, -1.2059e+00, -2.1638e-01, -3.4789e-01, -7.7912e-01,
         -1.5159e+00],
        [-5.9256e-01, -1.7412e+00, -6.7192e-01, -2.1932e+00, -1.5259e-01,
         -2.3279e-01],
        [-8.8805e-01, -4.5332e+00, -3.6019e+00, -3.5962e+00, -1.3015e+00,
         -1.1870e+00],
        [-3.2690e+00, -1.0853e+00, -4.0804e+00, -1.0053e+00, -5.7432e+00,
         -4.0697e+00],
        [-3.8791e+00, -1.8420e+00, -2.8327e+00, -5.7522e-01, -3.3045e+00,
         -3.5158e+00],
        [-1.0935e+00, -4.9443e+00, -3.3701e+00, -1.4453e+00, -4.6894e+00,
         -3.7976e+00],
        [-7.4963e-02, -9.4641e-02, -1.0835e+00, -1.2370e+00, -4.6207e-01,
         -3.4332e-05],
        [-3.6575e+00, -8.4926e-01, -2.5572e+00, -1.1922e+00, -2.1796e+00,
         -4.0375e-01]], device='cuda:0')
tensor([ -6.3901,  -7.7388, -10.1240,  -9.2229, -12.4806, -11.1395, -10.6375,
        -21.2647, -12.4699,  -7.0343,  -6.5344,  -8.3926,  -4.9287,  -5.5844,
        -15.1079, -19.2528, -15.9493, -19.3402,  -2.9522, -10.8395],
       device='cuda:0')
tensor([-2.9223, -3.3700, -2.2353, -3.5001, -0.0713,  0.1318, -1.5773,  4.9949,
        -2.6319, -2.4075, -3.0523, -3.1530, -4.0111, -3.2378,  0.5332,  0.9346,
        -0.5032,  2.7367, -4.9643, -0.7768], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.1308, -1.0210, -1.1991, -0.9573, -1.3544, -1.0641],
        [-3.0480, -3.4497, -1.2208, -1.0285, -1.7122, -1.4256],
        [-1.3351, -2.4785, -2.4195, -1.6112, -1.1648, -1.4628],
        [-1.0998, -2.5056, -1.6016, -2.4778, -1.7547, -1.0482],
        [-1.7939, -2.7262, -2.6568, -1.7766, -1.3253, -1.3971],
        [-1.1068, -1.3555, -1.2022, -1.0312, -1.7816, -1.3394],
        [-4.7426, -2.3833, -2.3297, -3.5346, -1.3966, -0.9390],
        [-3.9161, -1.8858, -2.0651, -1.2481, -1.8435, -2.7139],
        [-1.2306, -1.2286, -1.2854, -1.0480, -1.2611, -0.9927],
        [-2.7580, -1.5744, -1.9440, -1.1150, -1.2266, -1.0562],
        [-1.0510, -1.1292, -1.1712, -1.1895, -1.1950, -1.8522],
        [-1.9180, -1.0326, -1.9287, -1.0798, -1.0947, -1.3097],
        [-2.2563, -1.5500, -1.6514, -1.7843, -2.1116, -1.0431],
        [-0.9544, -1.9221, -2.9584, -1.2278, -1.2530, -1.0605],
        [-3.8739, -1.4459, -4.6485, -2.2678, -2.2491, -1.9885],
        [-1.5356, -1.5950, -1.7632, -2.3368, -3.0698, -1.7144],
        [-3.6846, -3.0130, -1.7273, -2.9043, -2.0693, -2.0301],
        [-3.4609, -2.6514, -2.4772, -1.1984, -2.1639, -1.4024],
        [-1.9265, -1.6039, -1.7766, -1.1892, -1.1918, -1.4020],
        [-3.1532, -1.0218, -2.1217, -1.3089, -1.3182, -1.0889]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -6.7268, -11.8848, -10.4718, -10.4876, -11.6760,  -7.8168, -15.3257,
        -13.6724,  -7.0463,  -9.6741,  -7.5882,  -8.3635, -10.3967,  -9.3761,
        -16.4736, -12.0147, -15.4287, -13.3541,  -9.0900, -10.0127],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.3239e-01, -1.7183e-02, -5.8029e-02, -2.6635e-02, -9.5639e-01,
         -1.1606e-02],
        [-2.6230e+00, -3.6081e+00, -4.8001e-01, -3.5318e-02, -1.3804e+00,
         -8.0048e-01],
        [-9.0819e-02, -2.9755e+00, -1.9204e+00, -7.8646e-01, -7.1488e-01,
         -5.2579e-01],
        [-3.3018e-01, -2.5200e+00, -1.3104e+00, -2.4893e+00, -1.2828e+00,
         -3.2977e-01],
        [-1.5042e+00, -2.5981e+00, -2.8079e+00, -1.6255e+00, -7.5415e-01,
         -8.3580e-01],
        [-3.9703e-02, -6.8569e-01, -4.4205e-01, -1.5098e-01, -1.3799e+00,
         -8.9439e-01],
        [-3.7019e+00, -1.9842e+00, -2.7597e+00, -3.9707e+00, -3.0579e-01,
         -3.8743e-04],
        [-2.5988e+00, -1.1925e+00, -2.2703e+00, -1.3822e-01, -1.7559e+00,
         -3.0874e+00],
        [-7.7311e-01, -4.8926e-01, -5.1878e-01, -1.0781e-01, -3.0028e-01,
         -2.6370e-02],
        [-3.2433e+00, -1.6339e+00, -1.8203e+00, -2.3609e-03, -8.7476e-01,
         -2.1566e-01],
        [-4.1380e-03, -4.0015e-02, -5.2998e-02, -2.2396e-01, -5.1731e-03,
         -1.3690e+00],
        [-1.4680e+00, -3.5792e-01, -1.9439e+00, -1.3131e-01, -4.5239e-02,
         -4.3115e-01],
        [-1.5210e+00, -1.3005e+00, -1.3036e+00, -1.2471e+00, -2.3233e+00,
         -5.1971e-01],
        [-1.0568e-02, -1.7920e+00, -3.0962e+00, -5.3976e-01, -4.0399e-01,
         -4.3981e-03],
        [-4.1851e+00, -6.7521e-01, -4.9555e+00, -2.4737e+00, -2.1885e+00,
         -1.6072e+00],
        [-1.2820e+00, -9.7678e-01, -1.5462e+00, -1.8521e+00, -3.1522e+00,
         -1.5850e+00],
        [-3.3173e+00, -2.3161e+00, -1.3567e+00, -3.4174e+00, -2.2309e+00,
         -1.7105e+00],
        [-2.5296e+00, -2.5175e+00, -2.6859e+00, -1.6825e-03, -2.2695e+00,
         -8.3770e-01],
        [-2.7826e+00, -1.3767e+00, -1.5120e+00, -8.1586e-01, -3.3395e-02,
         -1.1031e+00],
        [-3.8732e+00, -1.9056e-01, -2.4530e+00, -2.9207e-01, -6.7716e-01,
         -1.6746e-01]], device='cuda:0')
tensor([ -1.3022,  -8.9274,  -7.0139,  -8.2624, -10.1257,  -3.5928, -12.7226,
        -11.0430,  -2.2156,  -7.7903,  -1.6952,  -4.3775,  -8.2151,  -5.8468,
        -16.0852, -10.3943, -14.3490, -10.8419,  -7.6238,  -7.6534],
       device='cuda:0')
tensor([-5.4246, -2.9574, -3.4580, -2.2252, -1.5502, -4.2240, -2.6031, -2.6294,
        -4.8307, -1.8839, -5.8930, -3.9860, -2.1816, -3.5293, -0.3884, -1.6204,
        -1.0798, -2.5122, -1.4662, -2.3593], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.1584, -1.2246, -1.3533, -2.7048, -1.7915, -1.5530],
        [-2.0589, -1.3822, -2.2729, -1.2519, -1.2038, -0.9238],
        [-3.1108, -1.8495, -1.8092, -1.3732, -1.3760, -1.2342],
        [-1.8964, -1.4131, -1.3112, -2.0230, -1.3549, -1.0923],
        [-2.6166, -2.1372, -1.7128, -3.1628, -1.8554, -2.5634],
        [-3.7132, -1.3877, -2.4434, -2.4993, -1.5284, -0.9973],
        [-2.9073, -0.9376, -1.3203, -0.9249, -3.2394, -1.9514],
        [-1.9122, -1.1729, -1.5065, -1.1768, -1.3442, -5.6114],
        [-2.9678, -2.3147, -1.9876, -1.1346, -2.2763, -1.2488],
        [-2.7802, -3.0649, -1.8074, -1.5492, -2.4717, -2.0287],
        [-1.4917, -1.9528, -2.2145, -2.0188, -1.4024, -1.1765],
        [-2.1297, -1.4680, -2.5047, -1.4495, -1.5283, -1.1400],
        [-5.0998, -1.0897, -2.6092, -1.0853, -1.7387, -1.1632],
        [-2.9412, -1.0624, -1.3845, -1.3044, -1.3086, -1.1826],
        [-3.7441, -2.2430, -3.6239, -1.1037, -1.5811, -1.0218],
        [-2.8378, -2.0069, -1.9646, -1.4468, -1.7502, -1.5256],
        [-1.6714, -1.2311, -1.2106, -1.3093, -1.4451, -1.2806],
        [-1.9607, -1.2505, -3.2124, -1.2439, -2.5235, -2.1457],
        [-1.2980, -1.6511, -2.7205, -1.1745, -2.9497, -1.7331],
        [-2.7553, -1.3414, -1.6108, -1.4373, -1.4088, -1.1242]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.7856,  -9.0934, -10.7528,  -9.0909, -14.0482, -12.5694, -11.2809,
        -12.7239, -11.9299, -13.7021, -10.2568, -10.2202, -12.7859,  -9.1836,
        -13.3176, -11.5319,  -8.1482, -12.3367, -11.5269,  -9.6778],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.2839e-01, -1.6963e-03, -6.5660e-01, -2.7682e+00, -1.5568e+00,
         -1.1154e+00],
        [-2.4007e+00, -8.1260e-01, -2.6487e+00, -1.6280e-01, -2.3273e-01,
         -1.6063e-02],
        [-2.4659e+00, -9.7687e-01, -1.9607e+00, -5.1002e-01, -1.9745e-03,
         -5.4965e-01],
        [-1.0931e+00, -9.4261e-01, -7.6110e-01, -1.5123e+00, -1.6270e-01,
         -9.7211e-02],
        [-3.8853e+00, -3.0395e+00, -1.5481e+00, -3.0644e+00, -1.7808e+00,
         -1.6221e+00],
        [-3.9972e+00, -3.5939e-01, -2.8890e+00, -2.8996e+00, -1.0809e+00,
         -8.6474e-02],
        [-3.7505e+00, -2.2127e-01, -3.0070e-01, -7.9390e-03, -3.6779e+00,
         -2.4142e+00],
        [-2.4094e+00, -6.6784e-02, -1.1431e+00, -2.9064e-03, -3.9782e-01,
         -4.2570e+00],
        [-2.7859e+00, -2.4396e+00, -2.1811e+00, -2.5862e-01, -3.2315e+00,
         -1.7862e-01],
        [-2.5263e+00, -3.4665e+00, -1.3922e+00, -6.3435e-01, -3.7782e+00,
         -2.2566e+00],
        [-1.5165e+00, -2.3129e+00, -2.5978e+00, -1.4317e+00, -9.3599e-02,
         -8.4676e-02],
        [-3.1284e+00, -3.3545e-02, -3.4826e+00, -4.2130e-01, -1.0155e+00,
         -1.1069e-01],
        [-5.2086e+00, -6.0531e-02, -3.0415e+00, -5.9479e-03, -2.1453e+00,
         -6.0149e-02],
        [-3.2488e+00, -1.5327e-02, -5.0732e-01, -4.0914e-01, -6.1408e-01,
         -2.6246e-01],
        [-3.5673e+00, -2.2876e+00, -5.2016e+00, -2.7693e-01, -1.0608e+00,
         -1.5617e-02],
        [-2.3748e+00, -1.5621e+00, -1.9613e+00, -2.3848e-01, -1.2266e+00,
         -5.2296e-01],
        [-2.1709e+00, -1.7881e-01, -2.2296e-03, -3.6720e-01, -1.4055e-04,
         -2.6575e-01],
        [-2.3857e+00, -4.7219e-02, -4.0986e+00, -2.0925e-01, -3.0502e+00,
         -1.6716e+00],
        [-1.1591e+00, -1.9072e+00, -3.5037e+00, -3.9999e-02, -4.3541e+00,
         -1.5011e+00],
        [-2.8918e+00, -1.1612e-01, -1.6555e+00, -7.8714e-01, -8.0059e-01,
         -1.9959e-01]], device='cuda:0')
tensor([ -6.3271,  -6.2736,  -6.4651,  -4.5691, -14.9401, -11.3125, -10.3725,
         -8.2770, -11.0753, -14.0543,  -8.0371,  -8.1920, -10.5220,  -5.0571,
        -12.4098,  -7.8862,  -2.9850, -11.4625, -12.4652,  -6.4508],
       device='cuda:0')
tensor([-3.4586, -2.8198, -4.2877, -4.5218,  0.8920, -1.2568, -0.9084, -4.4469,
        -0.8546,  0.3522, -2.2197, -2.0282, -2.2639, -4.1265, -0.9078, -3.6456,
        -5.1632, -0.8742,  0.9383, -3.2270], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.6835, -3.8280, -1.7547, -1.7208, -1.6485, -1.6332],
        [-1.2647, -1.4907, -1.3606, -1.4417, -1.4392, -1.2590],
        [-4.1876, -1.2526, -3.4887, -1.2461, -1.8756, -1.0989],
        [-3.1356, -1.1982, -1.7567, -1.2298, -2.5801, -1.0499],
        [-1.8790, -1.1399, -2.2135, -1.6866, -1.4703, -1.0643],
        [-4.3578, -2.0365, -1.9463, -1.9752, -2.5052, -2.0134],
        [-2.5574, -1.9584, -3.4169, -1.7345, -1.5358, -1.2989],
        [-1.3776, -1.3321, -2.7366, -1.7636, -0.9646, -2.7834],
        [-2.8590, -1.6565, -1.9166, -1.3676, -1.2673, -1.4284],
        [-1.1071, -1.2277, -1.6558, -1.2695, -1.2673, -1.1716],
        [-2.0857, -0.8768, -1.8265, -1.1107, -1.6476, -1.5075],
        [-1.4258, -1.4561, -2.0436, -1.0247, -2.3036, -2.1308],
        [-1.8241, -3.3986, -2.0859, -1.0687, -1.8708, -2.1915],
        [-4.4591, -3.3195, -2.0470, -1.4103, -1.7872, -1.1861],
        [-1.5406, -1.8689, -1.4740, -4.2904, -1.5908, -1.6464],
        [-2.4188, -1.2329, -2.3276, -1.2695, -2.5130, -4.6439],
        [-2.6899, -1.2358, -2.2377, -1.1867, -1.7760, -2.5146],
        [-1.3701, -1.5629, -1.4088, -1.7379, -1.5615, -1.2756],
        [-4.4030, -3.1437, -1.6474, -1.6960, -1.8602, -1.8780],
        [-1.7171, -2.8858, -2.3413, -1.9518, -1.6617, -1.7065]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-14.2686,  -8.2559, -13.1496, -10.9503,  -9.4536, -14.8344, -12.5019,
        -10.9579, -10.4954,  -7.6990,  -9.0548, -10.3847, -12.4396, -14.2092,
        -12.4112, -14.4057, -11.6406,  -8.9168, -14.6283, -12.2643],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-3.3670e+00, -3.8695e+00, -1.2666e+00, -1.1122e+00, -1.5592e+00,
         -1.2332e+00],
        [-3.2631e-02, -1.0978e+00, -6.1133e-01, -5.7693e-01, -2.1262e-01,
         -4.8911e-01],
        [-5.1002e+00, -3.4093e-03, -4.0757e+00, -1.2628e-02, -2.2410e+00,
         -3.4156e-01],
        [-3.4760e+00, -1.8754e-01, -1.2697e+00, -3.9313e-01, -3.5454e+00,
         -1.0637e-03],
        [-1.9009e+00, -4.9590e-02, -2.5531e+00, -1.0483e+00, -1.1895e+00,
         -3.3555e-03],
        [-4.3096e+00, -1.6263e+00, -2.1823e+00, -2.2830e+00, -3.7667e+00,
         -2.3990e+00],
        [-1.5426e+00, -1.3482e+00, -2.9002e+00, -1.0915e+00, -5.2140e-01,
         -2.9586e-01],
        [-7.0456e-01, -1.0025e+00, -2.8478e+00, -1.3366e+00, -2.3849e-02,
         -2.6878e+00],
        [-3.0464e+00, -1.8620e+00, -1.8852e+00, -3.8012e-01, -8.3209e-01,
         -8.0059e-01],
        [-5.1669e-01, -7.0409e-02, -1.0106e+00, -4.4759e-01, -6.8065e-01,
         -5.2678e-01],
        [-2.8174e+00, -2.7551e-01, -1.4875e+00, -4.7480e-02, -1.2393e+00,
         -1.2053e+00],
        [-1.3424e+00, -3.1759e-01, -2.3307e+00, -1.4940e-01, -2.6020e+00,
         -1.7014e+00],
        [-7.6878e-01, -4.9021e+00, -2.7953e+00, -2.9433e-03, -1.9466e+00,
         -2.5295e+00],
        [-4.0411e+00, -2.9137e+00, -1.8651e+00, -8.3187e-01, -1.6774e+00,
         -4.6448e-01],
        [-5.3250e-01, -2.0545e+00, -9.7884e-01, -3.5973e+00, -1.0255e+00,
         -1.2100e+00],
        [-3.7573e+00, -3.5349e-02, -2.4630e+00, -4.0827e-02, -3.0050e+00,
         -2.3440e+00],
        [-2.9681e+00, -2.1423e-02, -2.2656e+00, -4.8184e-02, -2.0802e+00,
         -2.7300e+00],
        [-8.3808e-01, -6.3144e-01, -1.0376e+00, -1.4069e+00, -1.3802e+00,
         -4.3665e-01],
        [-4.6378e+00, -3.8816e+00, -9.3631e-01, -1.7418e+00, -1.9272e+00,
         -1.8705e+00],
        [-1.6285e-01, -3.0455e+00, -3.1946e+00, -8.2050e-01, -7.2132e-01,
         -1.6991e+00]], device='cuda:0')
tensor([-12.4078,  -3.0204, -11.7745,  -8.8728,  -6.7448, -16.5670,  -7.6997,
         -8.6032,  -8.8064,  -3.2527,  -7.0726,  -8.4435, -12.9452, -11.7936,
         -9.3986, -11.6455, -10.1135,  -5.7308, -14.9952,  -9.6438],
       device='cuda:0')
tensor([-1.8608, -5.2354, -1.3751, -2.0775, -2.7088,  1.7326, -4.8022, -2.3547,
        -1.6890, -4.4463, -1.9822, -1.9412,  0.5056, -2.4157, -3.0126, -2.7603,
        -1.5271, -3.1860,  0.3669, -2.6205], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.1675, -1.4269, -1.3321, -1.7136, -1.1012, -1.4041],
        [-2.0115, -1.8994, -1.7754, -2.6943, -1.8186, -1.1935],
        [-5.2712, -1.8394, -2.7788, -2.9451, -1.8477, -1.3165],
        [-2.1873, -0.9285, -1.9592, -1.3154, -3.2197, -2.5916],
        [-3.8811, -3.0249, -2.6004, -1.3857, -2.1692, -1.1326],
        [-2.3529, -2.7077, -2.6311, -2.4700, -2.3091, -1.9201],
        [-1.3759, -1.4736, -2.7183, -0.9966, -1.4651, -1.3988],
        [-2.3014, -1.3162, -2.5606, -1.3228, -1.5543, -0.8844],
        [-3.7581, -1.3060, -2.8260, -1.1703, -1.8915, -1.0977],
        [-5.1159, -1.1504, -3.2987, -1.1357, -1.9825, -1.5482],
        [-2.0309, -1.1705, -1.6965, -1.1730, -2.2102, -1.8399],
        [-3.4007, -2.0905, -2.1542, -1.1335, -1.9116, -1.2898],
        [-2.5860, -3.7831, -2.6133, -1.3971, -2.2507, -2.4234],
        [-1.3300, -1.2205, -1.6582, -1.2039, -1.4735, -1.1367],
        [-2.3210, -1.2198, -2.2683, -1.4965, -2.3851, -2.3654],
        [-2.5912, -1.3301, -2.0737, -1.2832, -1.8760, -2.6610],
        [-2.1261, -1.3170, -1.1077, -1.1980, -1.5787, -1.1388],
        [-2.8820, -1.9569, -1.6828, -1.1284, -1.6969, -1.5052],
        [-3.4363, -2.7306, -2.0328, -1.2162, -2.1086, -1.3719],
        [-1.1978, -1.1236, -1.1506, -0.9622, -1.3701, -1.2224]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -9.1454, -11.3927, -15.9988, -12.2017, -14.1940, -14.3908,  -9.4283,
         -9.9397, -12.0496, -14.2315, -10.1211, -11.9803, -15.0536,  -8.0228,
        -12.0560, -11.8153,  -8.4663, -10.8522, -12.8964,  -7.0266],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-1.2870e+00, -1.0389e+00, -7.7452e-01, -1.1211e+00, -1.3909e-01,
         -1.7912e+00],
        [-2.7944e+00, -2.3425e+00, -1.9058e+00, -1.8807e+00, -1.6611e+00,
         -2.0936e-01],
        [-5.4975e+00, -1.3061e+00, -3.2724e+00, -3.0203e+00, -1.6840e+00,
         -8.7095e-01],
        [-2.8926e+00, -3.8601e-03, -1.9334e+00, -4.3788e-01, -3.6691e+00,
         -3.0576e+00],
        [-3.8100e+00, -2.9841e+00, -3.8939e+00, -3.0412e-01, -2.9286e+00,
         -8.3175e-01],
        [-1.6717e+00, -3.1139e+00, -4.0042e+00, -1.3039e+00, -3.3867e+00,
         -2.1640e+00],
        [-2.1018e-03, -1.0927e+00, -3.3398e+00, -2.1886e-03, -6.4033e-01,
         -6.0267e-01],
        [-3.6310e+00, -2.9224e-03, -2.7862e+00, -1.1811e-02, -8.4632e-01,
         -6.7562e-03],
        [-3.6999e+00, -4.8238e-02, -3.3395e+00, -1.0620e-01, -2.3812e+00,
         -1.2657e-01],
        [-5.0093e+00, -3.1904e-01, -3.3023e+00, -1.5903e-01, -2.3892e+00,
         -1.2831e+00],
        [-2.5237e+00, -2.8631e-01, -1.7921e+00, -2.0376e-02, -2.4252e+00,
         -1.1527e+00],
        [-3.4777e+00, -1.5435e+00, -3.1328e+00, -6.5880e-02, -2.0090e+00,
         -4.3450e-02],
        [-2.2529e+00, -3.9752e+00, -3.2627e+00, -5.0933e-01, -3.1809e+00,
         -2.8838e+00],
        [-2.9228e-02, -4.7512e-02, -1.3991e+00, -1.9442e-01, -3.5237e-01,
         -3.3774e-02],
        [-3.3148e+00, -4.3208e-02, -2.4200e+00, -7.1768e-01, -2.8808e+00,
         -1.2801e+00],
        [-3.0783e+00, -2.7214e-01, -2.0394e+00, -1.5915e-01, -2.4304e+00,
         -2.5134e+00],
        [-2.0990e+00, -6.0111e-01, -3.6563e-02, -3.6437e-01, -1.4007e+00,
         -1.4271e-01],
        [-2.7419e+00, -1.4711e+00, -1.4533e+00, -1.5683e-01, -1.3628e+00,
         -1.0355e+00],
        [-3.5871e+00, -3.1487e+00, -2.0390e+00, -1.3418e-01, -2.9041e+00,
         -7.6903e-02],
        [-3.1739e-01, -1.7128e-03, -1.7170e-01, -1.3028e-02, -5.3345e-01,
         -2.1855e-01]], device='cuda:0')
tensor([ -6.1518, -10.7939, -15.6513, -11.9945, -14.7525, -15.6444,  -5.6798,
         -7.2850,  -9.7016, -12.4619,  -8.2004, -10.2724, -16.0649,  -2.0564,
        -10.6567, -10.4927,  -4.6445,  -8.2215, -11.8899,  -1.2558],
       device='cuda:0')
tensor([-2.9936, -0.5988, -0.3475, -0.2072,  0.5585,  1.2536, -3.7485, -2.6547,
        -2.3481, -1.7696, -1.9206, -1.7079,  1.0113, -5.9663, -1.3993, -1.3225,
        -3.8218, -2.6308, -1.0065, -5.7708], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-1.3544, -1.1836, -1.3921, -1.3083, -1.2809, -1.7759],
        [-1.7259, -1.0306, -2.3020, -1.6947, -1.9114, -1.5039],
        [-3.9602, -1.4630, -2.0843, -1.2609, -2.0760, -3.1412],
        [-3.2672, -1.3129, -2.4621, -2.3140, -1.4224, -1.3535],
        [-2.1762, -2.2108, -1.8866, -2.5295, -1.9235, -2.6675],
        [-2.2080, -1.8981, -2.6680, -0.9617, -1.7230, -1.4504],
        [-1.3711, -1.1417, -1.4232, -1.1736, -1.1400, -1.0903],
        [-3.3959, -1.2998, -2.6063, -1.5785, -1.1788, -0.9507],
        [-3.3143, -1.4865, -1.3700, -1.4216, -1.4175, -1.3222],
        [-1.6968, -1.3395, -1.0539, -1.1171, -1.2286, -2.0514],
        [-2.8075, -1.8262, -1.5591, -1.2692, -2.0396, -0.9947],
        [-1.1006, -1.1915, -2.1552, -2.4716, -1.2824, -1.1611],
        [-2.9939, -1.6183, -2.2096, -2.6402, -2.2820, -1.6614],
        [-1.9991, -3.4996, -1.5467, -1.6504, -3.5701, -1.8216],
        [-4.2796, -3.4814, -2.7281, -1.3772, -1.8390, -1.7053],
        [-3.4607, -2.9698, -2.2072, -2.2607, -2.2136, -2.8977],
        [-1.2241, -1.1472, -1.2557, -1.0703, -1.6349, -1.0671],
        [-1.4047, -1.1788, -4.6115, -1.3499, -1.9486, -1.7496],
        [-2.3807, -1.2728, -1.9362, -1.2286, -1.5814, -1.1656],
        [-2.5694, -1.2895, -1.5379, -1.2974, -1.9255, -1.8288]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([ -8.2953, -10.1687, -13.9856, -12.1321, -13.3941, -10.9092,  -7.3400,
        -11.0100, -10.3320,  -8.4872, -10.4963,  -9.3624, -13.4055, -14.0875,
        -15.4105, -16.0097,  -7.3993, -12.2433,  -9.5653, -10.4484],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-0.3016, -0.0120, -0.2074, -1.1507, -0.1888, -1.7476],
        [-1.8973, -0.1892, -2.3466, -1.2643, -2.0087, -1.1359],
        [-4.8479, -0.5643, -2.0171, -0.0892, -2.8403, -2.1231],
        [-4.1290, -0.1339, -3.6529, -2.0330, -1.1991, -0.8667],
        [-2.0892, -2.1453, -2.1458, -3.5039, -1.9884, -3.4303],
        [-1.0005, -1.2513, -2.3481, -0.0094, -1.1897, -0.6294],
        [-0.8750, -0.0207, -0.2142, -0.2888, -0.1064, -0.3474],
        [-4.0046, -0.0195, -3.3091, -0.8964, -0.0282, -0.0644],
        [-2.8071, -0.3562, -0.3413, -0.6532, -0.2431, -0.6485],
        [-0.8145, -0.7157, -0.0269, -0.0497, -0.1004, -1.9288],
        [-2.4649, -1.7261, -1.0799, -0.2351, -2.0481, -0.0355],
        [-0.0347, -0.3305, -2.5949, -2.3505, -0.1742, -0.1084],
        [-4.1772, -0.9711, -2.5418, -2.7537, -2.1070, -0.9732],
        [-2.3924, -2.6351, -0.9718, -0.9786, -3.9515, -1.9414],
        [-4.4299, -2.7249, -3.3937, -0.4489, -1.7432, -1.3536],
        [-4.3481, -3.7118, -3.1421, -1.0747, -2.6633, -3.7808],
        [-0.4489, -0.3645, -0.1443, -0.3956, -1.5977, -0.0597],
        [-1.3857, -0.2345, -5.2131, -0.3975, -1.9470, -0.8959],
        [-2.7736, -0.0782, -1.6803, -0.0128, -1.4113, -0.3761],
        [-2.5033, -0.1862, -0.7371, -0.6166, -2.4737, -1.8720]],
       device='cuda:0')
tensor([ -3.6081,  -8.8420, -12.4819, -12.0145, -15.3029,  -6.4284,  -1.8525,
         -8.3222,  -5.0493,  -3.6360,  -7.5895,  -5.5931, -13.5240, -12.8709,
        -14.0941, -18.7208,  -3.0106, -10.0737,  -6.3322,  -8.3889],
       device='cuda:0')
tensor([-4.6872, -1.3267, -1.5037, -0.1176,  1.9088, -4.4808, -5.4875, -2.6878,
        -5.2827, -4.8512, -2.9068, -3.7693,  0.1185, -1.2166, -1.3164,  2.7111,
        -4.3886, -2.1695, -3.2330, -2.0595], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-2.4393, -1.2119, -3.1246, -1.2153, -3.1047, -1.5949],
        [-2.9424, -1.1240, -1.5898, -1.1990, -2.1949, -1.6440],
        [-3.0524, -2.3164, -1.7702, -1.1590, -1.7660, -1.2834],
        [-3.1885, -2.6759, -1.7873, -1.6316, -1.8327, -1.4072],
        [-1.3160, -1.4820, -1.5477, -1.0512, -1.3076, -1.0786],
        [-2.6849, -1.2594, -2.0524, -1.4658, -1.9117, -1.0150],
        [-2.7978, -1.3446, -2.8443, -1.1969, -1.9091, -1.1465],
        [-1.5820, -1.2018, -1.8557, -2.1061, -1.5447, -1.3851],
        [-4.3686, -2.4594, -1.9148, -1.1926, -1.8583, -1.5288],
        [-3.3164, -2.6767, -1.8731, -1.3117, -1.7979, -1.3447],
        [-1.3168, -1.3760, -1.5585, -1.0724, -1.3088, -1.2195],
        [-1.9938, -1.5712, -2.9137, -1.3845, -1.6301, -1.1886],
        [-4.0371, -1.0526, -5.0543, -1.0956, -1.5908, -1.1806],
        [-4.4756, -1.1373, -1.8063, -1.3316, -2.3480, -1.7522],
        [-1.3501, -1.2207, -1.3747, -1.1864, -2.0329, -1.2356],
        [-3.5526, -2.3837, -2.0661, -1.1186, -2.1149, -1.3340],
        [-2.8044, -3.2310, -2.4805, -1.5959, -2.1768, -2.3025],
        [-1.4776, -1.3899, -1.3942, -1.0371, -1.4691, -1.1558],
        [-2.9220, -1.5252, -2.2812, -1.4608, -2.0293, -1.5707],
        [-2.1168, -1.2256, -2.9307, -1.6024, -1.6407, -1.2986]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.6906, -10.6943, -11.3474, -12.5232,  -7.7831, -10.3891, -11.2392,
         -9.6754, -13.3224, -12.3205,  -7.8520, -10.6819, -14.0111, -12.8510,
         -8.4004, -12.5700, -14.5912,  -7.9236, -11.7893, -10.8148],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-2.7211e+00, -2.3515e-02, -3.4638e+00, -2.0087e-01, -4.1124e+00,
         -1.1125e+00],
        [-3.4803e+00, -5.8318e-03, -1.3785e+00, -3.6071e-01, -2.3673e+00,
         -1.2286e+00],
        [-2.9261e+00, -2.4380e+00, -9.3710e-01, -3.8012e-01, -1.6350e+00,
         -1.0895e-01],
        [-3.2003e+00, -2.6114e+00, -1.4252e+00, -7.0665e-01, -1.9257e+00,
         -8.4997e-01],
        [-6.9107e-01, -1.2042e+00, -1.0302e+00, -5.8481e-02, -5.2876e-02,
         -8.8394e-02],
        [-3.8659e+00, -3.5738e-02, -2.0958e+00, -5.9892e-01, -2.0871e+00,
         -1.1352e-01],
        [-3.2594e+00, -1.4081e+00, -3.9684e+00, -3.3859e-01, -2.5018e+00,
         -1.3019e-01],
        [-1.4618e+00, -5.3567e-02, -1.9821e+00, -1.8195e+00, -1.4060e+00,
         -8.3709e-01],
        [-4.3084e+00, -2.9118e+00, -1.9304e+00, -6.4499e-01, -1.8230e+00,
         -8.2753e-01],
        [-3.5561e+00, -2.8620e+00, -1.5750e+00, -1.7666e-01, -1.7702e+00,
         -6.0352e-01],
        [-7.9081e-01, -9.4636e-01, -1.0811e+00, -3.3753e-02, -2.7994e-01,
         -5.1162e-01],
        [-2.7702e+00, -4.0259e-01, -4.3078e+00, -2.6407e-01, -1.2755e+00,
         -2.4320e-01],
        [-4.3730e+00, -5.2652e-02, -6.5086e+00, -9.6032e-02, -1.7359e+00,
         -2.8968e-05],
        [-4.6253e+00, -5.7764e-01, -1.1923e+00, -4.5105e-01, -3.1170e+00,
         -1.5931e+00],
        [-1.2659e+00, -1.5373e-01, -9.8626e-01, -1.9069e-01, -2.1503e+00,
         -3.2878e-01],
        [-3.4681e+00, -2.6625e+00, -2.6484e+00, -2.7427e-01, -2.8227e+00,
         -2.7439e-02],
        [-2.7164e+00, -3.3149e+00, -2.9693e+00, -7.5870e-01, -2.9501e+00,
         -2.6032e+00],
        [-1.4851e+00, -8.3256e-01, -5.9288e-01, -1.0049e-01, -6.2839e-01,
         -2.5727e-02],
        [-4.4811e+00, -3.4407e-01, -2.7879e+00, -5.3742e-01, -2.2525e+00,
         -6.4531e-01],
        [-2.4225e+00, -5.3197e-02, -3.8972e+00, -1.1562e+00, -1.8722e+00,
         -8.1903e-01]], device='cuda:0')
tensor([-11.6343,  -8.8213,  -8.4252, -10.7192,  -3.1252,  -8.7969, -11.6064,
         -7.5600, -12.4462, -10.5435,  -3.6436,  -9.2635, -12.7662, -11.5564,
         -5.0756, -11.9034, -15.3126,  -3.6651, -11.0482, -10.2202],
       device='cuda:0')
tensor([-1.0564, -1.8730, -2.9222, -1.8040, -4.6578, -1.5922,  0.3671, -2.1154,
        -0.8763, -1.7770, -4.2084, -1.4184, -1.2449, -1.2946, -3.3247, -0.6665,
         0.7214, -4.2585, -0.7410, -0.5946], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-4.5009, -1.9256, -2.3493, -0.8962, -1.8281, -1.2065],
        [-1.3320, -1.2557, -1.3181, -1.1373, -1.6925, -1.2372],
        [-3.1638, -1.1578, -2.6436, -1.1677, -2.5340, -0.9982],
        [-4.5028, -2.1017, -3.0878, -1.4884, -1.3450, -0.9839],
        [-3.2197, -1.0481, -5.4397, -1.1841, -1.6502, -1.3135],
        [-1.5662, -1.1672, -1.2515, -1.2555, -2.7076, -1.4116],
        [-2.8545, -1.9381, -2.2326, -1.1145, -2.0144, -1.3622],
        [-4.5794, -4.3524, -3.0290, -1.5813, -3.0839, -1.4816],
        [-1.5821, -1.3124, -1.5928, -1.5790, -1.4557, -1.2693],
        [-2.2594, -1.2756, -2.9636, -1.2569, -1.9626, -1.0994],
        [-2.1629, -1.2300, -4.0190, -1.1772, -2.0069, -1.6540],
        [-4.2466, -1.2054, -1.3637, -1.4754, -1.4378, -1.0698],
        [-5.3506, -1.7887, -1.9032, -1.2173, -2.5426, -1.4209],
        [-3.2517, -3.3522, -2.7932, -1.8259, -1.9348, -1.3324],
        [-2.0297, -1.0915, -1.0996, -0.9400, -1.4433, -1.1154],
        [-2.5430, -1.4767, -2.4643, -1.2452, -3.3466, -2.1565],
        [-2.1134, -1.2702, -2.2848, -1.1673, -1.9227, -1.2487],
        [-3.1412, -1.3422, -1.2795, -1.5176, -1.9348, -1.7331],
        [-5.1826, -1.7121, -1.5693, -1.8259, -2.8296, -3.0614],
        [-3.4754, -1.9541, -2.4929, -1.0611, -2.7812, -1.8646]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-12.7066,  -7.9728, -11.6650, -13.5095, -13.8555,  -9.3595, -11.5162,
        -18.1076,  -8.7914, -10.8177, -12.2502, -10.7987, -14.2234, -14.4901,
         -7.7196, -13.2323, -10.0071, -10.9483, -16.1810, -13.6293],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.3851e+00, -1.3705e+00, -2.3208e+00, -6.3193e-04, -1.6946e+00,
         -1.1387e-01],
        [-8.1499e-01, -2.9566e-02, -2.4116e-01, -3.9596e-01, -1.6915e+00,
         -6.7220e-01],
        [-4.2176e+00, -1.4657e-01, -3.5690e+00, -1.0103e-02, -2.8195e+00,
         -7.2152e-02],
        [-4.6294e+00, -1.7235e+00, -4.0540e+00, -9.0365e-01, -6.0793e-01,
         -6.4887e-02],
        [-4.1391e+00, -7.7528e-01, -5.9670e+00, -2.0189e-01, -1.3843e+00,
         -1.8024e-03],
        [-1.6458e+00, -1.4144e-02, -6.4947e-01, -3.9055e-03, -3.0947e+00,
         -5.8159e-01],
        [-2.4365e+00, -1.0799e+00, -3.5464e+00, -2.0520e-01, -2.4169e+00,
         -1.8620e-01],
        [-4.7391e+00, -4.2881e+00, -3.5774e+00, -8.8840e-01, -4.3235e+00,
         -9.6894e-01],
        [-1.8194e+00, -5.1034e-01, -1.2050e+00, -7.3838e-01, -8.9912e-02,
         -1.5597e-01],
        [-3.1251e+00, -3.8800e-03, -4.0090e+00, -1.9112e-02, -1.9964e+00,
         -2.0793e-01],
        [-2.4207e+00, -2.5045e-01, -5.6987e+00, -9.3179e-02, -2.7337e+00,
         -1.5212e+00],
        [-4.8408e+00, -1.6624e-01, -8.2053e-01, -7.4010e-01, -1.0788e+00,
         -1.3712e-01],
        [-5.9820e+00, -7.8836e-01, -2.0119e+00, -1.2857e-01, -3.8105e+00,
         -6.3622e-01],
        [-3.2111e+00, -3.9163e+00, -3.9047e+00, -7.9512e-01, -2.2607e+00,
         -3.3638e-01],
        [-3.0102e+00, -2.5051e-02, -7.6010e-03, -1.8995e-02, -1.0105e+00,
         -2.3541e-02],
        [-3.7913e+00, -3.1005e-01, -2.9670e+00, -1.7382e-01, -4.1425e+00,
         -1.4007e+00],
        [-2.4035e+00, -2.7138e-01, -2.6786e+00, -1.1440e-02, -2.4470e+00,
         -7.2029e-01],
        [-3.4024e+00, -1.0970e-01, -8.1398e-02, -9.1092e-01, -2.5451e+00,
         -1.2148e+00],
        [-4.6803e+00, -1.0348e+00, -1.3472e+00, -1.6999e+00, -3.7982e+00,
         -3.2707e+00],
        [-3.9394e+00, -1.4351e+00, -3.4242e+00, -1.6212e-05, -4.9920e+00,
         -1.9921e+00]], device='cuda:0')
tensor([ -9.8855,  -3.8454, -10.8350, -11.9834, -12.4694,  -5.9896,  -9.8711,
        -18.7855,  -4.5190,  -9.3615, -12.7179,  -7.7836, -13.3575, -14.4243,
         -4.0959, -12.7855,  -8.5322,  -8.2644, -15.8311, -15.7828],
       device='cuda:0')
tensor([-2.8211, -4.1274, -0.8300, -1.5261, -1.3861, -3.3699, -1.6451,  0.6779,
        -4.2724, -1.4561,  0.4677, -3.0151, -0.8658, -0.0658, -3.6236, -0.4468,
        -1.4749, -2.6840, -0.3499,  2.1535], device='cuda:0',
       grad_fn=<SubBackward0>)
tensor([[-3.4902, -1.3732, -3.0515, -1.3129, -1.9097, -2.4812],
        [-4.4491, -1.1725, -1.5062, -1.1314, -2.0425, -2.0251],
        [-1.9685, -1.3323, -1.2746, -1.8607, -2.1093, -1.4615],
        [-3.4671, -2.5038, -1.7040, -1.8287, -2.1992, -1.6854],
        [-2.8159, -2.2112, -1.7595, -1.2447, -2.2592, -1.2705],
        [-1.5439, -1.2389, -1.2200, -1.0169, -1.4348, -1.1423],
        [-3.2187, -1.3849, -2.3675, -1.2215, -1.8077, -2.2471],
        [-4.9002, -1.2334, -3.3348, -1.1203, -1.3574, -1.0593],
        [-3.0642, -1.2626, -2.0020, -1.2267, -1.8704, -1.5258],
        [-1.9137, -1.1469, -1.6720, -1.1269, -2.3355, -1.5069],
        [-4.5799, -2.5822, -1.7940, -1.1491, -2.0186, -1.2241],
        [-2.7683, -2.4801, -2.0198, -1.1701, -1.7554, -1.7220],
        [-1.3886, -1.2661, -1.8639, -1.1378, -1.4847, -1.0802],
        [-2.0792, -1.4065, -2.6400, -1.2913, -1.3508, -1.2067],
        [-2.3967, -1.1906, -2.4042, -1.1045, -1.5241, -1.1709],
        [-3.4270, -1.1242, -1.1261, -1.7356, -1.6932, -1.1544],
        [-4.0555, -2.1533, -1.7618, -1.1156, -2.0698, -1.2577],
        [-3.3179, -3.5275, -1.9171, -1.1469, -1.9632, -1.3228],
        [-1.1967, -1.4649, -1.6362, -1.4219, -1.3478, -1.1781],
        [-2.0697, -1.7579, -2.8236, -1.3001, -2.2223, -0.8752]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-13.6188, -12.3269, -10.0069, -13.3882, -11.5611,  -7.5969, -12.2474,
        -13.0053, -10.9517,  -9.7020, -13.3480, -11.9157,  -8.2213,  -9.9745,
         -9.7911, -10.2605, -12.4138, -13.1953,  -8.2456, -11.0487],
       device='cuda:0', grad_fn=<SumBackward1>)
tensor([[-4.8867e+00, -2.8378e-02, -4.3150e+00, -2.0017e-01, -1.8282e+00,
         -1.5108e+00],
        [-4.6555e+00, -2.1861e-01, -5.6283e-01, -2.6775e-02, -2.8068e+00,
         -2.0910e+00],
        [-1.8912e+00, -7.5292e-02, -7.6954e-01, -1.6140e+00, -2.9514e+00,
         -9.9001e-01],
        [-3.1745e+00, -2.8007e+00, -1.5316e+00, -1.8754e+00, -2.7879e+00,
         -1.3845e+00],
        [-2.4953e+00, -2.0001e+00, -1.2502e+00, -1.1176e-01, -3.1862e+00,
         -1.0441e-01],
        [-1.9097e+00, -5.4855e-01, -2.1570e-01, -1.2804e-01, -7.1827e-01,
         -4.1121e-02],
        [-4.8499e+00, -7.0644e-02, -2.9408e+00, -1.5839e-02, -1.6284e+00,
         -1.1618e+00],
        [-4.8288e+00, -1.3785e-01, -4.5125e+00, -2.1584e-02, -8.7544e-01,
         -8.2654e-02],
        [-3.1180e+00, -4.4829e-03, -1.4700e+00, -2.6433e-01, -2.3601e+00,
         -1.1564e+00],
        [-2.1793e+00, -3.9985e-01, -1.4778e+00, -2.9188e-01, -2.6372e+00,
         -1.0775e+00],
        [-5.1428e+00, -2.9786e+00, -1.0148e+00, -6.3646e-03, -2.4729e+00,
         -5.1228e-02],
        [-2.7026e+00, -2.3133e+00, -1.9620e+00, -1.2492e-01, -1.8637e+00,
         -1.6560e+00],
        [-1.1178e+00, -4.3349e-01, -1.8360e+00, -2.1555e-01, -6.3175e-01,
         -8.7881e-04],
        [-2.9928e+00, -1.9612e-02, -3.7081e+00, -3.7844e-02, -4.6091e-01,
         -2.0524e-01],
        [-2.6572e+00, -4.3729e-01, -2.8808e+00, -4.5028e-02, -1.5516e+00,
         -4.6259e-01],
        [-3.7009e+00, -9.8119e-03, -1.0735e-01, -1.3447e+00, -1.8173e+00,
         -2.3161e-01],
        [-4.1262e+00, -1.9206e+00, -1.6809e+00, -2.5348e-01, -2.5370e+00,
         -1.3407e-01],
        [-3.2193e+00, -4.3491e+00, -1.7171e+00, -6.2979e-02, -2.3145e+00,
         -1.2411e-02],
        [-5.0735e-01, -1.2589e+00, -1.2138e+00, -7.0158e-01, -1.1671e-04,
         -1.8598e-03],
        [-2.6749e+00, -9.0453e-01, -3.9526e+00, -1.2135e-01, -2.6488e+00,
         -1.5639e-02]], device='cuda:0')
