mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
ee97d80be2
Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/144208 Approved by: https://github.com/davidberard98
79 lines
1.9 KiB
Python
79 lines
1.9 KiB
Python
# mypy: allow-untyped-defs
|
|
from textwrap import dedent
|
|
from typing import Any
|
|
|
|
import torch.jit
|
|
|
|
|
|
def execWrapper(code, glob, loc):
|
|
exec(code, glob, loc)
|
|
|
|
|
|
def _gen_unsupported_methods_properties():
|
|
tensor_attrs = set(filter(lambda x: x[0] != "_", dir(torch.Tensor)))
|
|
tensor = torch.tensor([2])
|
|
funcs_template = dedent(
|
|
"""
|
|
def func(x):
|
|
return x.{op}()
|
|
"""
|
|
)
|
|
|
|
deprecated_apis = {
|
|
"volatile",
|
|
"resize",
|
|
"reinforce",
|
|
"new",
|
|
"name",
|
|
"map2_",
|
|
"has_names",
|
|
"grad_fn",
|
|
"resize_as",
|
|
}
|
|
tensor_attrs = tensor_attrs - deprecated_apis
|
|
|
|
properties = []
|
|
methods = []
|
|
sorted_tensor_attrs = sorted(tensor_attrs, key=lambda x: x.lower())
|
|
for attr in sorted_tensor_attrs:
|
|
funcs_str = funcs_template.format(op=attr)
|
|
scope: dict[str, Any] = {}
|
|
execWrapper(funcs_str, globals(), scope)
|
|
try:
|
|
torch.jit.CompilationUnit(funcs_str)
|
|
except Exception as e:
|
|
if "nonexistent attribute" not in repr(e):
|
|
continue
|
|
attr_repr = repr(getattr(tensor, attr))
|
|
if "bound method" in attr_repr or "built-in method" in attr_repr:
|
|
methods.append(attr)
|
|
else:
|
|
properties.append(attr)
|
|
|
|
mapped_methods = ("\t* :meth:`~torch.Tensor." + x + r"`" for x in methods)
|
|
mapped_properties = ("\t* :attr:`~torch.Tensor." + x + r"`" for x in properties)
|
|
return "\n".join(mapped_methods), "\n".join(mapped_properties)
|
|
|
|
|
|
def _list_unsupported_tensor_ops():
|
|
header = """\n\n
|
|
Unsupported Tensor Methods
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
"""
|
|
methods, properties = _gen_unsupported_methods_properties()
|
|
return (
|
|
header
|
|
+ "\n"
|
|
+ methods
|
|
+ """
|
|
|
|
Unsupported Tensor Properties
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
"""
|
|
+ "\n"
|
|
+ properties
|
|
)
|
|
|
|
|
|
__doc__ = _list_unsupported_tensor_ops()
|