tensor([[ 2.5571,  0.0120,  2.2805, -0.3635, -0.7773, -0.0230],
        [ 2.3571,  0.5425,  2.2679,  0.4903, -1.5274, -0.9987],
        [ 2.0996,  0.2282, -1.1405,  0.2727, -1.6026, -1.8638],
        [-1.9332, -2.4681, -1.7166,  0.4069,  1.1099, -0.3603],
        [-2.3109, -2.3646, -1.4676, -0.9093,  2.5349,  1.7925],
        [ 0.7870,  1.6745, -1.0899, -0.8481,  0.1659,  0.5173],
        [ 2.4787, -0.1648,  2.5052,  0.7348, -1.6568,  0.2299],
        [ 2.5608,  0.2040,  1.8584,  0.6953, -1.1109, -1.2167],
        [ 1.4604, -0.6655, -1.3429,  2.6506, -1.8556, -0.1519],
        [-1.7336, -2.2789, -1.6586,  0.8597,  2.4486,  0.0289],
        [-3.1228, -1.7661, -2.0062, -0.0034,  3.0074,  0.7095],
        [ 0.9704,  1.3416, -1.7346, -0.6516, -0.4663, -0.4639],
        [ 2.7065, -0.2414,  3.2271,  0.4813, -0.3661, -0.1490],
        [ 1.4840,  0.0656,  2.2354,  1.4245, -1.7730, -1.3849],
        [ 1.5939, -1.1953, -1.3727,  2.0568, -1.2987, -2.0875],
        [-2.2498, -2.7023, -2.3429,  1.0427,  2.2280, -0.6313],
        [-1.1754, -2.2087, -1.6797, -0.4098,  1.2213,  0.7559],
        [ 0.5216,  1.4740, -2.0013, -0.4187,  0.1103,  0.3975],
        [ 2.3649,  0.3823,  2.4331,  0.7201, -0.4859, -0.9881],
        [ 1.7866,  0.6292,  2.8232,  0.1120, -1.9530, -1.2932]],
       device='cuda:0')
tensor([ 2.5571,  0.0120,  2.2805, -0.3635, -0.7773, -0.0230], device='cuda:0')
