當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。