EnzymeAD / EnzymeAD/Reactant.jl
Bool mask indexing
Open
- Dominant language
- Julia
- Stars
- 370
- Forks
- 74
- Avg merge
- 18h 47m
- Merged PRs (30d)
- 30
Description
I want to implement an operation like this pytorch code
https://github.com/intell-sci-comput/PTS/blob/07088bda3e4c48fe8ac7eded9341855caa02b545/model/models.py#L46-L48
Note: the dr_mask here is a constant bool mask
This is super fast in pytorch, but i can not figure out why Reactant.jl / Julia can not do this.
I know this is not a problem unique to Reactant.jl, but I would like to know what the solution is on Julia. I'm just getting started with Julia. Many thanks!
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.