There’s a built-in Embedding layer in Flux and an updated implementation at https://github.com/FluxML/Flux.jl/pull/1656. You can use directly or as a reference for tutorial purposes.
There’s a built-in Embedding layer in Flux and an updated implementation at https://github.com/FluxML/Flux.jl/pull/1656. You can use directly or as a reference for tutorial purposes.