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