class DNN::Layers::Flatten
Public Instance Methods
compute_output_shape()
click to toggle source
# File lib/dnn/core/layers/basic_layers.rb, line 300 def compute_output_shape [@input_shape.reduce(:*)] end
forward(x)
click to toggle source
# File lib/dnn/core/layers/basic_layers.rb, line 296 def forward(x) Reshape.(x, @output_shape) end