Model Summaries

Structured model-summary rendering for checked sequential models.

def TorchLean.nn.paramCount (shapes : List Shape) : ℕ

def TorchLean.nn.shapeDims : Shape → List ℕ

Dimensions of a tensor shape, outermost first. Scalars have no dimensions.

def TorchLean.nn.shapeDisplay (s : Shape) : String

User-facing tensor shape display for one model-summary shape.

def TorchLean.nn.shapeListString (shapes : List Shape) : String

User-facing display for a list of parameter tensor shapes.

structure TorchLean.nn.LayerSummary : Type

Structured per-layer summary derived from a checked sequential model.

• index : ℕ

  Zero-based position in the sequential layer list.

• kind : String

  User-facing layer kind string.

• inputShape : Shape

  Checked input shape for this layer.

• outputShape : Shape

  Checked output shape for this layer.

• paramShapes : List Shape

  Parameter tensor shapes owned by this layer.

• paramCount : ℕ

  Total scalar parameter count for this layer.

def TorchLean.nn.LayerSummary.render (s : LayerSummary) : String

One-line rendering of one layer summary.

@[implicit_reducible]

instance TorchLean.nn.LayerSummary.instToString : ToString LayerSummary

structure TorchLean.nn.ModelSummary : Type

Structured whole-model summary derived from a checked sequential model.

• inputShape : Shape

  Checked input shape for the full model.

• outputShape : Shape

  Checked output shape for the full model.

• layers : List LayerSummary

  Per-layer summaries in order.

• layerCount : ℕ

  Total number of layers in the sequential model.

• totalParams : ℕ

  Total scalar parameter count across all layers.

def TorchLean.nn.ModelSummary.header (s : ModelSummary) : String

Header line for the model summary.

def TorchLean.nn.ModelSummary.render (s : ModelSummary) : String

Multi-line rendering of the structured model summary.

@[implicit_reducible]

instance TorchLean.nn.ModelSummary.instToString : ToString ModelSummary

def TorchLean.nn.layerSummariesAux {σ τ : Shape} : ℕ → Sequential σ τ → List LayerSummary

Recursive worker that records one summary row for each layer in a sequential model.

def TorchLean.nn.summary {σ τ : Shape} (model : Sequential σ τ) : ModelSummary

Structured checked model summary for a sequential model.

def TorchLean.nn.info {σ τ : Shape} (model : Sequential σ τ) : String

Model description derived from a sequential model value.

This walks the checked Seq itself, so the printed layer list stays attached to the model that will actually run.

def TorchLean.nn.printInfo {σ τ : Shape} (model : Sequential σ τ) : IO Unit

Print the checked model description for a sequential model.