Skip to content

Add dim parameter to LabelTensor.stack method #770

@GiovanniCanali

Description

@GiovanniCanali

Is your feature request related to a problem? Please describe.
The current implementation of the stack method in the LabelTensor class does not allow users to specify the dimension along which tensors are stacked. Since the method internally relies on torch.stack, it would be beneficial to expose this parameter to provide greater flexibility and consistency with the underlying PyTorch API.

Note: An additional check may be necessary. To preserve label integrity, the newly introduced dimension should likely not be the last one.

Metadata

Metadata

Labels

0.3Related to 0.3 releaseenhancementNew feature or request

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions