TorchLean API

NN.API.Public.Facade.Base.Root

TorchLean Public Names #

Root public names and training-log types used by import NN.

@[reducible, inline]

Tensor shape descriptor used to index spec-level tensors (Spec.Tensor α s).

Shape is an outermost-first tree:

  • .scalar for a scalar,
  • .dim n s for a length-n dimension whose entries have shape s.
Instances For
    @[reducible, inline]

    Runtime options such as backend, dtype, and CUDA fast-kernel selection.

    Instances For
      @[reducible, inline]

      One supervised training example with an input tensor and target tensor.

      Instances For
        @[reducible, inline]

        Shape-indexed sequential model. Most examples use the shorter nn.Sequential spelling.

        Instances For
          @[reducible, inline]
          abbrev TorchLean.ModelBuilder (α : Type u_1) :
          Type u_1

          Randomized model builder used by nn.run and built-in examples.

          Instances For
            @[reducible, inline]
            abbrev TorchLean.modelParamShapes {σ τ : Shape} (model : SequentialModel σ τ) :

            Parameter shapes required by a sequential model.

            Instances For
              @[reducible, inline]
              abbrev TorchLean.TensorPack (α : Type) (shapes : List Shape) :

              Shape-indexed tensor pack.

              This is the public name for TorchLean's typed tuple/argument-pack representation. A TensorPack α [s₁, s₂, ...] is a fixed tuple of tensors whose shapes are tracked by the type-level list.

              Instances For
                @[reducible, inline]
                abbrev TorchLean.ParamTensors (α : Type) (shapes : List Shape) :

                Concrete parameter tensors for a model or model slice.

                This is the root public name for shape-indexed parameter tensors. The nn.ParamTensors spelling points back to this same type so model code can stay on the shorter name.

                Instances For
                  @[reducible, inline]
                  abbrev TorchLean.ScalarModuleDef (paramShapes inputShapes : List Spec.Shape) :

                  Shape-indexed module definition used by the executable TorchLean training runtime.

                  Instances For
                    @[reducible, inline]

                    How a vector of per-example losses is reduced to one scalar loss.

                    Instances For
                      @[reducible, inline]

                      CSV parsing options used by the public data loaders.

                      Instances For
                        @[reducible, inline]

                        A scalar metric curve, usually a loss or accuracy series over training steps.

                        Instances For
                          @[reducible, inline]

                          JSON-serializable training log with metrics and run metadata.

                          Instances For
                            @[reducible, inline]

                            Mutable experiment log used by longer-running examples.

                            Instances For
                              @[reducible, inline]

                              Output destination for training logs.

                              Instances For
                                @[reducible, inline]

                                In-memory history for named training metrics.

                                Instances For
                                  @[reducible, inline]

                                  Finite in-memory dataset used by TorchLean trainers.

                                  Instances For
                                    @[reducible, inline]

                                    Stateful minibatch loader for finite datasets.

                                    Instances For