Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Support `Split` even when dimension is not evenly splittable

See original GitHub issue

Currently, Split operator splits the tensor to equal sized tensors if split input is not given.

I think this restriction is not very important. ~For example, pytorch’s split also supports the case when dim cannot be evenly split. https://pytorch.org/docs/1.12/generated/torch.split.html?highlight=split#torch.split~

UPD: Torch’s corresponding function was https://pytorch.org/docs/stable/generated/torch.tensor_split.html

Can we update the operator definition to support this case?

Issue Analytics

State:
Created a year ago
Comments:6 (6 by maintainers)

Top GitHub Comments

1reaction

p-wysockicommented, Oct 4, 2022

The PR has been merged. Please reopen the issue if needed.

1reaction

p-wysockicommented, Sep 1, 2022

PR: https://github.com/onnx/onnx/pull/4481

Top Results From Across the Web

Native batch ingestion - Apache Druid

Druid does not split files across tasks. One subtask will not process more files than maxNumFiles even when their total size is smaller...

Split a Field into Multiple Fields - Tableau Help

Not all data sources support SPLIT. You can tell if your data supports the SPLIT function by checking for the Split and Custom...

SPLIT_TO_TABLE - Snowflake Documentation

A unique sequence number associated with the input record; the sequence is not guaranteed to be gap-free or ordered in any particular way....

Optimizing Data-Intensive Computations in Existing Libraries ...

We introduce split annotations, a new technique for en- abling data movement optimization and automatic paral- lelization with no library modifications. 2. We ......

Shape inference fails with a Split node with an split attribute

When a split attribute is set to a Split node, ... This will happen even if the tensor a is divided evenly (by...