Top Level Namespace

Instance Method Summary collapse

Instance Method Details

#Const(scope, value, type = nil) ⇒ Object

Package op defines functions for adding TensorFlow operations to a Graph.

Functions for adding an operation to a graph take a Scope object as the first argument. The Scope object encapsulates a graph and a set of properties (such as a name prefix) for all operations being added to the graph.

WARNING: The API in this package has not been finalized and can change without notice. Const adds an operation to graph that produces value as output.



11
12
13
14
15
 
# File 'lib/tensorflow/op.rb', line 11

def Const(scope, value, type = nil)
    value = Tensorflow::Tensor.new(value, type)
    opspec = Tensorflow::OpSpec.new('', 'Const', 'dtype' => {value.type_num => 'DataType'}, 'value' => {value => 'tensor'})
    scope.AddOperation(opspec).output(0)
end

#CString(string) ⇒ Object

A simple makeshift function to convert a ruby string to C++ string



35
36
37
 
# File 'lib/tensorflow/op.rb', line 35

def CString(string)
    String.new(string)
end

#Placeholder(scope, dtype) ⇒ Object

A placeholder op for a value that will be fed into the computation.

N.B. This operation will fail with an error if it is executed. It is intended as a way to represent a value that will always be fed, and to provide attrs that enable the fed value to be checked at runtime.

Arguments: dtype: The type of elements in the tensor.

Returns A placeholder tensor that must be replaced using the feed mechanism.



27
28
29
30
31
32
 
# File 'lib/tensorflow/op.rb', line 27

def Placeholder(scope, dtype)
    optionalAttr = {}
    optionalAttr['dtype'] = dtype
    opspec = Tensorflow::OpSpec.new('', 'Placeholder', 'dtype' => {dtype => 'DataType'})
    scope.AddOperation(opspec).output(0)
end