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