TorchLean API

NN.API.Neural.Layers

Neural Layers #

Named configurations for attention, convolution, and pooling. The seeded constructors that consume them live in NN.API.Seeded; spatial operators state the trailing axes they consume, while batchShape records any axes mapped pointwise by the layer.