Contents

module TensorStream
  # Class that defines a TensorStream placeholder
  class Placeholder < Tensor
    def initialize(data_type, rank, shape, options = {})
      setup_initial_state(options)

      @data_type = data_type.to_sym
      @rank = rank
      @shape = TensorShape.new(shape, rank)
      @value = nil
      @is_const = false

      @name = [@graph.get_name_scope, options[:name] || build_name].compact.reject(&:empty?).join("/")
      @op = Graph.get_default_graph.add_op!(:placeholder, data_type: @data_type, shape: @shape, internal_name: @name)
    end

    def inspect
      "Placeholder(#{@name} shape: #{@shape || "?"} data_type: #{@data_type})"
    end

    private

    def build_name
      "Placeholder#{graph.get_placeholder_counter}"
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
tensor_stream-1.0.2	lib/tensor_stream/placeholder.rb
tensor_stream-1.0.1	lib/tensor_stream/placeholder.rb