public INDArray repeat(INDArray input, INDArray repeats, int axis) {
    NDValidation.validateNumerical("repeat", "input", input);
    NDValidation.validateNumerical("repeat", "repeats", repeats);
    return Nd4j.exec(new org.nd4j.linalg.api.ops.impl.shape.Repeat(input, repeats, axis))[0];
  }