Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
nn.rst 2.49 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
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 
oneflow.nn
===================================
Operators for neural networks
----------------------------------
.. currentmodule:: oneflow.nn
.. automodule:: oneflow.nn
    :members: AdaptiveAvgPool1d,
        AdaptiveAvgPool2d,
        AdaptiveAvgPool3d,
        AvgPool1d,
        AvgPool2d,
        AvgPool3d,
        BCELoss,
        BCEWithLogitsLoss,
        BatchNorm1d,
        BatchNorm2d,
        BatchNorm3d,
        COCOReader,
        CTCLoss,
        CoinFlip,
        ConstantPad1d,
        ConstantPad2d,
        ConstantPad3d,
        Conv1d,
        Conv2d,
        Conv3d,
        ConvTranspose1d,
        ConvTranspose2d,
        ConvTranspose3d,
        CosineSimilarity,
        CombinedMarginLoss,
        CropMirrorNormalize,
        CrossEntropyLoss,
        Dropout,
        ELU,
        CELU,
        Embedding,
        Flatten,
        GELU,
        GLU,
        GroupNorm,
        Hardsigmoid,
        Hardshrink,
        Hardswish,
        Hardtanh,
        Identity,
        InstanceNorm1d,
        InstanceNorm2d,
        InstanceNorm3d,
        KLDivLoss,
        L1Loss,
        LayerNorm,
        LeakyReLU,
        Linear,
        LogSigmoid,
        LogSoftmax,
        MSELoss,
        MarginRankingLoss,
        TripletMarginLoss,
        MaxPool1d,
        MaxPool2d,
        MaxPool3d,
        ModuleDict,
        ModuleList,
        Mish,
        NLLLoss,
        OFRecordImageDecoder,
        OFRecordImageDecoderRandomCrop,
        OFRecordRawDecoder,
        OFRecordReader,
        OFRecordBytesDecoder,
        PReLU,
        Parameter,
        ParameterDict,
        ParameterList,
        PixelShuffle,
        ReLU,
        ReLU6,
        ReflectionPad2d,
        ReplicationPad2d,
        Sequential, 
        SELU, 
        SiLU, 
        Sigmoid,
        SmoothL1Loss,
        Softmax,
        Softplus, 
        Softshrink,
        Softsign, 
        Tanh,
        Threshold,
        Upsample,
        UpsamplingBilinear2d,
        UpsamplingNearest2d,
        ZeroPad2d,
        MinMaxObserver,
        MovingAverageMinMaxObserver,
        FakeQuantization,
        Quantization, 
        FusedBatchNorm1d, 
        FusedBatchNorm2d, 
        FusedBatchNorm3d, 
        FusedMLP, 

.. autofunction:: oneflow.nn.modules.pixelshuffle.PixelShufflev2

.. autofunction:: oneflow.nn.parallel.DistributedDataParallel

.. currentmodule:: oneflow.nn.utils
.. autofunction:: oneflow.nn.utils.clip_grad_norm_
.. autofunction:: oneflow.nn.utils.weight_norm
.. autofunction:: oneflow.nn.utils.remove_weight_norm