Module candle_nn::ops

Structs

• Dropout

Functions

• dropout
• hard_sigmoid
• leaky_relu
• log_softmax
• pixel_shuffle
• pixel_unshuffle
• replication_pad2d
• rms_norm
• rms_norm_slow
• sigmoid
• silu
• softmax
  Applies the softmax function to the input tensor, rescaling the element so that elements on a slice of fixed index on dimension dim are between 0 and 1 and sum to 1.
• softmax_last_dim
• swiglu