Contrib

拡張コンテキスト

nnabla.contrib.context.extension_context(extension_name='cpu', **kw)

指定した拡張機能のコンテキストを取得できます。

すべての拡張モジュールにおいて、 context(**kw) 関数を提供する必要があります。

パラメータ:

• extension_name (str) -- nnabla_ext に関連するモジュールパス。

• kw (dict) -- 拡張モジュールにおける context 関数の追加キーワード引数。

戻り値:

現在の拡張機能コンテキスト。

戻り値の型:

nnabla.Context

注釈

非推奨です。代わりに、 :function:`nnabla.ext_utils.get_extension_context` を使用してください。

例

ctx = extension_context('cuda.cudnn', device_id=0)
nn.set_default_context(ctx)