Uses of Class
io.github.kirstenali.deepj.nn.Module

Packages that use Module
Package
Description
Neural networks composed from DeepJ tensors, layers, autograd and backend operations.
  • Uses of Module in io.github.kirstenali.deepj.nn

    Modifier and Type
    Class
    Description
    final class 
    Adapts DeepJ activations to the existing Layer contract.
    final class 
    Composable encoder/decoder, sharing the same parameter and autograd system as every DeepJ layer.
    final class 
    Channel-axis (axis 1) normalization, using backend reductions and persistent FLOAT32 running statistics.
    final class 
    Temporal cross-correlation for [batch, channels, length], sharing the 2D implementation.
    final class 
    Cross-correlation of [batch, channels, height, width] inputs, using gather and matrix multiplication.
    final class 
    Inverted dropout.
    final class 
    Bidirectional multi-head attention with optional padding masks and backend-resident softmax.
    final class 
    Flattens an inclusive axis range; defaults to preserving the first (batch) axis.
    final class 
    Averages every spatial axis of [batch, channels, ...], returning [batch, channels].
    final class 
    Batch-first GRU layer; compose layers with Sequential for stacked recurrence.
    final class 
    Reset-after GRU, with reset/update/candidate gate order and separate projection biases.
    final class 
    Batch-first LSTM layer; compose layers with Sequential for stacked recurrence.
    final class 
    LSTM with input/forget/candidate/output gate order and explicit cell memory.
    final class 
    Max/average pooling of [batch, channels, length] inputs.
    final class 
    Floor-mode max/average pooling of channel-first images.
    class 
    Batch-first recurrent sequence execution, with optional lengths and caller-owned initial/final state.
    class 
    Shared input/state validation and DeepJ linear projections for recurrent cells.
    final class 
    Reshapes the full tensor, with at most one inferred (-1) dimension.
    final class 
    Batch-first RNN layer; compose layers with Sequential for stacked recurrence.
    final class 
    Elman recurrent cell with tanh activation.
    final class 
    Ordered composition of existing or new DeepJ layers.
    final class 
    Stacked encoder for [batch, time, width] embeddings, with padding masks and masked mean pooling.
    final class 
    Pre-normalized, bidirectional encoder block reusing DeepJ linear layers and LayerNorm.