[GitHub] Roshrini opened a new issue #11594: [FeatureRequest]MXNet Split operator to support variable length splits

[GitBox <gi...@apache.org>]

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:
users@infra.apache.org


With regards,
Apache Git Services