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

用法:

參數：

返回：

返回類型：

例子：