Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[FeatureRequest]MXNet Split operator to support variable length splits #11594

Open
Roshrini opened this issue Jul 6, 2018 · 2 comments
Open

Comments

@Roshrini
Copy link
Member

Roshrini commented Jul 6, 2018

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)>]
@nswamy
Copy link
Member

nswamy commented Jul 19, 2018

Closing as the PR fixes.

@nswamy nswamy closed this as completed Jul 19, 2018
@MoritzMaxeiner
Copy link
Contributor

@nswamy Why was this feature request closed? The requested feature is not available in 1.2 release or master AFAICT - I could not find any PR fixing this.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

No branches or pull requests

4 participants