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