Module: TensorFlow::Ops

Included in:
TensorFlow
Defined in:
lib/tensorflow/ops.rb

Instance Method Summary collapse

Instance Method Details

#eye(num_rows, num_columns: nil) ⇒ Object 



4
5
6
7
8
9
 
# File 'lib/tensorflow/ops.rb', line 4

def eye(num_rows,  num_columns: nil)
  num_columns ||= num_rows
  zeros = self.zeros([num_rows, num_columns])
  ones = self.ones([num_rows])
  RawOps.matrix_set_diag(input: zeros, diagonal: ones)
end

#fill(dims, value) ⇒ Object 



11
12
13
 
# File 'lib/tensorflow/ops.rb', line 11

def fill(dims, value)
  RawOps.fill(dims: dims, value: value)
end

#identity(input) ⇒ Object 



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

def identity(input)
  RawOps.identity(input: input)
end

#matmul(a, b) ⇒ Object 



19
20
21
 
# File 'lib/tensorflow/ops.rb', line 19

def matmul(a, b)
  RawOps.mat_mul(a: a, b: b)
end

#ones(dims) ⇒ Object 



23
24
25
 
# File 'lib/tensorflow/ops.rb', line 23

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

#range(start, limit = nil, delta = 1) ⇒ Object 



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

def range(start, limit = nil, delta = 1)
  unless limit
    limit = start
    start = 0
  end
  RawOps.range(start: start, limit: limit, delta: delta)
end

#timestamp ⇒ Object 



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

def timestamp
  RawOps.timestamp
end

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



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

def transpose(x, perm: [1, 0])
  RawOps.transpose(x: x, perm: perm)
end

#zeros(dims) ⇒ Object 



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

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

#zeros_like(x) ⇒ Object 



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

def zeros_like(x)
  RawOps.zeros_like(x: x)
end