Module: TensorFlow

Extended by:
Utils
Defined in:
lib/tensorflow.rb,
lib/tensorflow/ffi.rb,
lib/tensorflow/utils.rb,
lib/tensorflow/tensor.rb,
lib/tensorflow/context.rb,
lib/tensorflow/version.rb,
lib/tensorflow/variable.rb

Defined Under Namespace

Modules: FFI, Utils Classes: Context, Error, Tensor, Variable

Constant Summary collapse

VERSION = 
"0.1.0"

Class Attribute Summary collapse

Class Method Summary collapse

Methods included from Utils

infer_type

Class Attribute Details

.ffi_lib ⇒ Object

Returns the value of attribute ffi_lib.



15
16
17
 
# File 'lib/tensorflow.rb', line 15

def ffi_lib
  @ffi_lib
end

Class Method Details

.abs(x) ⇒ Object 



54
55
56
 
# File 'lib/tensorflow.rb', line 54

def abs(x)
  execute("Abs", [x])
end

.add(x, y) ⇒ Object 



38
39
40
 
# File 'lib/tensorflow.rb', line 38

def add(x, y)
  execute("Add", [x, y])
end

.assign_add_variable_op(resource, value) ⇒ Object 



98
99
100
 
# File 'lib/tensorflow.rb', line 98

def assign_add_variable_op(resource, value)
  execute("AssignAddVariableOp", [resource, value])
end

.assign_sub_variable_op(resource, value) ⇒ Object 



102
103
104
 
# File 'lib/tensorflow.rb', line 102

def assign_sub_variable_op(resource, value)
  execute("AssignSubVariableOp", [resource, value])
end

.assign_variable_op(resource, value) ⇒ Object 



106
107
108
 
# File 'lib/tensorflow.rb', line 106

def assign_variable_op(resource, value)
  execute("AssignVariableOp", [resource, value])
end

.constant(value, dtype: nil, shape: nil) ⇒ Object 



29
30
31
 
# File 'lib/tensorflow.rb', line 29

def constant(value, dtype: nil, shape: nil)
  Tensor.new(value, dtype: dtype, shape: shape)
end

.convert_to_tensor(value, dtype: nil) ⇒ Object 



33
34
35
36
 
# File 'lib/tensorflow.rb', line 33

def convert_to_tensor(value, dtype: nil)
  value = Tensor.new(value, dtype: dtype) unless value.is_a?(Tensor)
  value
end

.divide(x, y) ⇒ Object 



50
51
52
 
# File 'lib/tensorflow.rb', line 50

def divide(x, y)
  execute("Div", [x, y])
end

.equal(x, y) ⇒ Object 



78
79
80
 
# File 'lib/tensorflow.rb', line 78

def equal(x, y)
  execute("Equal", [x, y])
end

.fill(dims, value) ⇒ Object 



86
87
88
 
# File 'lib/tensorflow.rb', line 86

def fill(dims, value)
  execute("Fill", [dims, value])
end

.floormod(x, y) ⇒ Object 



66
67
68
 
# File 'lib/tensorflow.rb', line 66

def floormod(x, y)
  execute("Mod", [x, y])
end

.library_version ⇒ Object 



25
26
27
 
# File 'lib/tensorflow.rb', line 25

def library_version
  FFI.TF_Version
end

.matmul(x, y) ⇒ Object 



62
63
64
 
# File 'lib/tensorflow.rb', line 62

def matmul(x, y)
  execute("MatMul", [x, y])
end

.multiply(x, y) ⇒ Object 



46
47
48
 
# File 'lib/tensorflow.rb', line 46

def multiply(x, y)
  execute("Mul", [x, y])
end

.ones(dims) ⇒ Object 



94
95
96
 
# File 'lib/tensorflow.rb', line 94

def ones(dims)
  fill(dims, 1)
end

.range(start, limit, delta) ⇒ Object 



70
71
72
 
# File 'lib/tensorflow.rb', line 70

def range(start, limit, delta)
  execute("Range", [start, limit, delta])
end

.read_variable_op(resource, dtype) ⇒ Object 



110
111
112
 
# File 'lib/tensorflow.rb', line 110

def read_variable_op(resource, dtype)
  execute("ReadVariableOp", [resource], dtype: dtype)
end

.sqrt(x) ⇒ Object 



58
59
60
 
# File 'lib/tensorflow.rb', line 58

def sqrt(x)
  execute("Sqrt", [x])
end

.subtract(x, y) ⇒ Object 



42
43
44
 
# File 'lib/tensorflow.rb', line 42

def subtract(x, y)
  execute("Sub", [x, y])
end

.transpose(x, perm: [1, 0]) ⇒ Object 



74
75
76
 
# File 'lib/tensorflow.rb', line 74

def transpose(x, perm: [1, 0])
  execute("Transpose", [x, perm])
end

.var_handle_op(dtype, shape, container: "", shared_name: "") ⇒ Object 



114
115
116
 
# File 'lib/tensorflow.rb', line 114

def var_handle_op(dtype, shape, container: "", shared_name: "")
  execute("VarHandleOp", [], container: container, shared_name: shared_name, dtype: dtype, shape: shape)
end

.var_is_initialized_op(resource) ⇒ Object 



118
119
120
 
# File 'lib/tensorflow.rb', line 118

def var_is_initialized_op(resource)
  execute("VarIsInitializedOp", [resource])
end

.zeros(dims) ⇒ Object 



90
91
92
 
# File 'lib/tensorflow.rb', line 90

def zeros(dims)
  fill(dims, 0)
end

.zeros_like(x) ⇒ Object 



82
83
84
 
# File 'lib/tensorflow.rb', line 82

def zeros_like(x)
  execute("ZerosLike", [x])
end