It may help if you can share a reproducible snippet. It’s unclear what is the shape of your data and dims.