当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python mxnet.ndarray.NDArray.context用法及代码示例


用法:

property context

阵列的设备上下文。

例子

>>> x = mx.nd.array([1, 2, 3, 4])
>>> x.context
cpu(0)
>>> type(x.context)
<class 'mxnet.context.Context'>
>>> y = mx.nd.zeros((2,3), mx.gpu(0))
>>> y.context
gpu(0)

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.ndarray.NDArray.context。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。