Contrib¶
拡張コンテキスト¶
- nnabla.contrib.context.extension_context(extension_name='cpu', **kw)[ソース]¶
指定した拡張機能のコンテキストを取得できます。
すべての拡張モジュールにおいて、
context(**kw)
関数を提供する必要があります。- パラメータ
- 戻り値
現在の拡張機能コンテキスト。
- 戻り値の型
注釈
非推奨です。代わりに、 :function:`nnabla.ext_utils.get_extension_context` を使用してください。
例
ctx = extension_context('cuda.cudnn', device_id=0) nn.set_default_context(ctx)