Skip to content
Merged
Show file tree
Hide file tree
Changes from 7 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
326 changes: 46 additions & 280 deletions python/paddle/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -286,6 +286,7 @@
from .tensor.math import frac # noqa: F401
from .tensor.math import sgn # noqa: F401
from .tensor.math import take # noqa: F401
from .tensor.math import frexp # noqa: F401

from .tensor.random import bernoulli # noqa: F401
from .tensor.random import poisson # noqa: F401
Expand Down Expand Up @@ -386,285 +387,50 @@
os.environ.setdefault('runtime_include_dir', runtime_include_dir)

disable_static()

__all__ = [ # noqa
'iinfo',
'dtype',
'uint8',
'int8',
'int16',
'int32',
'int64',
'float16',
'float32',
'float64',
'bfloat16',
'bool',
'complex64',
'complex128',
'addmm',
'allclose',
'isclose',
't',
'add',
'subtract',
'diag',
'diagflat',
'isnan',
'scatter_nd_add',
'unstack',
'get_default_dtype',
'save',
'multinomial',
'get_cuda_rng_state',
'rank',
'empty_like',
'eye',
'cumsum',
'cumprod',
'logcumsumexp',
'logit',
'LazyGuard',
'sign',
'is_empty',
'equal',
'equal_all',
'is_tensor',
'is_complex',
'is_integer',
'cross',
'where',
'log1p',
'cos',
'tan',
'mean',
'mode',
'mv',
'in_dynamic_mode',
'min',
'amin',
'any',
'slice',
'normal',
'logsumexp',
'full',
'unsqueeze',
'unsqueeze_',
'argmax',
'Model',
'summary',
'flops',
'sort',
'searchsorted',
'bucketize',
'split',
'vsplit',
'logical_and',
'full_like',
'less_than',
'kron',
'clip',
'Tensor',
'crop',
'ParamAttr',
'stanh',
'randint',
'randint_like',
'assign',
'gather',
'scale',
'zeros',
'rsqrt',
'squeeze',
'squeeze_',
'to_tensor',
'gather_nd',
'isinf',
'uniform',
'floor_divide',
'remainder',
'floor_mod',
'roll',
'batch',
'max',
'amax',
'logical_or',
'bitwise_and',
'bitwise_or',
'bitwise_xor',
'bitwise_not',
'mm',
'flip',
'rot90',
'bincount',
'histogram',
'multiplex',
'CUDAPlace',
'NPUPlace',
'empty',
'shape',
'real',
'imag',
'is_floating_point',
'complex',
'reciprocal',
'rand',
'less_equal',
'triu',
'sin',
'dist',
'unbind',
'meshgrid',
'arange',
'load',
'numel',
'median',
'nanmedian',
'quantile',
'nanquantile',
'no_grad',
'set_grad_enabled',
'is_grad_enabled',
'mod',
'abs',
'tril',
'pow',
'zeros_like',
'maximum',
'topk',
'index_select',
'CPUPlace',
'matmul',
'seed',
'acos',
'logical_xor',
'exp',
'expm1',
'bernoulli',
'poisson',
'sinh',
'round',
'DataParallel',
'argmin',
'prod',
'broadcast_shape',
'conj',
'neg',
'lgamma',
'lerp',
'erfinv',
'inner',
'outer',
'square',
'divide',
'ceil',
'atan',
'atan2',
'rad2deg',
'deg2rad',
'gcd',
'lcm',
'expand',
'broadcast_to',
'ones_like',
'index_sample',
'cast',
'grad',
'all',
'ones',
'not_equal',
'sum',
'nansum',
'nanmean',
'count_nonzero',
'tile',
'greater_equal',
'isfinite',
'create_parameter',
'dot',
'increment',
'erf',
'bmm',
'chunk',
'tolist',
'tensordot',
'greater_than',
'shard_index',
'argsort',
'tanh',
'tanh_',
'transpose',
'randn',
'strided_slice',
'unique',
'unique_consecutive',
'set_cuda_rng_state',
'set_printoptions',
'std',
'flatten',
'asin',
'multiply',
'disable_static',
'masked_select',
'var',
'trace',
'enable_static',
'scatter_nd',
'set_default_dtype',
'disable_signal_handler',
'expand_as',
'stack',
'sqrt',
'randperm',
'linspace',
'logspace',
'reshape',
'reshape_',
'reverse',
'nonzero',
'CUDAPinnedPlace',
'logical_not',
'add_n',
'minimum',
'scatter',
'scatter_',
'floor',
'cosh',
'log',
'log2',
'log10',
'concat',
'check_shape',
'trunc',
'frac',
'digamma',
'standard_normal',
'diagonal',
'broadcast_tensors',
'einsum',
'set_flags',
'get_flags',
'asinh',
'acosh',
'atanh',
'as_complex',
'as_real',
'diff',
'angle',
'fmax',
'fmin',
'moveaxis',
'repeat_interleave',
'clone',
'kthvalue',
'renorm',
'take_along_axis',
'put_along_axis',
'heaviside',
'tril_indices',
'index_add',
"index_add_",
'sgn',
'triu_indices',
'take',
'iinfo', 'dtype', 'uint8', 'int8', 'int16', 'int32', 'int64', 'float16',
'float32', 'float64', 'bfloat16', 'bool', 'complex64', 'complex128',
'addmm', 'allclose', 'isclose', 't', 'add', 'subtract', 'diag', 'diagflat',
'isnan', 'scatter_nd_add', 'unstack', 'get_default_dtype', 'save',
'multinomial', 'get_cuda_rng_state', 'rank', 'empty_like', 'eye', 'cumsum',
'cumprod', 'logcumsumexp', 'logit', 'LazyGuard', 'sign', 'is_empty',
'equal', 'equal_all', 'is_tensor', 'is_complex', 'is_integer', 'cross',
'where', 'log1p', 'cos', 'tan', 'mean', 'mode', 'mv', 'in_dynamic_mode',
'min', 'amin', 'any', 'slice', 'normal', 'logsumexp', 'full', 'unsqueeze',
'unsqueeze_', 'argmax', 'Model', 'summary', 'flops', 'sort', 'searchsorted',
'bucketize', 'split', 'vsplit', 'logical_and', 'full_like', 'less_than',
'kron', 'clip', 'Tensor', 'crop', 'ParamAttr', 'stanh', 'randint',
'randint_like', 'assign', 'gather', 'scale', 'zeros', 'rsqrt', 'squeeze',
'squeeze_', 'to_tensor', 'gather_nd', 'isinf', 'uniform', 'floor_divide',
'remainder', 'floor_mod', 'roll', 'batch', 'max', 'amax', 'logical_or',
'bitwise_and', 'bitwise_or', 'bitwise_xor', 'bitwise_not', 'mm', 'flip',
'rot90', 'bincount', 'histogram', 'multiplex', 'CUDAPlace', 'NPUPlace',
'empty', 'shape', 'real', 'imag', 'is_floating_point', 'complex',
'reciprocal', 'rand', 'less_equal', 'triu', 'sin', 'dist', 'unbind',
'meshgrid', 'arange', 'load', 'numel', 'median', 'nanmedian', 'quantile',
'nanquantile', 'no_grad', 'set_grad_enabled', 'is_grad_enabled', 'mod',
'abs', 'tril', 'pow', 'zeros_like', 'maximum', 'topk', 'index_select',
'CPUPlace', 'matmul', 'seed', 'acos', 'logical_xor', 'exp', 'expm1',
'bernoulli', 'poisson', 'sinh', 'round', 'DataParallel', 'argmin', 'prod',
'broadcast_shape', 'conj', 'neg', 'lgamma', 'lerp', 'erfinv', 'inner',
'outer', 'square', 'divide', 'ceil', 'atan', 'atan2', 'rad2deg', 'deg2rad',
'gcd', 'lcm', 'expand', 'broadcast_to', 'ones_like', 'index_sample', 'cast',
'grad', 'all', 'ones', 'not_equal', 'sum', 'nansum', 'nanmean',
'count_nonzero', 'tile', 'greater_equal', 'isfinite', 'create_parameter',
'dot', 'increment', 'erf', 'bmm', 'chunk', 'tolist', 'tensordot',
'greater_than', 'shard_index', 'argsort', 'tanh', 'tanh_', 'transpose',
'randn', 'strided_slice', 'unique', 'unique_consecutive',
'set_cuda_rng_state', 'set_printoptions', 'std', 'flatten', 'asin',
'multiply', 'disable_static', 'masked_select', 'var', 'trace',
'enable_static', 'scatter_nd', 'set_default_dtype',
'disable_signal_handler', 'expand_as', 'stack', 'sqrt', 'randperm',
'linspace', 'logspace', 'reshape', 'reshape_', 'reverse', 'nonzero',
'CUDAPinnedPlace', 'logical_not', 'add_n', 'minimum', 'scatter', 'scatter_',
'floor', 'cosh', 'log', 'log2', 'log10', 'concat', 'check_shape', 'trunc',
'frac', 'digamma', 'standard_normal', 'diagonal', 'broadcast_tensors',
'einsum', 'set_flags', 'get_flags', 'asinh', 'acosh', 'atanh', 'as_complex',
'as_real', 'diff', 'angle', 'fmax', 'fmin', 'moveaxis', 'repeat_interleave',
'clone', 'kthvalue', 'renorm', 'take_along_axis', 'put_along_axis',
'heaviside', 'tril_indices', 'index_add', "index_add_", 'sgn',
'triu_indices', 'take', 'frexp'
]
Loading