You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Currently, Burn's repeat method only supports repeating along one dimension at a time. To enhance functionality and align with common deep learning frameworks like PyTorch, we need to implement a new repeat operation that accepts multiple dimensions simultaneously. This will provide the same behavior as PyTorch's repeat.
To avoid confusion and maintain consistency with other Burn methods, we should rename the existing repeat method to repeat_dim, similar to other *_dim methods in the library.
Implementing this multi-dimensional repeat operation is crucial for supporting the Tile ONNX operator, which is commonly used in many user-submitted models. The lack of this functionality has been reported by users, as highlighted in issue #1714.
Tasks:
Implement a new repeat method that accepts multiple dimensions and repeats the tensor along those dimensions simultaneously.
Rename the existing repeat method to repeat_dim for consistency with other *_dim methods.
Update the documentation and examples to reflect the new repeat method and the renamed repeat_dim method.
Add unit tests to verify the correctness of the multi-dimensional repeat operation.
Ensure compatibility with the Tile ONNX operator and test with user-submitted models that require this functionality.
By completing these tasks, we will provide a more comprehensive and user-friendly repeat operation in Burn, enabling users to easily repeat tensors along multiple dimensions and improving compatibility with ONNX models.
The text was updated successfully, but these errors were encountered:
Currently, Burn's
repeat
method only supports repeating along one dimension at a time. To enhance functionality and align with common deep learning frameworks like PyTorch, we need to implement a new repeat operation that accepts multiple dimensions simultaneously. This will provide the same behavior as PyTorch's repeat.To avoid confusion and maintain consistency with other Burn methods, we should rename the existing
repeat
method torepeat_dim
, similar to other*_dim
methods in the library.Implementing this multi-dimensional repeat operation is crucial for supporting the Tile ONNX operator, which is commonly used in many user-submitted models. The lack of this functionality has been reported by users, as highlighted in issue #1714.
Tasks:
repeat
method that accepts multiple dimensions and repeats the tensor along those dimensions simultaneously.repeat
method torepeat_dim
for consistency with other*_dim
methods.repeat
method and the renamedrepeat_dim
method.By completing these tasks, we will provide a more comprehensive and user-friendly repeat operation in Burn, enabling users to easily repeat tensors along multiple dimensions and improving compatibility with ONNX models.
The text was updated successfully, but these errors were encountered: