TorchLean API

Docs HomeGuideExamplesGraphs

NN.Spec.Module.Autoencoder

Autoencoder as an NNModuleSpec #

The autoencoder spec model defines the forward pass and its VJP pieces. This file adds the NNModuleSpec wrapper so it can be composed with other modules and exported.

source
def Spec.autoencoderActivationToPyTorch (activation_type : String) :
String

Small helper used by exporters to render a PyTorch-style activation string.

Instances For
    source
    def Spec.AutoencoderModuleSpec {α : Type} [Context α] {inputDim hiddenDim : ℕ} (m : AutoencoderSpec α inputDim hiddenDim) :
    ModSpec.NNModuleSpec α (Shape.dim inputDim Shape.scalar) (Shape.dim inputDim Shape.scalar)

    Autoencoder module specification following NNModuleSpec.

    Instances For