Contents

module TensorFlow
  module Keras
    module Layers
      class Dropout
        def initialize(rate)
        end

        def build(input_shape)
          @output_shape = input_shape
        end

        def call(inputs)
          # TODO implement
          TensorFlow.identity(inputs)
        end

        def output_shape
          @output_shape
        end

        def count_params
          0
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
tensorflow-0.2.0	lib/tensorflow/keras/layers/dropout.rb