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
Subclasses of Module in io.github.kirstenali.deepj.nnModifier and TypeClassDescriptionfinal classAdapts DeepJ activations to the existing Layer contract.final classComposable encoder/decoder, sharing the same parameter and autograd system as every DeepJ layer.final classChannel-axis (axis 1) normalization, using backend reductions and persistent FLOAT32 running statistics.final classTemporal cross-correlation for [batch, channels, length], sharing the 2D implementation.final classCross-correlation of [batch, channels, height, width] inputs, using gather and matrix multiplication.final classInverted dropout.final classBidirectional multi-head attention with optional padding masks and backend-resident softmax.final classFlattens an inclusive axis range; defaults to preserving the first (batch) axis.final classAverages every spatial axis of [batch, channels, ...], returning [batch, channels].final classBatch-first GRU layer; compose layers with Sequential for stacked recurrence.final classReset-after GRU, with reset/update/candidate gate order and separate projection biases.final classBatch-first LSTM layer; compose layers with Sequential for stacked recurrence.final classLSTM with input/forget/candidate/output gate order and explicit cell memory.final classMax/average pooling of [batch, channels, length] inputs.final classFloor-mode max/average pooling of channel-first images.classBatch-first recurrent sequence execution, with optional lengths and caller-owned initial/final state.classShared input/state validation and DeepJ linear projections for recurrent cells.final classReshapes the full tensor, with at most one inferred (-1) dimension.final classBatch-first RNN layer; compose layers with Sequential for stacked recurrence.final classElman recurrent cell with tanh activation.final classOrdered composition of existing or new DeepJ layers.final classStacked encoder for [batch, time, width] embeddings, with padding masks and masked mean pooling.final classPre-normalized, bidirectional encoder block reusing DeepJ linear layers and LayerNorm.