Package io.github.kirstenali.deepj.nn

Class Autoencoder

java.lang.Object

io.github.kirstenali.deepj.nn.Module

io.github.kirstenali.deepj.nn.Autoencoder

All Implemented Interfaces:

Layer, Trainable

public final class Autoencoder
extends Module

Composable encoder/decoder, sharing the same parameter and autograd system as every DeepJ layer.

Constructor Summary

Constructors

Constructor	Description
Autoencoder(Layer encoder, Layer decoder)

Method Summary

Modifier and Type	Method	Description
protected List<Layer>	children()
Tensor	decode(Tensor encoded)
Tensor	encode(Tensor input)
Tensor	forward(Tensor input)

Methods inherited from class io.github.kirstenali.deepj.nn.Module

buffers, eval, isTraining, parameters, setTraining, train

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface io.github.kirstenali.deepj.layers.Layer

forward, supportsIncrementalInference

Methods inherited from interface io.github.kirstenali.deepj.training.Trainable

zeroGrad

Constructor Details

Autoencoder

public Autoencoder(Layer encoder,
 Layer decoder)

Method Details

encode

public Tensor encode(Tensor input)

decode

public Tensor decode(Tensor encoded)

children

protected List<Layer> children()

Overrides:

children in class Module

forward

public Tensor forward(Tensor input)