Neural Network Normalization: Unsupported data rank

My NN implementation runs without normalization, but when I apply the NormalizerStandardize as follows I get the error
java.lang.RuntimeException: Unsupported data rank
I can’t figure out where this error is thrown, any help much appreciated

NormalizerStandardize normalizer = new NormalizerStandardize();
DataSet dataSet = new DataSet(features,labels,null,null);
normalizer.fit(dataSet);
normalizer.transform(dataSet);

@jiapei100 what shape is your data?