Scalar indexing GPU problem in Flux.jl model

using cuDNN is missing

2 Likes