pytorch is warning me of named tensors…but I don’t see any…what is going on?
x = torch.max_pool2d(input, kernel_size, stride, padding, dilation, ceil_mode)
<ipython-input-3-b78ccc85411a>:1: UserWarning: Named tensors and all their associated APIs are an experimental feature and subject to change. Please do not use them for anything important until they are released as stable. (Triggered internally at /tmp/pip-req-build-accccgfm/c10/core/TensorImpl.h:1156.)
x = torch.max_pool2d(input, kernel_size, stride, padding, dilation, ceil_mode)
x.names
Out[4]: (None, None, None, None)
input.names
Out[5]: (None, None, None, None)
the tensor is:
input
Out[6]:
tensor([[[[0.0000e+00, 5.3843e-01, 3.2706e-01, ..., 0.0000e+00,
0.0000e+00, 4.7521e-01],
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
1.1444e+00, 5.4548e-01],
[3.5387e-01, 3.4596e-01, 7.5767e-01, ..., 5.9062e-01,
0.0000e+00, 0.0000e+00],
...,
[8.5284e-01, 6.1768e-01, 2.6013e-01, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
[3.2465e-01, 0.0000e+00, 1.3492e+00, ..., 1.9810e-01,
0.0000e+00, 0.0000e+00],
[0.0000e+00, 1.2310e+00, 1.1530e+00, ..., 0.0000e+00,
5.1107e-01, 1.3043e-01]],
[[0.0000e+00, 2.4027e+00, 0.0000e+00, ..., 8.2062e-01,
0.0000e+00, 1.1750e+00],
[0.0000e+00, 4.4811e-01, 1.1390e+00, ..., 0.0000e+00,
8.5415e-01, 0.0000e+00],
[7.2516e-01, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
3.0788e-01, 0.0000e+00],
...,
[3.2195e-01, 0.0000e+00, 7.5724e-01, ..., 0.0000e+00,
1.3083e+00, 1.5439e+00],
[6.8650e-01, 3.1598e-01, 0.0000e+00, ..., 1.8237e+00,
0.0000e+00, 1.9643e+00],
[0.0000e+00, 1.5988e+00, 0.0000e+00, ..., 9.7256e-02,
2.3014e-01, 7.7640e-02]],
[[3.9484e-01, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 7.6755e-01],
[1.2628e+00, 7.8810e-02, 0.0000e+00, ..., 1.9742e+00,
0.0000e+00, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 1.0417e+00, ..., 0.0000e+00,
7.0454e-01, 3.1250e-01],
...,
[1.9359e-01, 0.0000e+00, 9.8210e-01, ..., 0.0000e+00,
3.8394e-01, 0.0000e+00],
[0.0000e+00, 3.2974e-01, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 2.0021e-01],
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 5.9649e-01,
0.0000e+00, 0.0000e+00]],
...,
[[0.0000e+00, 0.0000e+00, 8.5655e-01, ..., 0.0000e+00,
0.0000e+00, 1.0052e+00],
[3.6246e-01, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
1.8805e+00, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 4.4412e-01, ..., 1.1574e+00,
0.0000e+00, 0.0000e+00],
...,
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
[7.5930e-01, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 1.1444e+00, ..., 0.0000e+00,
3.1190e-01, 0.0000e+00]],
[[0.0000e+00, 1.3781e+00, 1.3457e+00, ..., 2.8191e-01,
6.3913e-01, 0.0000e+00],
[0.0000e+00, 7.7578e-01, 3.8718e-01, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
[1.3419e+00, 0.0000e+00, 0.0000e+00, ..., 1.4992e+00,
0.0000e+00, 0.0000e+00],
...,
[0.0000e+00, 1.1803e+00, 0.0000e+00, ..., 4.3173e-01,
1.7696e+00, 1.3127e+00],
[3.9189e-01, 9.5379e-01, 0.0000e+00, ..., 1.3779e+00,
6.0881e-01, 0.0000e+00],
[1.0152e+00, 1.1613e+00, 3.6996e-02, ..., 0.0000e+00,
2.8808e-01, 0.0000e+00]],
[[1.3395e+00, 0.0000e+00, 1.0644e-01, ..., 0.0000e+00,
8.8641e-01, 1.4134e+00],
[7.6206e-01, 0.0000e+00, 0.0000e+00, ..., 4.9055e-01,
5.4889e-01, 7.2144e-02],
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
1.8425e+00, 0.0000e+00],
...,
[3.0384e-01, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
1.3174e+00, 6.1765e-01],
[0.0000e+00, 0.0000e+00, 7.6771e-01, ..., 1.6826e-01,
5.4228e-01, 6.7642e-01],
[0.0000e+00, 0.0000e+00, 1.0298e+00, ..., 6.9759e-01,
3.5460e-01, 1.5688e-01]]],
[[[0.0000e+00, 0.0000e+00, 8.3083e-01, ..., 8.2208e-01,
0.0000e+00, 3.3902e-01],
[4.5702e-01, 1.1061e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
[9.1549e-01, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 3.5947e-01],
...,
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 6.7693e-01,
0.0000e+00, 1.0915e-01],
[5.3579e-01, 3.1282e-01, 3.3284e-01, ..., 2.6364e-01,
0.0000e+00, 1.2258e-01],
[7.9742e-01, 6.2433e-01, 0.0000e+00, ..., 1.5049e+00,
0.0000e+00, 6.6239e-01]],
[[1.7167e+00, 0.0000e+00, 6.9336e-01, ..., 1.1487e+00,
7.6293e-01, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
1.3644e+00, 0.0000e+00],
[1.2310e+00, 0.0000e+00, 2.8227e-01, ..., 1.6871e-01,
0.0000e+00, 8.1177e-01],
...,
[0.0000e+00, 9.5035e-01, 0.0000e+00, ..., 7.0090e-01,
0.0000e+00, 1.7451e-01],
[7.6924e-01, 8.5412e-01, 6.6429e-01, ..., 0.0000e+00,
3.4891e-01, 1.0943e+00],
[0.0000e+00, 3.9243e-01, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00]],
[[0.0000e+00, 0.0000e+00, 6.6179e-01, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
[6.5661e-01, 4.0987e-01, 1.1061e+00, ..., 0.0000e+00,
0.0000e+00, 6.3384e-04],
[0.0000e+00, 6.0305e-01, 0.0000e+00, ..., 0.0000e+00,
1.4141e+00, 0.0000e+00],
...,
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 8.1925e-01, ..., 1.1502e+00,
7.7306e-01, 3.8062e-01],
[3.6750e-01, 0.0000e+00, 5.8874e-01, ..., 0.0000e+00,
1.1679e+00, 0.0000e+00]],
...,
[[7.2572e-02, 0.0000e+00, 0.0000e+00, ..., 1.2981e+00,
0.0000e+00, 7.5984e-01],
[1.0215e+00, 1.2642e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 1.2246e+00],
[5.2841e-01, 0.0000e+00, 1.0517e+00, ..., 6.2395e-01,
0.0000e+00, 0.0000e+00],
...,
[0.0000e+00, 0.0000e+00, 8.8293e-01, ..., 4.1917e-01,
0.0000e+00, 7.1817e-02],
[0.0000e+00, 5.4760e-02, 0.0000e+00, ..., 1.3879e-01,
7.9032e-02, 4.8557e-01],
[1.3336e+00, 0.0000e+00, 0.0000e+00, ..., 7.1977e-01,
0.0000e+00, 0.0000e+00]],
[[1.4812e+00, 0.0000e+00, 0.0000e+00, ..., 1.5886e+00,
0.0000e+00, 1.0172e+00],
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 2.7406e-01,
0.0000e+00, 1.4390e+00],
[1.8079e-02, 0.0000e+00, 7.5307e-01, ..., 3.8386e-01,
0.0000e+00, 1.5611e+00],
...,
[2.9982e-01, 7.8633e-01, 0.0000e+00, ..., 2.0574e-01,
8.5534e-01, 1.6281e-01],
[3.6866e-01, 4.5012e-01, 0.0000e+00, ..., 2.7055e-01,
1.6929e-01, 4.8798e-01],
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00]],
[[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 3.3577e-01,
0.0000e+00, 0.0000e+00],
[1.3914e+00, 0.0000e+00, 1.9247e+00, ..., 0.0000e+00,
3.4227e-02, 7.0526e-02],
[0.0000e+00, 0.0000e+00, 5.9702e-02, ..., 9.0311e-01,
1.2063e+00, 8.6109e-02],
...,
[1.9551e-01, 6.0416e-01, 1.1880e+00, ..., 0.0000e+00,
0.0000e+00, 3.5600e-01],
[4.6817e-01, 0.0000e+00, 0.0000e+00, ..., 1.8308e+00,
1.3346e+00, 2.1078e+00],
[3.9993e-01, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
9.0906e-02, 0.0000e+00]]],
[[[0.0000e+00, 1.1589e+00, 6.8628e-01, ..., 5.8383e-01,
8.1593e-02, 0.0000e+00],
[5.4499e-01, 4.8885e-01, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
[0.0000e+00, 2.7747e-02, 0.0000e+00, ..., 1.3477e-02,
2.9468e-01, 1.0412e+00],
...,
[7.4950e-02, 1.9002e-01, 1.3108e+00, ..., 0.0000e+00,
0.0000e+00, 1.8968e+00],
[6.2398e-01, 0.0000e+00, 0.0000e+00, ..., 1.4393e+00,
0.0000e+00, 1.7784e-01],
[0.0000e+00, 2.8042e-01, 3.9977e-01, ..., 1.1489e+00,
0.0000e+00, 0.0000e+00]],
[[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 1.2484e+00,
3.1758e-01, 8.8609e-01],
[2.2066e+00, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
1.0030e+00, 0.0000e+00],
[0.0000e+00, 1.6279e+00, 3.7213e-01, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
...,
[8.4627e-01, 1.3089e-01, 0.0000e+00, ..., 5.3309e-01,
9.0879e-02, 0.0000e+00],
[1.4561e-01, 6.9590e-02, 0.0000e+00, ..., 1.0687e+00,
0.0000e+00, 1.4466e-01],
[0.0000e+00, 3.8945e-02, 1.9406e+00, ..., 0.0000e+00,
4.2170e-01, 5.1668e-01]],
[[0.0000e+00, 1.4705e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 5.4783e-02, ..., 9.3180e-01,
0.0000e+00, 1.3193e+00],
[5.8835e-01, 0.0000e+00, 6.6633e-01, ..., 4.1009e-01,
0.0000e+00, 5.2082e-01],
...,
[0.0000e+00, 1.2277e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 2.4909e-02],
[1.0347e+00, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
9.6890e-01, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
1.0135e+00, 2.8619e-01]],
...,
[[0.0000e+00, 1.4298e-01, 1.4583e-01, ..., 0.0000e+00,
0.0000e+00, 2.1926e-01],
[2.4764e-01, 1.6878e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 7.5591e-02, ..., 1.4293e+00,
1.0836e+00, 9.5980e-01],
...,
[0.0000e+00, 4.0301e-01, 1.8300e+00, ..., 0.0000e+00,
0.0000e+00, 8.7928e-01],
[3.8023e-02, 7.5328e-01, 1.4639e+00, ..., 0.0000e+00,
0.0000e+00, 2.1443e+00],
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00]],
[[0.0000e+00, 0.0000e+00, 3.5555e-01, ..., 1.1441e+00,
2.1699e-01, 7.1550e-01],
[0.0000e+00, 6.5907e-01, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 2.0855e+00, ..., 0.0000e+00,
0.0000e+00, 4.2980e-01],
...,
[1.5165e+00, 2.4846e-01, 9.7999e-01, ..., 0.0000e+00,
0.0000e+00, 1.3794e+00],
[2.0921e-01, 0.0000e+00, 1.1321e+00, ..., 5.9194e-01,
0.0000e+00, 1.7787e+00],
[0.0000e+00, 1.3980e-01, 7.5684e-03, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00]],
[[0.0000e+00, 1.0029e+00, 0.0000e+00, ..., 9.3311e-01,
5.4762e-01, 2.1139e-01],
[5.7416e-01, 0.0000e+00, 0.0000e+00, ..., 1.1260e-01,
0.0000e+00, 1.2596e+00],
[5.1453e-01, 0.0000e+00, 9.7247e-01, ..., 7.4106e-01,
0.0000e+00, 1.6046e-01],
...,
[0.0000e+00, 3.9179e-01, 0.0000e+00, ..., 1.6839e+00,
0.0000e+00, 1.3445e-01],
[2.7410e-02, 1.2619e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
[0.0000e+00, 1.2642e-01, 3.8120e-01, ..., 0.0000e+00,
6.6013e-01, 8.9280e-01]]],
[[[0.0000e+00, 2.3729e-01, 1.7070e-01, ..., 1.1353e-01,
1.3327e+00, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 1.6408e-01, ..., 6.0199e-01,
0.0000e+00, 1.8030e-02],
[0.0000e+00, 0.0000e+00, 1.1612e+00, ..., 0.0000e+00,
1.3867e+00, 7.3637e-01],
...,
[0.0000e+00, 0.0000e+00, 8.9469e-01, ..., 7.2455e-01,
1.3070e+00, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 2.6492e-01, ..., 4.3001e-01,
0.0000e+00, 9.8854e-01],
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 9.0555e-01]],
[[1.5223e+00, 0.0000e+00, 0.0000e+00, ..., 7.0300e-01,
0.0000e+00, 0.0000e+00],
[1.4837e-01, 1.4082e+00, 0.0000e+00, ..., 9.4753e-01,
0.0000e+00, 0.0000e+00],
[0.0000e+00, 6.6915e-01, 0.0000e+00, ..., 0.0000e+00,
3.4305e-01, 0.0000e+00],
...,
[9.6070e-01, 1.5629e+00, 0.0000e+00, ..., 1.3924e+00,
0.0000e+00, 0.0000e+00],
[0.0000e+00, 7.6587e-01, 0.0000e+00, ..., 1.2394e+00,
7.3497e-01, 0.0000e+00],
[0.0000e+00, 2.4575e-02, 0.0000e+00, ..., 4.0591e-02,
0.0000e+00, 0.0000e+00]],
[[2.8708e-01, 0.0000e+00, 8.7484e-01, ..., 0.0000e+00,
0.0000e+00, 1.7880e+00],
[0.0000e+00, 0.0000e+00, 2.7959e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
...,
[1.2609e-01, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 8.7795e-01],
[3.9389e-01, 4.0679e-01, 0.0000e+00, ..., 0.0000e+00,
8.4093e-01, 7.4454e-01],
[6.2753e-01, 0.0000e+00, 1.8964e-01, ..., 5.5085e-01,
5.3822e-02, 3.4173e-01]],
...,
[[7.2486e-01, 4.6403e-04, 1.6452e-01, ..., 0.0000e+00,
7.9595e-01, 1.0951e+00],
[7.5681e-01, 0.0000e+00, 2.1609e-02, ..., 8.8101e-01,
0.0000e+00, 1.3852e-01],
[0.0000e+00, 0.0000e+00, 1.3872e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
...,
[0.0000e+00, 0.0000e+00, 1.8147e+00, ..., 0.0000e+00,
2.1428e-01, 5.0763e-01],
[1.8989e-01, 1.9251e+00, 0.0000e+00, ..., 1.2933e+00,
0.0000e+00, 1.4109e-01],
[1.0464e+00, 0.0000e+00, 0.0000e+00, ..., 7.2314e-01,
0.0000e+00, 1.5447e+00]],
[[6.5672e-01, 8.8381e-02, 0.0000e+00, ..., 1.1558e+00,
3.6011e-01, 0.0000e+00],
[0.0000e+00, 3.0102e-01, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 4.8790e-01],
[1.6768e+00, 0.0000e+00, 1.8726e-01, ..., 3.6481e-02,
1.2215e+00, 7.3722e-01],
...,
[6.6561e-01, 0.0000e+00, 1.2700e+00, ..., 1.3990e+00,
9.3496e-01, 1.3797e+00],
[0.0000e+00, 9.7489e-02, 0.0000e+00, ..., 1.1864e+00,
9.7903e-01, 4.2837e-01],
[0.0000e+00, 0.0000e+00, 8.1179e-01, ..., 0.0000e+00,
0.0000e+00, 6.1072e-01]],
[[4.9630e-01, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 8.9547e-01],
[7.2495e-01, 0.0000e+00, 8.6785e-01, ..., 0.0000e+00,
1.0072e+00, 3.3873e-01],
[0.0000e+00, 0.0000e+00, 2.4915e-01, ..., 0.0000e+00,
3.4924e-01, 0.0000e+00],
...,
[5.2499e-02, 9.1254e-01, 0.0000e+00, ..., 2.7929e-01,
0.0000e+00, 2.8307e-01],
[5.0577e-01, 4.8957e-01, 0.0000e+00, ..., 1.1805e+00,
0.0000e+00, 1.0580e+00],
[6.6028e-01, 0.0000e+00, 1.8995e-01, ..., 3.3843e-01,
0.0000e+00, 2.4419e-01]]],
[[[4.9783e-01, 0.0000e+00, 2.7451e-01, ..., 0.0000e+00,
0.0000e+00, 2.3531e-01],
[3.2154e-01, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
1.7924e+00, 0.0000e+00],
[7.0188e-01, 5.8265e-01, 0.0000e+00, ..., 1.3178e+00,
9.4335e-01, 0.0000e+00],
...,
[0.0000e+00, 0.0000e+00, 1.9460e+00, ..., 4.5137e-01,
3.3290e+00, 5.0397e-01],
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 2.1804e-01,
1.9568e-02, 8.2031e-01],
[0.0000e+00, 0.0000e+00, 9.5533e-01, ..., 0.0000e+00,
0.0000e+00, 1.0995e+00]],
[[0.0000e+00, 0.0000e+00, 4.9483e-01, ..., 2.4929e-01,
9.3641e-01, 0.0000e+00],
[2.3749e-01, 1.4653e-01, 0.0000e+00, ..., 1.2070e+00,
1.9448e+00, 0.0000e+00],
[3.7172e-01, 6.6949e-01, 1.1680e-01, ..., 0.0000e+00,
0.0000e+00, 2.8329e-01],
...,
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
[3.8689e-01, 0.0000e+00, 0.0000e+00, ..., 9.9326e-01,
4.7188e-01, 0.0000e+00],
[8.0520e-01, 1.0054e+00, 6.1926e-01, ..., 0.0000e+00,
0.0000e+00, 2.0081e-01]],
[[7.2211e-01, 1.9520e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 2.1821e+00,
1.5979e-01, 9.3610e-01],
[4.6789e-01, 1.0894e+00, 5.5708e-01, ..., 0.0000e+00,
0.0000e+00, 7.4993e-01],
...,
[6.7221e-01, 1.0692e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 6.6579e-01],
[0.0000e+00, 7.4521e-01, 1.6384e+00, ..., 1.2648e+00,
0.0000e+00, 0.0000e+00],
[1.4972e+00, 0.0000e+00, 0.0000e+00, ..., 8.4341e-01,
0.0000e+00, 5.6929e-01]],
...,
[[1.7058e+00, 7.2182e-01, 1.8254e-01, ..., 0.0000e+00,
2.9969e-01, 5.5203e-01],
[0.0000e+00, 1.0202e-01, 0.0000e+00, ..., 0.0000e+00,
4.8342e-01, 2.0406e-01],
[0.0000e+00, 0.0000e+00, 4.0311e-02, ..., 4.8591e-01,
8.9188e-01, 4.8178e-01],
...,
[1.1258e+00, 0.0000e+00, 0.0000e+00, ..., 0.0000e+00,
8.6303e-01, 6.9366e-01],
[0.0000e+00, 1.4386e+00, 0.0000e+00, ..., 1.6804e+00,
5.0834e-01, 0.0000e+00],
[7.2391e-01, 0.0000e+00, 3.5029e-01, ..., 1.0345e+00,
0.0000e+00, 8.8588e-01]],
[[0.0000e+00, 0.0000e+00, 3.1934e-01, ..., 6.1964e-01,
1.3395e+00, 1.4913e-01],
[4.2243e-01, 1.0406e+00, 5.0063e-01, ..., 3.7116e-02,
0.0000e+00, 0.0000e+00],
[1.4079e-01, 0.0000e+00, 1.7899e-01, ..., 0.0000e+00,
0.0000e+00, 4.7468e-01],
...,
[8.9060e-01, 0.0000e+00, 3.7541e-01, ..., 4.3928e-01,
0.0000e+00, 0.0000e+00],
[1.3567e+00, 2.0902e+00, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
[4.1899e-01, 0.0000e+00, 7.0255e-02, ..., 1.2891e-01,
0.0000e+00, 9.2617e-01]],
[[0.0000e+00, 1.3865e+00, 0.0000e+00, ..., 0.0000e+00,
1.7737e+00, 0.0000e+00],
[0.0000e+00, 1.7273e-01, 0.0000e+00, ..., 1.2442e+00,
0.0000e+00, 0.0000e+00],
[0.0000e+00, 3.5963e-01, 0.0000e+00, ..., 0.0000e+00,
0.0000e+00, 0.0000e+00],
...,
[0.0000e+00, 0.0000e+00, 0.0000e+00, ..., 1.7262e-01,
0.0000e+00, 0.0000e+00],
[0.0000e+00, 2.0881e-02, 2.1506e-01, ..., 2.2239e+00,
0.0000e+00, 0.0000e+00],
[2.0762e+00, 8.1629e-01, 0.0000e+00, ..., 1.4129e+00,
0.0000e+00, 9.0114e-01]]]], grad_fn=<ReluBackward0>)