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