cupyx 名前空間に属する関数の使い方について

Question

前提・実現したいこと

python Numpy の初心者です。
Numpy と　cupy のコードは概ね同じシグネチャということで，大きな変更を伴わずに使えるということで学習を始めたのですが，例えば，numpy のadd.at　関数を cupyではscatter_add関数に置き換えるところまではわかったのですが，レファレンスのサンプルを動かしてもエラーが出ます。というのも，scatter_add関数はcupyx という名前空間にあるということが理由のようでした。どなたか，解決方法をご教示いただけませんか。

発生している問題・エラーメッセージ

traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'cupyx' is not defined

該当のソースコード

>>> import numpy
>>> import cupy
>>> a = cupy.zeros((6,), dtype=numpy.float32)
>>> i = cupy.array([1, 0, 1])
>>> v = cupy.array([1., 1., 1.])
>>> cupyx.scatter_add(a, i, v);
>>> a
array([1., 2., 0., 0., 0., 0.], dtype=float32)

Answer 1

cupyの init.py に

import cupyx as _cupyx

とありますので、

cupyx.scatter_add(a, i, v);

を

cupy._cupyx.scatter_add(a, i, v);

と変えたら実行できますけど、最初に「_」が付いてるのを使うのは行儀が良くないかもしれません

質問者さんが自分で回答に書かれてるように、

import cupyx

を追加する方がいいのではないですかね

Comments

[jbe00214]

ありがとうございます。環境を整えてもう一度やってみます。

[jbpb0]

https://docs.cupy.dev/en/v7.8.0/reference/generated/cupyx.scatter_add.html
の一番下の「Note」に
「As of v4, this function is moved from cupy package to cupyx package.」
と書かれてますので、やはり下記が必要なのではないですかね

import cupyx

[jbe00214]

そうですね。sample コードにも単にimport文がなかっただけでした。解決できました。ありがとうございました。

Answer 2

import cupyx

このように 単にimport 文がなかっただけなのでしょうか。
別なエラーが出ていますが，未定義エラーではありません。当たり前のような気がします。
cupy マニュアルにあるnamespace 名前空間の意味が，モジュールのインポートとは異なると思うのですが，その点の知識がなかったということかもしれません。
そういうことでしょうか。

Answer 3

cupyx.scatter_add(a, i, v);

このcupyxとはいったいなんでしょうか

Comments

[jbe00214]

それが質問です。上記のコードは質問にあるリンクにあるコード例です。

[y_waiwai]

あなたが書いたコードじゃないですか？

[y_waiwai]

んじゃたんに、cupyの打ち間違いなんでは。
エラーメッセージでは、cupyxは未定義だ、といってますよ

[jbe00214]

コード例はcupyx.scatter_addの公式レファレンスのページにあるサンプルコードです。そもそも，https://docs.cupy.dev/en/stable/upgrade.html　には，cupyx Namespaceのことが書かれています。ですから，コードは正しいと思います。未定義エラーがでないようにするにはどうしたらよいかの質問です。