Module dfdx::nn::builders

Simple specification of network structure, without worrying about device or dtype.

Re-exports

• pub use super::*;

Structs

• Abs
  Calls abs().
• AddInto
  Add inputs together into a single tensor. T should be a tuple
• AvgPool2D
  Average pool with 2d kernel that operates on images (3d) and batches of images (4d). Each patch reduces to the average of the values in the patch.
• AvgPoolGlobal
  Applies average pooling over an entire image, fully reducing the height and width dimensions:
• BatchNorm1D
• BatchNorm2D
• Bias2D
• Conv2D
• Cos
  Calls cos().
• Dropout
  Does nothing as a Module, and calls dropout() as ModuleMut with probability 1.0 / N.
• DropoutOneIn
  Does nothing as a Module, and calls dropout() as ModuleMut with probability 1.0 / N.
• Embedding
• Exp
  Calls exp().
• Flatten2D
  Requires Nightly Flattens 3d tensors to 1d, and 4d tensors to 2d.
• GeLU
  Calls gelu().
• GeneralizedResidual
  A residual connection R around F: F(x) + R(x), as introduced in Deep Residual Learning for Image Recognition.
• LayerNorm1D
• Linear
• Ln
  Calls ln().
• MaxPool2D
  Max pool with 2d kernel that operates on images (3d) and batches of images (4d). Each patch reduces to the maximum value in that patch.
• MaxPoolGlobal
  Applies max pooling over an entire image, fully reducing the height and width dimensions:
• MinPool2D
  Minimum pool with 2d kernel that operates on images (3d) and batches of images (4d). Each patch reduces to the minimum of the values in the patch.
• MinPoolGlobal
  Applies min pooling over an entire image, fully reducing the height and width dimensions:
• MultiHeadAttention
• ReLU
  Calls relu().
• Repeated
  Repeats T N times. This requires that T’s input is the same as it’s output.
• Residual
  A residual connection around F: F(x) + x, as introduced in Deep Residual Learning for Image Recognition.
• Sigmoid
  Calls sigmoid().
• Sin
  Calls sin().
• Softmax
  Calls softmax().
• SplitInto
  Splits input into multiple heads. T should be a tuple, where every element of the tuple accepts the same input type.
• Sqrt
  Calls sqrt().
• Square
  Calls square().
• Tanh
  Calls tanh().
• Transformer
• TransformerDecoder
• TransformerDecoderBlock
• TransformerEncoder
• TransformerEncoderBlock
• UnbiasedLinear