Is there anything like vmap to vectorize a computation

[Leander]

according to this:

How to do mapslices() in parallel for 3D arrays

Update my solutions here. Shorter but not efficient enough solution. This solution aligns each 3D array layer in 2D array, then computes the result of 2D array multiplication and last converts back to 3D array. using CUDA A = CUDA.rand(4, 4, 3) B = CUDA.rand(4, 4) C = reshape(permutedims(A, [1, 3, 2]), size(A, 1) * size(A, 3), * B C = permutedims(reshape(C, size(A, 1), size(A, 3), :), [1, 3, 2]) Longer but more efficient solution. Use for loop to get each entry of the final 3D array. Thi…

this is not really working afaik