Package io.github.kirstenali.deepj.nn

Class Flatten

java.lang.Object

io.github.kirstenali.deepj.nn.Module

io.github.kirstenali.deepj.nn.Flatten

All Implemented Interfaces:

Layer, Trainable

public final class Flatten
extends Module

Flattens an inclusive axis range; defaults to preserving the first (batch) axis.

Constructor Summary

Constructors

Constructor	Description
Flatten()
Flatten(int first, int last)

Method Summary

Modifier and Type	Method	Description
Tensor	forward(Tensor input)

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

buffers, children, 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

Flatten

public Flatten()

Flatten

public Flatten(int first,
 int last)

Method Details

forward

public Tensor forward(Tensor input)