vm.torch.nn.functional

Functions

affine(…) : Apply the affine transformation to input.

avg_pool2d(…) : Apply the 2d average pooling to input.

batch_norm(…) : Apply the batch normalization to input. [Ioffe & Szegedy, 2015].

binary_cross_entropy_with_logits(…) : Compute the sigmoid cross entropy with contiguous target.

conv2d(…) : Apply 2d convolution to the input.

conv_transpose2d(…) : Apply 2d deconvolution to the input.

cross_entropy(…) : Compute the softmax cross entropy with sparse labels.

ctc_loss(…) : Compute the ctc loss with batched labels. [Graves & Gomez, 2006].

depthwise_conv2d(…) : Apply 2d depthwise convolution to the input. [Chollet, 2016].

drop_block2d(…) : Set the spatial blocks over input to zero randomly. [Ghiasi et.al, 2018].

drop_path(…) : Set the examples over input to zero randomly. [Larsson et.al, 2016].

dropout(…) : Set the elements of the input to zero randomly. [Srivastava et.al, 2014].

elu(…) : Apply the exponential linear unit to input. [Clevert et.al, 2015].

group_norm(…) : Apply the group normalization to input. [Wu & He, 2018].

l1_loss(…) : Compute the element-wise absolute value difference.

leaky_relu(…) : Apply the leaky rectified linear unit to input.

linear(…) : Apply the linear transformation to input.

local_response_norm(…) : Apply the local response normalization to input. [Krizhevsky et.al, 2012].

log_softmax(…) : Apply the composite of logarithm and softmax to input.

interpolate(…) : Resize input via interpolating neighborhoods.

max_pool2d(…) : Apply the 2d max pooling to input.

mse_loss(…) : Compute the element-wise squared error.

nll_loss(…) : Compute the negative likelihood loss with sparse labels.

normalize(…) : Apply the \(L_{p}\) normalization to the input.

pad(…) : Pad the input according to the given sizes.

prelu(…) : Apply the parametric rectified linear unit to input. [He et.al, 2015].

relu(…) : Apply rectified linear unit to input. [Nair & Hinton, 2010].

relu6(…) : Apply the clipped-6 rectified linear unit to input. [Krizhevsky, 2010].

selu(…) : Compute the sigmoid focal loss with sparse labels. [Lin et.al, 2017].

smooth_l1_loss(…) : Compute the element-wise error transited from L1 and L2. [Girshick, 2015].

softmax(…) : Apply the softmax function to input.

softmax_cross_entropy_with_logits(…) : Compute the softmax cross entropy with contiguous targets.

sync_batch_norm(…) : Apply the sync batch normalization to input. [Ioffe & Szegedy, 2015].

tanh(…) : Apply the tanh function to input.

upsample(…) : Upsample input via interpolating neighborhoods.

upsample_bilinear(…) : Upsample input via bilinear interpolating.

upsample_nearest(…) : Upsample input via nearest interpolating.