DO NOT EDIT.
This file was autogenerated. Do not edit it by hand, since your modifications would be overwritten.
Functions
adaptive_average_pool(...): Adaptive average pooling operation.
adaptive_max_pool(...): Adaptive max pooling operation.
average_pool(...): Average pooling operation.
batch_normalization(...): Normalizes x by mean and variance.
binary_crossentropy(...): Computes binary cross-entropy loss between target and output tensor.
categorical_crossentropy(...): Computes categorical cross-entropy loss between target and output tensor.
celu(...): Continuously-differentiable exponential linear unit.
conv(...): General N-D convolution.
conv_transpose(...): General N-D convolution transpose.
ctc_decode(...): Decodes the output of a CTC model.
ctc_loss(...): CTC (Connectionist Temporal Classification) loss.
depth_to_space(...): Rearranges data from depth into blocks of spatial data.
depthwise_conv(...): General N-D depthwise convolution.
dot_product_attention(...): Scaled dot product attention function.
elu(...): Exponential Linear Unit activation function.
fold(...): Combines an array of sliding local blocks into a large containing tensor (reverses unfold).
gelu(...): Gaussian Error Linear Unit (GELU) activation function.
glu(...): Gated Linear Unit (GLU) activation function.
hard_shrink(...): Hard Shrink activation function.
hard_sigmoid(...): Hard sigmoid activation function.
hard_silu(...): Hard SiLU activation function, also known as Hard Swish.
hard_swish(...): Hard SiLU activation function, also known as Hard Swish.
hard_tanh(...): Applies the HardTanh function element-wise.
layer_normalization(...): Layer normalization layer (Ba et al., 2016).
leaky_relu(...): Leaky version of a Rectified Linear Unit activation function.
log_sigmoid(...): Logarithm of the sigmoid activation function.
log_softmax(...): Log-softmax activation function.
max_pool(...): Max pooling operation.
moments(...): Calculates the mean and variance of x.
multi_hot(...): Encodes integer labels as multi-hot vectors.
normalize(...): Normalizes x over the specified axis.
one_hot(...): Converts integer tensor x into a one-hot tensor.
polar(...): Constructs a complex tensor whose elements are Cartesian coordinates corresponding to the polar coordinates with absolute value abs and angle angle.
psnr(...): Peak Signal-to-Noise Ratio (PSNR) function.
relu(...): Rectified linear unit activation function.
relu6(...): Rectified linear unit activation function with upper bound of 6.
rms_normalization(...): Performs Root Mean Square (RMS) normalization on x.
selu(...): Scaled Exponential Linear Unit (SELU) activation function.
separable_conv(...): General N-D separable convolution.
sigmoid(...): Sigmoid activation function.
silu(...): Sigmoid Linear Unit (SiLU) activation function, also known as Swish.
soft_shrink(...): Soft Shrink activation function.
softmax(...): Softmax activation function.
softplus(...): Softplus activation function.
softsign(...): Softsign activation function.
space_to_depth(...): Rearranges blocks of spatial data into depth.
sparse_categorical_crossentropy(...): Computes sparse categorical cross-entropy loss.
sparse_plus(...): SparsePlus activation function.
sparse_sigmoid(...): Sparse sigmoid activation function.
sparsemax(...): Sparsemax activation function.
squareplus(...): Squareplus activation function.
swish(...): Sigmoid Linear Unit (SiLU) activation function, also known as Swish.
tanh_shrink(...): Applies the tanh shrink function element-wise.
threshold(...): Threshold activation function.
unfold(...): Extract sliding local blocks from a 4-D input (batched image).