code question

[dengchengxifrank]

Thank you for your work.When I test on your code ,I do not understand why the second parameter is 2.

When the input is torch.randn(5, 1, 3, 127, 127), I get the result [1,2,32,32,32] ,I do not understand why the second parameter of result is 2.Waiting for your reply!