Hello,
I would like to apply a BatchNorm1d just after a LinearLayer. My input is a 3D multivariate time series of shape [batch_size, n_variables, timesteps]. The LinearLayer performs the linear transformation on the third dimension so that the new shape is [batch_size, n_variables, LinearLayer_out_features]. My problem occurs with the BatchNorm1d, I would like to apply it on the third dimension but, for a 3D input, BatchNorm1d operation is done over the second dimension (even for a 3D tensor). Do you have any suggestion on how to do that? Something like axis permuation?
Thank you in advance.
Filo