__init__.py 7.8 KB
Newer Older
1
# Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserved
2 3 4 5 6 7 8 9 10 11 12 13
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
14 15 16 17 18
from __future__ import print_function

#from .math import *
#from .creation import *
#from .linalg import *
19

C
Chengmo 已提交
20
# TODO: define alias in tensor and framework directory
21 22
# from .creation import create_tensor   #DEFINE_ALIAS
# from .creation import create_lod_tensor   #DEFINE_ALIAS
23
# from .creation import create_random_int_lod   #DEFINE_ALIAS
24
# from .creation import crop_tensor   #DEFINE_ALIAS
25
# from .creation import diag   #DEFINE_ALIAS
26
from .creation import eye  #DEFINE_ALIAS
27
# from .creation import fill_constant   #DEFINE_ALIAS
28 29
# from .creation import get__from_selected_rows   #DEFINE_ALIAS
from .creation import linspace  #DEFINE_ALIAS
30 31 32 33 34
# from .creation import ones   #DEFINE_ALIAS
# from .creation import ones_like   #DEFINE_ALIAS
# from .creation import range   #DEFINE_ALIAS
# from .creation import zeros   #DEFINE_ALIAS
# from .creation import zeros_like   #DEFINE_ALIAS
35
from .creation import arange  #DEFINE_ALIAS
36
# from .creation import eye   #DEFINE_ALIAS
C
Chengmo 已提交
37
from .creation import full  # DEFINE_ALIAS
38
# from .creation import linspace   #DEFINE_ALIAS
P
Pei Yang 已提交
39
from .creation import full_like  #DEFINE_ALIAS
W
WuHaobo 已提交
40 41
from .creation import triu  #DEFINE_ALIAS
from .creation import tril  #DEFINE_ALIAS
S
suytingwan 已提交
42
from .creation import meshgrid  #DEFINE_ALIAS
43 44 45 46
# from .stat import mean   #DEFINE_ALIAS
# from .stat import reduce_mean   #DEFINE_ALIAS
# from .stat import std   #DEFINE_ALIAS
# from .stat import var   #DEFINE_ALIAS
47
from .logic import equal  #DEFINE_ALIAS
48 49 50 51 52 53 54 55 56 57 58 59 60
# from .logic import greater_equal   #DEFINE_ALIAS
# from .logic import greater_than   #DEFINE_ALIAS
# from .logic import is_empty   #DEFINE_ALIAS
# from .logic import isfinite   #DEFINE_ALIAS
# from .logic import less_equal   #DEFINE_ALIAS
# from .logic import less_than   #DEFINE_ALIAS
# from .logic import logical_and   #DEFINE_ALIAS
# from .logic import logical_not   #DEFINE_ALIAS
# from .logic import logical_or   #DEFINE_ALIAS
# from .logic import logical_xor   #DEFINE_ALIAS
# from .logic import not_equal   #DEFINE_ALIAS
# from .logic import reduce_all   #DEFINE_ALIAS
# from .logic import reduce_any   #DEFINE_ALIAS
Z
Zhen Wang 已提交
61
from .logic import allclose  #DEFINE_ALIAS
62
from .logic import elementwise_equal  #DEFINE_ALIAS
63 64 65 66 67 68 69
# from .logic import isnan   #DEFINE_ALIAS
# from . import Tensor   #DEFINE_ALIAS
# from . import LoDTensor   #DEFINE_ALIAS
# from . import LoDTensorArray   #DEFINE_ALIAS
# from .random import gaussin   #DEFINE_ALIAS
# from .random import uniform   #DEFINE_ALIAS
# from .random import shuffle   #DEFINE_ALIAS
G
guofei 已提交
70
from .random import randn  #DEFINE_ALIAS
71
# from .random import rand   #DEFINE_ALIAS
S
silingtong123 已提交
72
from .random import randint  #DEFINE_ALIAS
C
cc 已提交
73
from .random import randperm
74 75 76
# from .math import abs   #DEFINE_ALIAS
# from .math import acos   #DEFINE_ALIAS
# from .math import asin   #DEFINE_ALIAS
77
from .math import atan  #DEFINE_ALIAS
78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93
# from .math import ceil   #DEFINE_ALIAS
# from .math import cos   #DEFINE_ALIAS
# from .math import cumsum   #DEFINE_ALIAS
# from .math import elementwise_add   #DEFINE_ALIAS
# from .math import elementwise_div   #DEFINE_ALIAS
# from .math import elementwise_floordiv   #DEFINE_ALIAS
# from .math import elementwise_max   #DEFINE_ALIAS
# from .math import elementwise_min   #DEFINE_ALIAS
# from .math import elementwise_mod   #DEFINE_ALIAS
# from .math import elementwise_mul   #DEFINE_ALIAS
# from .math import elementwise_pow   #DEFINE_ALIAS
# from .math import elementwise_sub   #DEFINE_ALIAS
# from .math import exp   #DEFINE_ALIAS
# from .math import floor   #DEFINE_ALIAS
# from .math import increment   #DEFINE_ALIAS
# from .math import log   #DEFINE_ALIAS
94
from .math import mul  #DEFINE_ALIAS
95
# from .math import multiplex   #DEFINE_ALIAS
96
from .math import pow  #DEFINE_ALIAS
97 98 99 100 101 102 103 104 105
# from .math import reciprocal   #DEFINE_ALIAS
# from .math import reduce_max   #DEFINE_ALIAS
# from .math import reduce_min   #DEFINE_ALIAS
# from .math import reduce_prod   #DEFINE_ALIAS
# from .math import reduce_sum   #DEFINE_ALIAS
# from .math import round   #DEFINE_ALIAS
# from .math import rsqrt   #DEFINE_ALIAS
# from .math import scale   #DEFINE_ALIAS
# from .math import sign   #DEFINE_ALIAS
106 107
from .math import sin  #DEFINE_ALIAS
from .math import sqrt  #DEFINE_ALIAS
108 109
# from .math import square   #DEFINE_ALIAS
# from .math import stanh   #DEFINE_ALIAS
110
from .math import sum  #DEFINE_ALIAS
111
# from .math import sums   #DEFINE_ALIAS
112
from .math import tanh  #DEFINE_ALIAS
113
from .math import elementwise_sum  #DEFINE_ALIAS
114 115
from .math import max  #DEFINE_ALIAS
from .math import min  #DEFINE_ALIAS
116
from .math import mm  #DEFINE_ALIAS
117 118
from .math import div  #DEFINE_ALIAS
from .math import add  #DEFINE_ALIAS
119
# from .math import atan   #DEFINE_ALIAS
120
from .math import logsumexp  #DEFINE_ALIAS
121
# from .math import inverse   #DEFINE_ALIAS
122
from .math import log1p  #DEFINE_ALIAS
123
# from .math import erf   #DEFINE_ALIAS
124
from .math import addcmul  #DEFINE_ALIAS
125
from .math import addmm  #DEFINE_ALIAS
126
from .math import clamp  #DEFINE_ALIAS
127 128 129 130
# from .attribute import rank   #DEFINE_ALIAS
# from .attribute import shape   #DEFINE_ALIAS
# from .io import save   #DEFINE_ALIAS
# from .io import load   #DEFINE_ALIAS
131
from .linalg import matmul  #DEFINE_ALIAS
L
liuwei1031 已提交
132
from .linalg import dot  #DEFINE_ALIAS
133
from .linalg import bmm  #DEFINE_ALIAS
134
# from .linalg import einsum   #DEFINE_ALIAS
135
from .linalg import norm  #DEFINE_ALIAS
136
# from .linalg import transpose   #DEFINE_ALIAS
Z
Zhang Ting 已提交
137
from .linalg import dist  #DEFINE_ALIAS
138
from .linalg import t  #DEFINE_ALIAS
139
from .linalg import cross  #DEFINE_ALIAS
140 141 142 143 144 145
# from .linalg import cholesky   #DEFINE_ALIAS
# from .manipulation import cast   #DEFINE_ALIAS
# from .manipulation import concat   #DEFINE_ALIAS
# from .manipulation import expand   #DEFINE_ALIAS
# from .manipulation import expand_as   #DEFINE_ALIAS
# from .manipulation import flatten   #DEFINE_ALIAS
146
from .manipulation import gather  #DEFINE_ALIAS
147 148 149 150 151 152 153 154
# from .manipulation import gather_nd   #DEFINE_ALIAS
# from .manipulation import reshape   #DEFINE_ALIAS
# from .manipulation import reverse   #DEFINE_ALIAS
# from .manipulation import scatter   #DEFINE_ALIAS
# from .manipulation import scatter_nd_add   #DEFINE_ALIAS
# from .manipulation import scatter_nd   #DEFINE_ALIAS
# from .manipulation import shard_index   #DEFINE_ALIAS
# from .manipulation import slice   #DEFINE_ALIAS
155 156 157
from .manipulation import split  #DEFINE_ALIAS
from .manipulation import squeeze  #DEFINE_ALIAS
from .manipulation import stack  #DEFINE_ALIAS
158 159 160 161
# from .manipulation import strided_slice   #DEFINE_ALIAS
# from .manipulation import transpose   #DEFINE_ALIAS
# from .manipulation import unique   #DEFINE_ALIAS
# from .manipulation import unique_with_counts   #DEFINE_ALIAS
162
from .manipulation import unsqueeze  #DEFINE_ALIAS
163
# from .manipulation import unstack   #DEFINE_ALIAS
W
Wilber 已提交
164
from .manipulation import flip  #DEFINE_ALIAS
165
# from .manipulation import unbind   #DEFINE_ALIAS
166
from .manipulation import roll  #DEFINE_ALIAS
167
from .search import argmax  #DEFINE_ALIAS
168 169 170 171 172 173
# from .search import argmin   #DEFINE_ALIAS
# from .search import argsort   #DEFINE_ALIAS
# from .search import has_inf   #DEFINE_ALIAS
# from .search import has_nan   #DEFINE_ALIAS
# from .search import masked_select   #DEFINE_ALIAS
# from .search import topk   #DEFINE_ALIAS
174
from .search import where  #DEFINE_ALIAS
175
from .search import index_select  #DEFINE_ALIAS
C
Chengmo 已提交
176
from .search import index_sample  # DEFINE_ALIAS
177
from .search import nonzero  #DEFINE_ALIAS
178
from .search import sort  #DEFINE_ALIAS