Skip to content

Support Split even when dimension is not evenly splittable #4438

@xuzijian629

Description

@xuzijian629

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?

Metadata

Metadata

Assignees

No one assigned

    Labels

    topic: operatorIssues related to ONNX operators

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions