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