1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
extend("gpu") A = gpu:to_tensor([ [1,2,3,4], [5,6,7,8], [9,10,11,12], [13,14,15,16] ]) B = gpu:to_tensor([ [1,1,1,1], [2,2,2,2], [3,3,3,3], [4,4,4,4] ]) C = gpu:add(A, B) slog(gpu:to_array(C))