code: def _generate_voxel_cube(dims, plane_orientation=None):
      if plane_orientation == "horizontal":
          voxels_no_batch = _generate_voxels_horizontal_plane(dims[-4:])
      elif plane_orientation == "vertical":
          voxels_no_batch = _generate_voxels_vertical_plane(dims[-4:])
      else:
          voxels_no_batch = np.zeros(dims[-4:])
  
      voxels = np.zeros(dims)
      voxels[..., :, :, :, :] = voxels_no_batch
      return voxels
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import numpy
  from six.moves import range
  import tensorflow
  from tensorflow_graphics.geometry.representation import grid
  from tensorflow_graphics.geometry.transformation import rotation_matrix_3d
  from tensorflow_graphics.math.interpolation import trilinear
  from tensorflow_graphics.util import test_case
label_desc: Use image node_id 8 for calling a global function with example usage: _generate_voxel_cube(dims, plane_orientation) and returns: voxels