TorchLean API

Docs HomeInstallationGuideExamplesGraphsTools

NN.Spec.Module.Autoencoder

Autoencoder as an Spec.Module #

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

source
def Spec.Module.autoencoder {α : Type} [TorchLean.Storage α] [Context α] {inputDim hiddenDim : ℕ} (m : AutoencoderSpec α inputDim hiddenDim) :
Module α [inputDim] [inputDim]

Autoencoder module specification following Spec.Module.

Instances For