Roshrini opened a new issue #11594: [FeatureRequest]MXNet Split operator to 
support variable length splits
URL: https://github.com/apache/incubator-mxnet/issues/11594
 
 
   ## Description
   MXNet's split operator only support equal splits of a tensor according to 
http://mxnet.incubator.apache.org/api/python/symbol/symbol.html#mxnet.symbol.split
   
   This issue is to request the support for variable length splits.
   For example:
   ```
   input = mx.nd.array([1., 2., 3., 4., 5., 6.])
   mx.nd.split(input, axis=0, num_outputs=[2,4]) # mayebe accepting num_outputs 
as a list 
   expected_output = [
    [1. 2.]
    <NDArray 2 @cpu(0)>, 
   [ [3. 4. 5. 6.]
    <NDArray 4 @cpu(0)>]
   ```
   

----------------------------------------------------------------
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:
[email protected]


With regards,
Apache Git Services

Reply via email to