Hi,
I want to know how to change the tensor’s dimension in net.
self.net.add_module("out1", nn.softmax(dim=1)

and I got a tensor[ 1681, 1, 1 ] from output.But I want to  change it into tensor[ 41, 41, 1, 1 ].
Can someone please give me some advice.
Thanks.