public SDVariable setDiag(SDVariable in, SDVariable diag) {
    SDValidation.validateNumerical("setDiag", "in", in);
    SDValidation.validateNumerical("setDiag", "diag", diag);
    return new org.nd4j.linalg.api.ops.impl.transforms.custom.MatrixSetDiag(sd,in, diag).outputVariable();
  }