tensor([[ 2.2449, -0.5146,  2.0494,  0.1373, -1.1046, -0.1967],
        [ 1.7764, -0.8497,  2.6515, -0.0907, -0.9679, -0.6338],
        [ 3.8405,  0.1615,  1.4825,  1.0604, -0.5508, -2.5320],
        [ 1.5042, -0.7690, -0.7518,  0.6553, -1.8622, -2.0593],
        [-2.7742, -1.0909, -1.8306,  0.6618,  1.9722, -0.9015],
        [-2.7638, -2.7341, -1.8809, -0.7472,  2.9776,  1.8173],
        [ 0.3820,  0.5641, -1.4608, -0.4308,  0.1394, -0.5065],
        [ 1.9883,  0.0060,  1.5536,  0.9876, -1.2151,  0.2524],
        [ 3.0710,  0.1878,  1.5546,  0.3380, -2.2291, -0.5479],
        [ 1.7028, -0.1354, -1.6143,  2.0217, -1.5215,  0.1679],
        [-2.5631, -1.6480, -1.8752,  1.2438,  2.4602,  1.4164],
        [-1.7629, -2.1343, -1.6993, -0.5545,  1.5637,  0.8344],
        [ 1.1423,  1.0163, -1.4106,  0.0632,  0.4034, -0.0197],
        [ 1.4495, -0.6583,  2.0824,  0.5464, -1.4991, -0.8341],
        [ 2.8550,  0.4890,  2.8008,  0.1302, -0.9762, -1.7078],
        [ 2.0025, -0.0540, -0.7922,  0.9757, -2.2209, -0.6861],
        [-2.4428, -1.9497, -1.3910,  1.4853,  2.3193,  0.9679],
        [-2.2587, -2.4982, -1.3722, -0.8739,  1.5580,  1.0803],
        [ 1.9746,  0.9871, -1.7028, -0.4080,  0.6690,  0.0622],
        [ 2.2814, -0.4198,  1.9244,  0.0808, -0.0460,  0.5590]],
       device='cuda:0')
tensor([ 2.2449, -0.5146,  2.0494,  0.1373, -1.1046, -0.1967], device='cuda:0')
tensor([[-2.7181, -1.1660, -2.5563, -1.1338, -1.3837, -1.0484],
        [-2.8504, -1.0687, -2.5671, -1.1271, -1.3069, -1.0476],
        [-3.0501, -1.2215, -2.5281, -1.1341, -1.2856, -0.9118],
        [-2.9315, -1.1559, -2.5737, -1.0583, -1.4257, -1.0564],
        [-2.7402, -1.1889, -2.8696, -1.2487, -1.5176, -1.4605],
        [-3.6518, -1.4134, -2.9062, -1.3601, -1.6595, -1.6240],
        [-3.4142, -1.1681, -2.5557, -1.2559, -1.5627, -1.2727],
        [-2.5886, -1.1399, -2.5777, -1.1281, -1.4107, -1.0929],
        [-2.6164, -1.0574, -2.7247, -1.2739, -1.2960, -1.0638],
        [-3.2552, -1.2275, -2.5412, -1.0906, -1.3541, -0.9644],
        [-2.6324, -1.1445, -2.6954, -1.2975, -1.4559, -1.3840],
        [-3.7839, -1.2877, -2.7901, -1.4526, -1.7346, -1.6838],
        [-3.8857, -1.2243, -2.6541, -1.2061, -1.5786, -1.3635],
        [-2.5377, -1.1387, -2.6491, -1.2066, -1.3801, -1.0470],
        [-2.9105, -1.0628, -2.3574, -1.1747, -1.3611, -1.1193],
        [-3.0093, -1.0767, -2.4397, -1.1992, -1.2596, -0.9011],
        [-2.6849, -1.0833, -2.7161, -1.2901, -1.4302, -1.3890],
        [-3.6103, -1.2137, -2.9757, -1.4853, -1.6800, -1.7403],
        [-3.5926, -1.1844, -2.4272, -1.2390, -1.5541, -1.3112],
        [-2.6714, -1.1424, -2.5493, -1.1391, -1.3692, -1.0383]],
       device='cuda:0', grad_fn=<SubBackward0>)
tensor([-10.0062,  -9.9677, -10.1312, -10.2014, -11.0255, -12.6150, -11.2293,
         -9.9379, -10.0323, -10.4329, -10.6098, -12.7327, -11.9123,  -9.9591,
         -9.9859,  -9.8857, -10.5935, -12.7053, -11.3084,  -9.9097],
       device='cuda:0', grad_fn=<SumBackward1>)
