code: def transport_plan(K, u, v):
      i_dim = K.shape[0]
      j_dim = K.shape[1]
      result = np.empty_like(K)
      for i in numba.prange(i_dim):
          for j in range(j_dim):
              result[i, j] = u[i] * K[i, j] * v[j]
  
      return result
  
imports: import numpy
  import numba
  from collections import namedtuple
  from enum import Enum, IntEnum
label_desc: Use image node_id 19 for calling a global function with example usage: transport_plan(K, u, v) and returns: result