class TensorFlow::Keras::Layers::Flatten

Public Class Methods

new(input_shape: nil) click to toggle source
# File lib/tensorflow/keras/layers/flatten.rb, line 5
def initialize(input_shape: nil)
  @input_shape = input_shape
end

Public Instance Methods

call(inputs) click to toggle source
# File lib/tensorflow/keras/layers/flatten.rb, line 18
def call(inputs)
  flattened_dim = inputs.shape[1..-1].inject(&:*)
  TensorFlow.reshape(inputs, [-1, flattened_dim])
end
count_params() click to toggle source
# File lib/tensorflow/keras/layers/flatten.rb, line 14
def count_params
  0
end
output_shape() click to toggle source
# File lib/tensorflow/keras/layers/flatten.rb, line 9
def output_shape
  flattened_dim = @input_shape.inject(&:*)
  [-1, flattened_dim]
end