mumu 0.10.0

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,0,0,0],
  [0,1,0,0],
  [0,0,1,0],
  [0,0,0,1]
])

C = gpu:multiply(A, B)
slog(gpu:to_array(C))