oneflow.rst 4.32 KB
Newer Older
yuguo's avatar
yuguo committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
oneflow
===================================
oneflow
----------------------------------
.. currentmodule:: oneflow
.. automodule:: oneflow
    :members: adaptive_avg_pool1d, 
            adaptive_avg_pool2d, 
            adaptive_avg_pool3d, 
            abs, 
            acos, 
            acosh, 
            add, 
            addcmul, 
            addmm, 
            all, 
            amin, 
            amax,
            any, 
            arccos, 
            arcsin, 
            arcsinh, 
            arccosh, 
            arctan, 
            arctanh, 
            argmax, 
            argmin, 
            arange, 
            argsort, 
            argwhere,
            asin,  
            asinh, 
            atan, 
            atan2, 
            atanh, 
            bernoulli,
            broadcast_like, 
            batch_gather,
            bmm,
            cat, 
            concat,
            cast, 
            ceil, 
            chunk, 
            clamp, 
            clip, 
            cos, 
            cosh, 
            diag, 
            select,
            diagonal,
            movedim,
            tensor_split,
            hsplit,
            vsplit,
            as_strided,
            div, 
            dot, 
            eq,
            einsum,
            equal, 
            expand, 
            eye,
            exp, 
            expm1, 
            erf, 
            erfc, 
            erfinv, 
            flatten, 
            flip, 
            floor, 
            floor_,
            fmod,
            full, 
            gather, 
            gather_nd, 
            gelu, 
            gt, 
            in_top_k, 
            index_select,
            linspace,
            logical_and,
            logical_or,
            logical_not,
            logical_xor,
            load, 
            log, 
            log2,
            log1p, 
            lt, 
            le, 
            masked_fill, 
            masked_select, 
            matmul, 
            mm, 
            mv, 
            narrow, 
            max, 
            mean,
            median,
            mish,  
            min, 
            meshgrid,
            mul, 
            neg, 
            negative, 
            new_ones,
            nonzero,
            normal,
            numel, 
            ne, 
            empty,
            ones, 
            ones_like, 
            pow,
            prod,  
            rand,
            randn,
            repeat, 
            repeat_interleave,
            reshape, 
            randint,
            randperm,
            reciprocal,
            roc_auc_score,
            roll,
            round, 
            rsqrt,
            save, 
            scatter,
            scatter_add,
            scatter_nd, 
            tensor_scatter_nd_update,
            sin, 
            sin_, 
            sinh, 
            sign, 
            selu, 
            silu, 
            slice, 
            slice_update,
            softsign, 
            sort, 
            softplus, 
            sigmoid, 
            softmax, 
            squeeze, 
            split, 
            stack, 
            std,
            sub, 
            sum, 
            sqrt, 
            square,  
            swapaxes, 
            swapdims, 
            tan, 
            tanh, 
            tensor, 
            tensordot,
            tile, 
            transpose,
            t,
            tril, 
            unsqueeze, 
            unbind, 
            permute,
            var, 
            where, 
            zeros, 
            zeros_like,
            is_nonzero,
            is_tensor,
            no_grad,
            set_grad_enabled,
            enable_grad,
            inference_mode,
            is_grad_enabled,
            is_floating_point,
            set_printoptions,
            decode_onerec,
            from_numpy,
            as_tensor,
            cumsum,
            topk,
            nms,
            cumprod,
            HalfTensor,
            FloatTensor,
            DoubleTensor,
            BoolTensor,
            ByteTensor,
            CharTensor,
            IntTensor,
            LongTensor,
            seed,
            manual_seed,
            initial_seed,
            get_rng_state,
            set_rng_state,
            isnan,
            isinf,
            searchsorted

.. autofunction:: oneflow.relu
.. autofunction:: oneflow.set_num_threads