class Num::NN::FlattenLayer(T)

Defined in:

nn/layers/flatten.cr

Constructors

Instance Method Summary

Instance methods inherited from class Num::NN::Layer(T)

forward(input : Num::Grad::Variable(T)) forward, output_shape : Array(Int32) output_shape, variables : Array(Num::Grad::Variable(T)) variables

Constructor Detail

def self.new(context : Num::Grad::Context(T), shape : Array(Int32)) #

[View source]

Instance Method Detail

def forward(input : Num::Grad::Variable(T)) : Num::Grad::Variable(T) #

[View source]
def output_shape : Array(Int32) #

[View source]
def shape : Array(Int32) #

[View source]