Sha256: 373beb2b63322939f203d3cec6a4b1906983147cca7e74fbe5bc5647a0c72fa4

Contents?: true

Size: 1.27 KB

Versions: 6

Compression:

Stored size: 1.27 KB

Contents

/*
  Array element referenece or slice view.
  @overload [](dim0,...,dimL)
  @param [Numeric,Range,etc] dim0,...,dimL  Multi-dimensional Index.
  @return [Numeric,NArray::<%=class_name%>] Element object or NArray view.

  --- Returns the element at +dim0+, +dim1+, ... are Numeric indices
  for each dimension, or returns a NArray View as a sliced subarray if
  +dim0+, +dim1+, ... includes other than Numeric index, e.g., Range
  or Array or true.

      a = Numo::DFloat.new(4,5).seq
      => Numo::DFloat#shape=[4,5]
      [[0, 1, 2, 3, 4],
       [5, 6, 7, 8, 9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19]]

      a[1,1]
      => 6.0

      a[1..3,1]
      => Numo::DFloat#shape=[3]
      [6, 11, 16]

      a[1,[1,3,4]]
      => Numo::DFloat#shape=[3]
      [6, 8, 9]

      a[true,2].fill(99)
      a
      => Numo::DFloat#shape=[4,5]
      [[0, 1, 99, 3, 4],
       [5, 6, 99, 8, 9],
       [10, 11, 99, 13, 14],
       [15, 16, 99, 18, 19]]
 */
static VALUE
<%=c_func%>(int argc, VALUE *argv, VALUE self)
{
    ssize_t pos;
    char *ptr;

    pos = na_get_scalar_position(self, argc, argv, sizeof(dtype));
    if (pos == -1) {
        return na_aref_main(argc, argv, self, 0);
    } else {
        ptr = na_get_pointer_for_read(self) + pos;
        return m_extract(ptr);
    }
}

Version data entries

6 entries across 6 versions & 1 rubygems

Version Path
numo-narray-0.9.0.3-x86-mingw32 ext/numo/narray/gen/tmpl/aref.c
numo-narray-0.9.0.3-x64-mingw32 ext/numo/narray/gen/tmpl/aref.c
numo-narray-0.9.0.3 ext/numo/narray/gen/tmpl/aref.c
numo-narray-0.9.0.1-x64-mingw32 ext/numo/narray/gen/tmpl/aref.c
numo-narray-0.9.0.1 ext/numo/narray/gen/tmpl/aref.c
numo-narray-0.9.0.1-x86-mingw32 ext/numo/narray/gen/tmpl/aref.c