oneflow.nn.functional

.. The documentation is referenced from: https://pytorch.org/docs/1.10/nn.functional.html.

.. contents:: oneflow.nn.functional :depth: 2 :local: :class: this-will-duplicate-information-and-it-is-still-useful-here :backlinks: top

.. currentmodule:: oneflow.nn.functional

Convolution functions

.. autosummary:: :toctree: generated :nosignatures:

conv1d
conv2d
conv3d
conv_transpose1d
conv_transpose2d
conv_transpose3d
fold
unfold

Normalization functions

.. autosummary:: :toctree: generated :nosignatures:

batch_norm
layer_norm
normalize
group_norm

Pooling functions

.. autosummary:: :toctree: generated :nosignatures:

avg_pool1d
avg_pool2d
avg_pool3d
max_pool1d
max_pool2d
max_pool3d
max_unpool1d
max_unpool2d
max_unpool3d
adaptive_avg_pool1d
adaptive_avg_pool2d
adaptive_avg_pool3d
adaptive_max_pool1d
adaptive_max_pool2d
adaptive_max_pool3d

Non-linear activation functions

.. autosummary:: :toctree: generated :nosignatures:

threshold
relu
hardtanh
hardswish
relu6
elu
selu
celu
leaky_relu
square_relu
prelu
glu
gelu
quick_gelu
logsigmoid
hardshrink
softsign
softplus
softmax
softshrink
log_softmax
gumbel_softmax
tanh
sigmoid
hardsigmoid
silu
mish

Linear functions

.. autosummary:: :toctree: generated :nosignatures:

linear

Dropout functions

.. autosummary:: :toctree: generated :nosignatures:

dropout
dropout1d
dropout2d
dropout3d

Sparse functions

.. autosummary:: :toctree: generated :nosignatures:

embedding
one_hot

Distance functions

.. autosummary:: :toctree: generated :nosignatures:

cosine_similarity
pairwise_distance

Loss functions

.. autosummary:: :toctree: generated :nosignatures:

sparse_softmax_cross_entropy
cross_entropy
ctc_loss
l1_loss
mse_loss
smooth_l1_loss
triplet_margin_loss
binary_cross_entropy
binary_cross_entropy_with_logits

Vision functions

.. autosummary:: :toctree: generated :nosignatures:

deform_conv2d
pad
interpolate
upsample
grid_sample
affine_grid

Greedy decoder

.. autosummary:: :toctree: generated :nosignatures:

ctc_greedy_decoder