Contents

module TensorStream
  class VariableScope
    attr_accessor :name, :reuse, :initializer

    def initialize(name: '', reuse: nil, initializer: nil)
      @name = name
      @reuse = reuse
      @initializer = initializer
    end

    def get_variable(name, dtype: nil, shape: nil, initializer: nil, trainable: true, collections: nil, validate_shape: false)
      TensorStream::Variable.new(dtype || :float32, nil, shape, self, collections: collections, name: name, initializer: initializer, trainable: trainable)
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
tensor_stream-0.8.0	lib/tensor_stream/variable_scope.rb