burn icon indicating copy to clipboard operation
burn copied to clipboard

Feature Request: Add Bitwise Operations for Tensors

Open wcshds opened this issue 1 year ago • 2 comments

Bitwise operations are very common, and most deep learning frameworks support these operations. However, Burn seems to lack support for them.

For me, if bitwise operations were available, I could easily implement NumPy's packbits and unpackbits functions in Burn. These functions would help me save 87.5% of the space when storing tensors in one-hot or multi-hot representations.

wcshds avatar Sep 01 '24 17:09 wcshds

is it okay if i work on this ? i'm new to open source

aybdee avatar Oct 28 '24 08:10 aybdee

i'm new to open source

@aybdee Check out the Contributing file and Adding a New Operation in the book. You can open PR (or draft PR) if you have decided to contribute. Link an issue if there is any (see this to link issues). Members will review if it is ready.

AsherJingkongChen avatar Nov 02 '24 18:11 AsherJingkongChen

I believe this is already implemented as of burn 0.17

antimora avatar May 05 '25 03:05 antimora