Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "74dcfba3727c90fc18cff9965545c37bc94b1ae9"
Switch branch/tag
Find file Normal viewHistoryPermalink
functional.rst 1.51 KB
Newer Older
yuguo's avatar
Oneflow 0.8 for DCU  
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 
oneflow.nn.functional
===========================================
Functional operations for neural networks
-------------------------------------------
.. currentmodule:: oneflow.nn.functional
.. autofunction:: conv1d
.. autofunction:: conv2d
.. autofunction:: conv3d
.. autofunction:: conv_transpose1d
.. autofunction:: conv_transpose2d
.. autofunction:: conv_transpose3d
.. autofunction:: adaptive_avg_pool1d
.. autofunction:: adaptive_avg_pool2d
.. autofunction:: adaptive_avg_pool3d
.. autofunction:: relu
.. autofunction:: hardsigmoid
.. autofunction:: hardshrink
.. autofunction:: hardswish
.. autofunction:: hardtanh
.. autofunction:: normalize
.. autofunction:: layer_norm
.. autofunction:: leaky_relu
.. autofunction:: elu
.. autofunction:: celu
.. autofunction:: selu
.. autofunction:: sigmoid
.. autofunction:: pad
.. autofunction:: prelu
.. autofunction:: logsigmoid 
.. autofunction:: log_softmax
.. autofunction:: gelu
.. autofunction:: glu
.. autofunction:: softsign
.. autofunction:: softmax 
.. autofunction:: softplus
.. autofunction:: tanh 
.. autofunction:: threshold
.. autofunction:: softshrink 
.. autofunction:: silu
.. autofunction:: mish
.. autofunction:: one_hot
.. autofunction:: triplet_margin_loss
.. autofunction:: dropout 
.. autofunction:: affine_grid
.. autofunction:: grid_sample
.. autofunction:: interpolate
.. autofunction:: ctc_greedy_decoder
.. autofunction:: sparse_softmax_cross_entropy
.. autofunction:: embedding
.. autofunction:: linear
.. autofunction:: cosine_similarity
.. autofunction:: cross_entropy