Dimension 0 is the batch dimension and gives the number of samples in the current batch.
You can consider  x  being a single batch of samples.
Dimension 1 is the feature dimension in my use case and gives the number of different features.