Python mxnet.ndarray.contrib.cond用法及代码示例

用法: mxnet.ndarray.contrib.cond(pred, then_func, else_func)

参数：

pred：(a MXNet NDArray representing a scalar.) - 分支条件。
then_func：(a Python function.) - 要执行的计算 ifpred是真的。
else_func：(a Python function.) - 要执行的计算 ifpred是假的。

outputs：

返回类型：

NDArray 或 NDArray 的嵌套列表，表示计算结果。

使用用户定义的条件和计算运行if-then-else

此运算符模拟if-like 分支，该分支根据指定条件选择执行两个自定义计算之一。

pred 是一个标量 MXNet NDArray，指示应该使用哪个计算分支。

then_func 是一个用户定义的函数，用作 then 分支的计算。它产生 outputs ，这是一个 NDArrays 列表。 then_func 的签名应该是 then_func() => NDArray or nested List[NDArray] 。

else_func 是用户定义的函数，用作else 分支的计算。它产生 outputs ，这是一个 NDArrays 列表。 else_func 的签名应该是 else_func() => NDArray or nested List[NDArray] 。

由then_func 和else_func 生成的outputs 应具有相同数量的元素，所有元素应具有相同的形状、相同的dtype 和stype。

该函数返回一个符号列表，表示计算结果。

例子：

>>> a, b = mx.nd.array([1]), mx.nd.array([2])
>>> pred = a * b < 5
>>> then_func = lambda: (a + 5) * (b + 5)
>>> else_func = lambda: (a - 5) * (b - 5)
>>> outputs = mx.nd.contrib.cond(pred, then_func, else_func)
>>> outputs[0]
[42.]
<NDArray 1 @cpu(0)>

相关用法

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

用法:

参数：

返回：

返回类型：

例子：