I notice that Nd4j#scatterUpdate is deprecated but it’s still used in Embedding Layer.
I’ve been trying to replicate the instant-nerf architecture with a custom embedding layer and my performance bottleneck is there. I’m wondering if there’s a replacement and possibly faster method to achieve the same functionality.