public SDVariable[] split(SDVariable input, SDVariable numSplit, int splitDim) {
    SDValidation.validateNumerical("split", "input", input);
    SDValidation.validateNumerical("split", "numSplit", numSplit);
    return new org.nd4j.linalg.api.ops.impl.shape.Split(sd,input, numSplit, splitDim).outputVariables();
  }