Is it possible to extend/apply the transforms.Normalize to normalize multidimensional tensor in custom pytroch dataset class? I have a tensor with shape (S x C x A x W) and I want to normalize on C dimension, here S: sequence length, C: feature channel, A: feature attributes, W: window length of each sub-sequence data.