I have a dataset in the form (batch,2,data_points) where 2 represents data points for in-phase and quadrature components. I want to normalize this, but not sure on which axis. S