Your map
function needs to be scalar, i.e., x ^ 2
. It’s not allowed to use other vectorized operations as operators with GPU array abstractions (as a consequence, functions like eachslices
aren’t supported).
1 Like