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

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


用法:

as_in_context(context)

参数：

context：(mxnet.context.Context) - 目标上下文。

返回：

目标数组。

返回类型：

NDArray、CSRNDArray或RowSparseNDArray

在目标设备上返回一个与此数组具有相同值的数组。

如果目标上下文与 self.context 相同，则返回 self。否则，将制作一份副本。

例子：

>>> x = mx.nd.ones((2,3))
>>> y = x.as_in_context(mx.cpu())
>>> y is x
True
>>> z = x.as_in_context(mx.gpu(0))
>>> z is x
False

相关用法


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