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