public INDArray where(INDArray x, INDArray condition) {
    NDValidation.validateNumerical("where", "x", x);
    NDValidation.validateBool("where", "condition", condition);
    return Nd4j.exec(new org.nd4j.linalg.api.ops.impl.controlflow.Where(x, condition))[0];
  }