Marko Petkovic

scatter_max returns 0s on GPU

5

Hi, When I am using the scatter_max function on cpu, it works as expected. However, when moving all tensors and models to GPU, the functions stops working, return 0 for...