class_docstr: Domain class describing a rectangular domain
  
  Attributes:
      size (Tuple[int]): The size of the domain in absolute units.
      dx (Tuple(float)): The unit of measure
class_docstr_tok: ["Domain","class","describing","a","rectangular","domain","Attributes",":","size","(","Tuple","[","int","]",")",":","The","size","of","the","domain","in","absolute","units",".","dx","(","Tuple","(","float",")",")",":","The","unit","of","measure"]
docstr: A grid of spatial position, of size
  `Nx x Ny x Nz x ... x num_axis` such that the element
  `[x1,x2,x3, .., :]` is a coordinate vector.
  
  Returns:
      jnp.array: A grid of spatial position
docstr_tok: ["A","grid","of","spatial","position",",","of","size","`","Nx","x","Ny","x","Nz","x","...","x","num_axis","`","such","that","the","element","`","[","x1",",","x2",",","x3",",","..",",",":","]","`","is","a","coordinate","vector",".","Returns",":","jnp.array",":","A","grid","of","spatial","position"]
code: def grid(self):
      """A grid of spatial position, of size
      `Nx x Ny x Nz x ... x num_axis` such that the element
      `[x1,x2,x3, .., :]` is a coordinate vector.
  
      Returns:
          jnp.array: A grid of spatial position
  
      """
      axis = self.spatial_axis
      return self._make_grid_from_axis(axis)
  
file_path: jaxdf/jaxdf/geometry.py
imports: from functools import reduce
  from typing import Iterable
  import equinox
  import jax
  from jax import numpy
  from jax import random
  from .mods import Module
label_desc: Use image node_id 9 for calling the Domain obj's underlying member method code with example usage: obj.grid() and returns: self