def cond(flag, value_if_true, value_if_false, name=None):
"""
return either value_if_true or value_if_false based on the value of flag.
If flag != 0 value_if_true is returned, otherwise value_if_false.
Behaves analogously to numpy.where(...).
Example:
>>> C.eval(C.cond([-10, -1, 0, 0.3, 100], [1, 10, 100, 1000, 10000], [ 2, 20, 200, 2000, 20000]))
[array([[ 1.00000000e+00, 1.00000000e+01, 2.00000000e+02,
1.00000000e+03, 1.00000000e+04]])]
Args:
flag: tensor
value_if_true: tensor
value_if_false: tensor
name (str): the name of the node in the network
Returns:
:class:`cntk.graph.ComputationNode`
"""
from cntk.ops.cntk1 import If
op = If(flag, value_if_true, value_if_false, name=name)
wrap_numpy_arrays(op)
op.rank = max(op.cond.rank, op.thenVal.rank, op.elseVal.rank)
return op
def cond(flag, value_if_true, value_if_false, name=None):
"""
return either value_if_true or value_if_false based on the value of flag.
If flag != 0 value_if_true is returned, otherwise value_if_false.
Behaves analogously to numpy.where(...).
Example:
>>> C.eval(C.cond([-10, -1, 0, 0.3, 100], [1, 10, 100, 1000, 10000], [ 2, 20, 200, 2000, 20000]))
[array([[ 1.00000000e+00, 1.00000000e+01, 2.00000000e+02,
1.00000000e+03, 1.00000000e+04]])]
Args:
flag: tensor
value_if_true: tensor
value_if_false: tensor
name (str): the name of the node in the network
Returns:
:class:`cntk.graph.ComputationNode`
"""
from cntk.ops.cntk1 import If
op = If(flag, value_if_true, value_if_false, name=name)
wrap_numpy_arrays(op)
op.rank = max(op.cond.rank, op.thenVal.rank, op.elseVal.rank)
return op
def cond(flag, value_if_true, value_if_false, name=None):
"""
Return either value_if_true or value_if_false based on the value of flag.
If flag != 0 value_if_true is returned, otherwise value_if_false.
Behaves analogously to numpy.where(...).
Example:
>>> cond([-10, -1, 0, 0.3, 100], [1, 10, 100, 1000, 10000], [ 2, 20, 200, 2000, 20000])
# [1, 10, 200, 1000, 10000]
Args:
flag: tensor
value_if_true: tensor
value_if_false: tensor
name: the name of the node in the network
Returns:
:class:`cntk.graph.ComputationNode`
"""
return If(flag, value_if_true, value_if_false, var_name = name)