tensor([[[ 0.1062, -0.0706,  0.0343,  ...,  0.0896,  0.1241,  0.0811],
         [ 0.1552, -0.0632,  0.0497,  ...,  0.0547,  0.0725,  0.0960],
         [ 0.0970, -0.0783,  0.0647,  ...,  0.0812,  0.0578,  0.1070],
         ...,
         [ 0.1979, -0.0234,  0.0873,  ...,  0.1047,  0.1507,  0.0587],
         [ 0.2031, -0.0630,  0.1304,  ...,  0.0985,  0.1816,  0.0514],
         [ 0.1303, -0.1082,  0.0683,  ...,  0.0532,  0.1669,  0.0524]],

        [[ 0.1829, -0.0706,  0.0533,  ...,  0.0783,  0.1255,  0.1043],
         [ 0.1891, -0.0596,  0.0698,  ...,  0.1247,  0.0781,  0.1504],
         [ 0.1921, -0.0589,  0.0724,  ...,  0.1275,  0.0724,  0.2050],
         ...,
         [ 0.1394, -0.0822,  0.0662,  ...,  0.1140,  0.1395,  0.1234],
         [ 0.1750, -0.0778,  0.0945,  ...,  0.0925,  0.1522,  0.0401],
         [ 0.1537, -0.1094,  0.0306,  ...,  0.0350,  0.1453,  0.0510]],

        [[ 0.1791, -0.0545,  0.0312,  ...,  0.1040,  0.1212,  0.0353],
         [ 0.2351, -0.0766,  0.0298,  ...,  0.0786,  0.0569,  0.1059],
         [ 0.2285, -0.0711,  0.0330,  ...,  0.0844,  0.0642,  0.1682],
         ...,
         [ 0.1871, -0.0736, -0.0312,  ...,  0.1648,  0.1705, -0.0026],
         [ 0.1876, -0.0933, -0.0158,  ...,  0.1435,  0.1671, -0.0308],
         [ 0.1806, -0.1350, -0.0156,  ...,  0.1419,  0.1417, -0.0586]],

        [[ 0.2451, -0.0512,  0.0207,  ...,  0.1242,  0.1442,  0.0446],
         [ 0.2269, -0.0461, -0.0124,  ...,  0.1284,  0.0918,  0.1082],
         [ 0.2298, -0.0476, -0.0281,  ...,  0.1397,  0.0669,  0.1410],
         ...,
         [ 0.0859, -0.0831,  0.0446,  ...,  0.1194,  0.1112,  0.0999],
         [ 0.1291, -0.0871,  0.0768,  ...,  0.1306,  0.1285,  0.0413],
         [ 0.1610, -0.1204,  0.0627,  ...,  0.1046,  0.1033,  0.0654]]],
       device='cuda:0')
tensor([[[-0.0857, -0.1034, -0.1035,  ..., -0.1175, -0.0724, -0.1227],
         [-0.0146, -0.0138, -0.0380,  ..., -0.0315, -0.0465,  0.0175],
         [ 0.0160,  0.0007,  0.0163,  ...,  0.0147,  0.0308,  0.0090],
         ...,
         [-0.0352, -0.0108, -0.0365,  ..., -0.1049, -0.0922, -0.0531],
         [ 0.2708,  0.2582,  0.2379,  ...,  0.2625,  0.2461,  0.2378],
         [ 0.0994,  0.0935,  0.0687,  ...,  0.1915,  0.1793,  0.0974]],

        [[-0.1221, -0.1453, -0.1377,  ..., -0.0979, -0.1344, -0.1147],
         [ 0.0085, -0.0006, -0.0129,  ..., -0.0306, -0.0060,  0.0057],
         [-0.0065,  0.0018,  0.0445,  ...,  0.0048,  0.0242,  0.0162],
         ...,
         [-0.0500, -0.0181, -0.0331,  ..., -0.0701, -0.0888, -0.0527],
         [ 0.3233,  0.3220,  0.2983,  ...,  0.2219,  0.2088,  0.2299],
         [ 0.1187,  0.0565,  0.0609,  ...,  0.1153,  0.1416,  0.0839]],

        [[-0.1152, -0.1286, -0.0901,  ..., -0.1428, -0.1430, -0.1392],
         [-0.0378, -0.0819, -0.0742,  ..., -0.0811, -0.0488, -0.0468],
         [ 0.0232, -0.0236,  0.0366,  ..., -0.0430, -0.0368,  0.0521],
         ...,
         [-0.0138, -0.0094, -0.0153,  ...,  0.0405,  0.0340, -0.0206],
         [ 0.3609,  0.3894,  0.3172,  ...,  0.3283,  0.2934,  0.2837],
         [ 0.1029,  0.1177,  0.1286,  ...,  0.0946,  0.0702,  0.0427]],

        [[-0.1581, -0.1968, -0.1531,  ..., -0.1368, -0.1294, -0.1570],
         [-0.0574, -0.0339, -0.0765,  ..., -0.1366, -0.1136, -0.0715],
         [ 0.0139,  0.0585,  0.0980,  ...,  0.0896,  0.0811,  0.1045],
         ...,
         [-0.0468, -0.0367, -0.0364,  ..., -0.0763, -0.0872, -0.0648],
         [ 0.4041,  0.4157,  0.4007,  ...,  0.3263,  0.2994,  0.2842],
         [ 0.1058,  0.1614,  0.1824,  ...,  0.1432,  0.1486,  0.1457]]],
       device='cuda:0')