1 2 3 4 5 6 7 8 9 10 11
extend("gpu") X = gpu:to_tensor([ [1,2,3,4], [5,6,7,8], [9,10,11,12], [13,14,15,16] ]) total = gpu:reduce_sum(X) slog(gpu:to_array(total))