Class: SparseTensor

Inherits:
Hash
  • Object
show all
Defined in:
lib/graphkit.rb

Overview

A simple sparse tensor

Defined Under Namespace

Classes: RankError

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Methods inherited from Hash

#modify

Constructor Details

#initialize(rank = 2) ⇒ SparseTensor

Create a new tensor.



55
56
57
58
59
 
# File 'lib/graphkit.rb', line 55

def initialize(rank = 2)
  @rank = rank
  @shape = [0]*rank
  super()
end

Instance Attribute Details

#rank ⇒ Object (readonly)

Returns the value of attribute rank.



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

def rank
  @rank
end

#shape ⇒ Object (readonly)

Returns the value of attribute shape.



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

def shape
  @shape
end

Class Method Details

.diagonal(rank, array) ⇒ Object

Create a new diagonal tensor from an array. E.g. if rank was 2, then tensor = array tensor = array Etc.



66
67
68
69
70
71
72
 
# File 'lib/graphkit.rb', line 66

def self.diagonal(rank, array)
  tensor = new(rank)
  for index in 0...array.size
    tensor[[index] * rank] = array[index]
  end
  tensor
end

.from_hash(hash) ⇒ Object 



161
162
163
164
165
 
# File 'lib/graphkit.rb', line 161

def self.from_hash(hash)
  st = new(hash.keys[0].size)
  hash.each{|k,v| st[k] = v}
  st
end

Instance Method Details

#+(other) ⇒ Object 



135
136
137
 
# File 'lib/graphkit.rb', line 135

def +(other)
  scalar_binary(other){|a, b| a + b}
end

#-(other) ⇒ Object 



138
139
140
 
# File 'lib/graphkit.rb', line 138

def -(other)
  scalar_binary(other){|a, b| a - b}
end

#[](*args) ⇒ Object

Access an element of the tensor. E.g. for a rank 2 tensor

a = tensor

Raises:



78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
 
# File 'lib/graphkit.rb', line 78

def [](*args)
  args = args[0] if args.size == 1 and not args.size == @rank and args[0].size == @rank
#     p args

  raise RankError.new("Rank is #@rank, not #{args.size}") unless args.size == @rank
  return nil unless keys.include? args
  #if self.keys.include?(args) or @default_val == nil
    #eputs args.pretty_inspect
    #eputs self.pretty_inspect
    #eputs self.class, self.class.ancestors
    super(args)
  #else
    #return @default_val
  #end

end

#[]=(*args) ⇒ Object

Set an element of the tensor. E.g. for a rank 2 tensor

tensor = a_variable

Raises:



99
100
101
102
103
104
105
106
107
 
# File 'lib/graphkit.rb', line 99

def []=(*args)
  value = args.pop
  args = args[0] if args.size == 1 and args[0].size == @rank
  raise RankError.new("Rank is #@rank, not #{args.size}") unless args.size == @rank
  args.each_with_index do |arg, index|
    @shape[index] = [@shape[index], arg + 1].max
  end
  super(args, value)
end

#alter!(&block) ⇒ Object 



156
157
158
159
160
 
# File 'lib/graphkit.rb', line 156

def alter!(&block)
  self.keys.each do |k|
    self[k] = yield(self[k])
  end
end

#inspect ⇒ Object 



166
167
168
 
# File 'lib/graphkit.rb', line 166

def inspect
  "SparseTensor.from_hash(#{super})"
end

#max(&block) ⇒ Object

Find the maximum element of the tensor. See Enumerable#max.



144
145
146
 
# File 'lib/graphkit.rb', line 144

def max(&block)
  return self.values.max(&block)
end

#min(&block) ⇒ Object

Find the minimum element of the tensor. See Enumerable#max.



150
151
152
 
# File 'lib/graphkit.rb', line 150

def min(&block)
  return self.values.min(&block)
end

#scalar_binary(other, &block) ⇒ Object

Raises:

  • (ArgumentError) 


120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
 
# File 'lib/graphkit.rb', line 120

def scalar_binary(other, &block)
  raise ArgumentError unless other.class == self.class
  raise RankError.new("Different ranks: #@rank, #{other.rank}") unless other.rank == @rank

  new = self.class.new(@rank)
  self.keys.each do |key|
    if other[key]
      new[key] = yield(self[key], other[key])
    else
      new[key] = self[key]
    end
  end
  (other.keys - self.keys).each{|key| new[key] = other[key]}
  new
end