Add Docstrings to ./torch/_refs/__init__.py, ./torch/_refs/special/__init__.py, ./torch/masked/_ops.py

[svekars]

Add docstrings on the following lines. Follow these docstring guidelines. A docstring should have:

• A short one-sentence description in imperative mode.
• A longer description.
• A list of associated Args.
• A return value (for functions)
• An example
  Follow the existing patterns in the file you are editing. After adding the docstring, run:

lintrunner -a

• File: ./torch/_refs/special/__init__.py, Line: 231
• File: ./torch/masked/_ops.py, Line: 1479
• File: ./torch/_refs/__init__.py, Line: 1080
• File: ./torch/_refs/__init__.py, Line: 1090
• File: ./torch/_refs/__init__.py, Line: 1098
• File: ./torch/_refs/__init__.py, Line: 1106

cc @carljparker

[bjhargrave]

/assigntome

[bjhargrave]

What about torch/masked/_ops.py? I don't think this is covered by https://github.com/pytorch/pytorch/pull/113197/files. I have this one mostly done.

[kit1980]

@bjhargrave for the one function in torch/masked/_ops.py yes, the docstring is still needed.