How to access like Ruby each?

[naitoh]

• Ruby Array each.

  > a = [[[0.0, 1.0], [2.0, 3.0]], [[4.0, 5.0], [6.0, 7.0]]]
  > a.each {|i| p i}
  [[0.0, 1.0], [2.0, 3.0]]
  [[4.0, 5.0], [6.0, 7.0]]

• Numo::NArray each.

  > a = Numo::DFloat.new(2,2,2).seq
  => Numo::DFloat#shape=[2,2,2]
  [[[0, 1], 
  [2, 3]], 
  [[4, 5], 
  [6, 7]]] 
  > a.each {|i| p i}
  0.0
  1.0
  2.0
  3.0
  4.0
  5.0
  6.0
  7.0

As below, I want to use each array in the first dimension of a multidimensional array.

a.shape[0].times{|i| p a[i, false]}
Numo::DFloat(view)#shape=[2,2]
[[0, 1], 
 [2, 3]]
Numo::DFloat(view)#shape=[2,2]
[[4, 5], 
 [6, 7]]

Is there a way to write this more easily?

[masa16]

How about NArray#each_over_axis ? https://github.com/ruby-numo/numo-narray/commit/2a5aae80805a4bba88e7bdb2919b7a9812d8838f Method name comes from numpy.apply_over_axes : https://docs.scipy.org/doc/numpy/reference/generated/numpy.apply_over_axes.html#numpy.apply_over_axes

[naitoh]

Thank you for your reply.

How about NArray#each_over_axis ?

I think it's good. I am glad if this can be used.

[naitoh]

I confirmed on numo-narray 0.9.1.3. Thank you new feature.