Class: TorchRec::Sparse::JaggedTensor

Inherits:
Object
  • Object
show all
Defined in:
lib/torchrec/sparse/jagged_tensor.rb

Instance Method Summary collapse

Constructor Details

#initialize(values, weights: nil, lengths: nil, offsets: nil) ⇒ JaggedTensor

Returns a new instance of JaggedTensor.



4
5
6
7
8
9
10
11
12
13
14
15
16
 
# File 'lib/torchrec/sparse/jagged_tensor.rb', line 4

def initialize(values, weights: nil, lengths: nil, offsets: nil)
  @values = values
  @weights = weights
  assert_offsets_or_lengths_is_provided(offsets, lengths)
  if !offsets.nil?
    assert_tensor_has_no_elements_or_has_integers(offsets, "offsets")
  end
  if !lengths.nil?
    assert_tensor_has_no_elements_or_has_integers(lengths, "lengths")
  end
  @lengths = lengths
  @offsets = offsets
end