code: def test_ndarrays_transmute_ops():
      u = hl.utils.range_table(10, n_partitions=10)
      u = u.annotate(x=hl.nd.array([u.idx]), y=hl.nd.array([u.idx]))
      u = u.transmute(xxx=u.x @ u.y)
      assert u.xxx.collect() == [x * x for x in range(10)]
  
imports: import math
  import numpy
  import re
  from ..helpers import *
  import pytest
  from hail.utils.java import FatalError, HailUserError
label_desc: Use image node_id 30 for calling a global function with example usage: test_ndarrays_transmute_ops() without return types