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.
  This method is compatible with Numo NArray's [] method.
  @example
      a = Cumo::DFloat.new(4,5).seq
      => Cumo::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]
      => Cumo::DFloat#shape=[3]
      [6, 11, 16]
      a[1,[1,3,4]]
      => Cumo::DFloat#shape=[3]
      [6, 8, 9]
      a[true,2].fill(99)
      a
      => Cumo::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(-1)%>(int argc, VALUE *argv, VALUE self)
{
    int result_nd;
    size_t pos;
    char *ptr;

    result_nd = na_get_result_dimension(self, argc, argv, sizeof(dtype), &pos);
    if (result_nd) {
        return na_aref_main(argc, argv, self, 0, result_nd, pos);
    } else {
        ptr = na_get_pointer_for_read(self) + pos;
        SHOW_SYNCHRONIZE_WARNING_ONCE("<%=name%>", "<%=type_name%>");
        cumo_cuda_runtime_check_status(cudaDeviceSynchronize());
        return m_extract(ptr);
    }
}

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
cumo-0.1.0	ext/cumo/narray/gen/tmpl/aref_cpu.c