public INDArray bitShift(INDArray x, INDArray shift) {
    NDValidation.validateInteger("bitShift", "x", x);
    NDValidation.validateInteger("bitShift", "shift", shift);
    return Nd4j.exec(new org.nd4j.linalg.api.ops.impl.transforms.custom.ShiftBits(x, shift))[0];
  }