class

Num::NN::FlattenLayer(T)

Inherits Num::NN::Layer / Reference / Object

Constructors

new(context : Num::Grad::Context(T), shape : Array(Int32))
Source

Instance methods

forward(input : Num::Grad::Variable(T)) : Num::Grad::Variable(T)
Source
output_shape
Source
shape
Source