code: def eval_sh(deg, sh, dirs):
      """
      Evaluate spherical harmonics at unit directions
      using hardcoded SH polynomials.
      Works with torch/np/jnp.
      ... Can be 0 or more batch dimensions.
      Args:
          deg: int SH deg. Currently, 0-3 supported
          sh: jnp.ndarray SH coeffs [..., C, (deg + 1) ** 2]
          dirs: jnp.ndarray unit directions [..., 3]
      Returns:
          [..., C]
      """
      assert deg <= 4 and deg >= 0
      coeff = (deg + 1) ** 2
      assert sh.shape[-1] >= coeff
  
      result = C0 * sh[..., 0]
      if deg > 0:
          x, y, z = dirs[..., 0:1], dirs[..., 1:2], dirs[..., 2:3]
          result = (
              result
              - C1 * y * sh[..., 1]
              + C1 * z * sh[..., 2]
              - C1 * x * sh[..., 3]
          )
  
          if deg > 1:
              xx, yy, zz = x * x, y * y, z * z
              xy, yz, xz = x * y, y * z, x * z
              result = (
                  result
                  + C2[0] * xy * sh[..., 4]
                  + C2[1] * yz * sh[..., 5]
                  + C2[2] * (2.0 * zz - xx - yy) * sh[..., 6]
                  + C2[3] * xz * sh[..., 7]
                  + C2[4] * (xx - yy) * sh[..., 8]
              )
  
              if deg > 2:
                  result = (
                      result
                      + C3[0] * y * (3 * xx - yy) * sh[..., 9]
                      + C3[1] * xy * z * sh[..., 10]
                      + C3[2] * y * (4 * zz - xx - yy) * sh[..., 11]
                      + C3[3]
                      * z
                      * (2 * zz - 3 * xx - 3 * yy)
                      * sh[..., 12]
                      + C3[4] * x * (4 * zz - xx - yy) * sh[..., 13]
                      + C3[5] * z * (xx - yy) * sh[..., 14]
                      + C3[6] * x * (xx - 3 * yy) * sh[..., 15]
                  )
  
                  if deg > 3:
                      result = (
                          result
                          + C4[0] * xy * (xx - yy) * sh[..., 16]
                          + C4[1] * yz * (3 * xx - yy) * sh[..., 17]
                          + C4[2] * xy * (7 * zz - 1) * sh[..., 18]
                          + C4[3] * yz * (7 * zz - 3) * sh[..., 19]
                          + C4[4]
                          * (zz * (35 * zz - 30) + 3)
                          * sh[..., 20]
                          + C4[5] * xz * (7 * zz - 3) * sh[..., 21]
                          + C4[6]
                          * (xx - yy)
                          * (7 * zz - 1)
                          * sh[..., 22]
                          + C4[7] * xz * (xx - 3 * yy) * sh[..., 23]
                          + C4[8]
                          * (xx * (xx - 3 * yy) - yy * (3 * xx - yy))
                          * sh[..., 24]
                      )
      return result
  
imports: import torch
label_desc: Use image node_id 1 for calling a global function with example usage: eval_sh(deg, sh, dirs) and returns: result