public SDVariable diag(String name, SDVariable x) {
    SDValidation.validateNumerical("diag", "x", x);
    SDVariable out =  new org.nd4j.linalg.api.ops.impl.shape.Diag(sd,x).outputVariable();
    return sd.updateVariableNameAndReference(out, name);
  }