mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
4fef3763dd
Summary: Original PR: https://github.com/pytorch/pytorch/pull/37419 cc mattip suo Pull Request resolved: https://github.com/pytorch/pytorch/pull/37778 Differential Revision: D21385774 Pulled By: ezyang fbshipit-source-id: 5de532faab8bae132736b6b5189e0ee2ac9935be
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
import torch.jit
|
|
from textwrap import dedent
|
|
|
|
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 = set(["volatile", "resize", "reinforce", "new", "name", "map2_", "has_names", "grad_fn", "resize_as"])
|
|
tensor_attrs = tensor_attrs - deprecated_apis
|
|
|
|
properties = []
|
|
methods = []
|
|
sorted_tensor_attrs = sorted(list(tensor_attrs), key=lambda x: x.lower())
|
|
for attr in sorted_tensor_attrs:
|
|
funcs_str = funcs_template.format(op=attr)
|
|
scope = {}
|
|
execWrapper(funcs_str, globals(), scope)
|
|
try:
|
|
cu = 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)
|
|
|
|
methods = map(lambda x: "\t* :meth:`~torch.Tensor." + x + r"`", methods)
|
|
properties = map(lambda x: "\t* :attr:`~torch.Tensor." + x + r"`", properties)
|
|
return "\n".join(methods), "\n".join(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()
|