__init__.py 7.9 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 23
# from .creation import create_tensor   #DEFINE_ALIAS
# from .creation import create_lod_tensor   #DEFINE_ALIAS
24
# from .creation import create_random_int_lod   #DEFINE_ALIAS
25
# from .creation import crop_tensor   #DEFINE_ALIAS
26
# from .creation import diag   #DEFINE_ALIAS
27
from .creation import eye  #DEFINE_ALIAS
28
# from .creation import fill_constant   #DEFINE_ALIAS
29 30
# from .creation import get__from_selected_rows   #DEFINE_ALIAS
from .creation import linspace  #DEFINE_ALIAS
31 32 33 34 35
# 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
36
from .creation import arange  #DEFINE_ALIAS
37
# from .creation import eye   #DEFINE_ALIAS
C
Chengmo 已提交
38
from .creation import full  # DEFINE_ALIAS
39
# from .creation import linspace   #DEFINE_ALIAS
P
Pei Yang 已提交
40
from .creation import full_like  #DEFINE_ALIAS
W
WuHaobo 已提交
41 42
from .creation import triu  #DEFINE_ALIAS
from .creation import tril  #DEFINE_ALIAS
S
suytingwan 已提交
43
from .creation import meshgrid  #DEFINE_ALIAS
44 45
# from .stat import mean   #DEFINE_ALIAS
# from .stat import reduce_mean   #DEFINE_ALIAS
L
Liufang Sang 已提交
46
from .stat import std  #DEFINE_ALIAS
47
# from .stat import var   #DEFINE_ALIAS
48
from .logic import equal  #DEFINE_ALIAS
49 50 51 52 53 54 55 56 57 58 59 60 61
# 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 已提交
62
from .logic import allclose  #DEFINE_ALIAS
63
from .logic import elementwise_equal  #DEFINE_ALIAS
64 65 66 67 68 69 70
# 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 已提交
71
from .random import randn  #DEFINE_ALIAS
X
Xing Wu 已提交
72
from .random import rand  #DEFINE_ALIAS
S
silingtong123 已提交
73
from .random import randint  #DEFINE_ALIAS
C
cc 已提交
74
from .random import randperm
75 76 77
# from .math import abs   #DEFINE_ALIAS
# from .math import acos   #DEFINE_ALIAS
# from .math import asin   #DEFINE_ALIAS
78
from .math import atan  #DEFINE_ALIAS
79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94
# 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
95
from .math import mul  #DEFINE_ALIAS
96
# from .math import multiplex   #DEFINE_ALIAS
97
from .math import pow  #DEFINE_ALIAS
98 99 100 101 102 103 104 105 106
# 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
107 108
from .math import sin  #DEFINE_ALIAS
from .math import sqrt  #DEFINE_ALIAS
109 110
# from .math import square   #DEFINE_ALIAS
# from .math import stanh   #DEFINE_ALIAS
111
from .math import sum  #DEFINE_ALIAS
112
# from .math import sums   #DEFINE_ALIAS
113
from .math import tanh  #DEFINE_ALIAS
114
from .math import elementwise_sum  #DEFINE_ALIAS
115 116
from .math import max  #DEFINE_ALIAS
from .math import min  #DEFINE_ALIAS
117
from .math import mm  #DEFINE_ALIAS
118 119
from .math import div  #DEFINE_ALIAS
from .math import add  #DEFINE_ALIAS
120
# from .math import atan   #DEFINE_ALIAS
121
from .math import logsumexp  #DEFINE_ALIAS
122
# from .math import inverse   #DEFINE_ALIAS
123
from .math import log1p  #DEFINE_ALIAS
124
# from .math import erf   #DEFINE_ALIAS
B
Bai Yifan 已提交
125
from .math import addcmul  #DEFINE_ALIAS
126
from .math import addmm  #DEFINE_ALIAS
127
from .math import clamp  #DEFINE_ALIAS
128 129 130 131
# from .attribute import rank   #DEFINE_ALIAS
# from .attribute import shape   #DEFINE_ALIAS
# from .io import save   #DEFINE_ALIAS
# from .io import load   #DEFINE_ALIAS
132
from .linalg import matmul  #DEFINE_ALIAS
L
liuwei1031 已提交
133
from .linalg import dot  #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
# from .linalg import cholesky   #DEFINE_ALIAS
141 142
# from .linalg import dot   #DEFINE_ALIAS
from .linalg import bmm  #DEFINE_ALIAS
143 144 145 146 147
# 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
148
from .manipulation import gather  #DEFINE_ALIAS
149 150 151 152 153 154 155 156
# 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
157 158 159
from .manipulation import split  #DEFINE_ALIAS
from .manipulation import squeeze  #DEFINE_ALIAS
from .manipulation import stack  #DEFINE_ALIAS
160 161 162 163
# 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
164
from .manipulation import unsqueeze  #DEFINE_ALIAS
165
# from .manipulation import unstack    #DEFINE_ALIAS
W
Wilber 已提交
166
from .manipulation import flip  #DEFINE_ALIAS
167
# from .manipulation import unbind   #DEFINE_ALIAS
168
from .manipulation import roll  #DEFINE_ALIAS
169
from .search import argmax  #DEFINE_ALIAS
170 171 172 173 174 175
# 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
176
from .search import where  #DEFINE_ALIAS
177
from .search import index_select  #DEFINE_ALIAS
C
Chengmo 已提交
178
from .search import index_sample  # DEFINE_ALIAS
179
from .search import nonzero  #DEFINE_ALIAS
180
from .search import sort  #DEFINE_ALIAS