Skip to content

✨[Converter] Implement aten::chunk in FX and torchscript #1801

Closed
@apbose

Description

@apbose

Torch op : chunk(Tensor(a -> *) self, int chunks, int dim=0) -> Tensor(a)[]
Aten op: torch.ops.aten.chunk.Tensor
Pytorch implementation:
https://pytorch.org/docs/stable/generated/torch.chunk.html

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions