public INDArray[] split(INDArray input, INDArray numSplit, int splitDim) {
    NDValidation.validateNumerical("split", "input", input);
    NDValidation.validateNumerical("split", "numSplit", numSplit);
    return Nd4j.exec(new org.nd4j.linalg.api.ops.impl.shape.Split(input, numSplit, splitDim));
  }