Junru Shao created MXNET-626:
--------------------------------
Summary: Add while_loop
Key: MXNET-626
URL: https://issues.apache.org/jira/browse/MXNET-626
Project: Apache MXNet
Issue Type: Improvement
Reporter: Junru Shao
This PR adds a control flow operator: while_loop. It takes a Python function as
input and run the function over the elements in the input array. while_loop is
similar to while_loop in TensorFlow.
This PR is part of the proposal of adding a set of control flow operators to
MXNet.
[https://cwiki.apache.org/confluence/display/MXNET/Optimize+dynamic+neural+network+models+with+control+flow+operators]
See also MXNET-432
--
This message was sent by Atlassian JIRA
(v7.6.3#76005)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]