I have batch x 4096 x 6(time samples) data in nn.
I want to nn.Relu, so I want to reduce the data dimension to
batch x 512 x 6(time samples)
but the nn.Relu layer take the last dimension, if I understand correct.
How can I do a Relu, in the wanted dimension…?