1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
extend("matrix") extend("gpu") a = [ [1, 2], [3, 4] ] b = [ [5, 6], [7, 8] ] ta = gpu:to_tensor(a) tb = gpu:to_tensor(b) tc = gpu:add(ta, tb) result = gpu:to_array(tc) slog(result)