szha commented on a change in pull request #12209: Make the output format of
control flow operators consistent with their UDFs
URL: https://github.com/apache/incubator-mxnet/pull/12209#discussion_r211483059
##########
File path: python/mxnet/ndarray/contrib.py
##########
@@ -135,7 +167,7 @@ def foreach(body, data, init_states):
Define computation in an iteration.
data: an NDArray or a list of NDArrays.
The input data.
- init_states: an NDArray or a list of NDArrays.
+ init_states: a list of NDArrays.
Review comment:
can we use the flatten regroup trick to relax this constraint?
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
