I’m trying to check if the result is good with one piece of data, but I keep seeing an error.
What should I do?

It occure the error
‘RuntimeError: CHECK_ARG(tensor->nDimension >= 2 && tensor->nDimension <= 5) failed at torch/csrc/cudnn/BatchNorm.cpp:13’