Module: NDAV::Torch::Tensor::MemoryViewable

Defined in:
lib/ndav/torch/tensor.rb

Instance Method Summary collapse

Instance Method Details

#ndav_descriptor ⇒ Object 



50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
 
# File 'lib/ndav/torch/tensor.rb', line 50

def ndav_descriptor(**)
  format = TYPE_TO_FORMAT[dtype]
  unless format
    warn "unsupported dtype: #{dtype}, currently supported: #{TYPE_TO_FORMAT.keys}"
    return false
  end

  byte_size = numel * element_size

  {
    data: ::Fiddle::Pointer.new(_data_ptr, byte_size),
    shape:,
    strides: stride.collect {|s| s * element_size},
    format:,
    byte_size:,
    readonly?: false,
    sub_offsets: nil
  }
end