shapes
================

.. py:module:: jaxns.internals.shapes

.. rubric:: :code:`jaxns.internals.shapes`


.. rubric:: Module Contents

.. py:function:: broadcast_dtypes(*dtypes)

   Returns the dtype with highest precision.

   :param \*dtypes: list of JAX dtypes.

   Returns: dtype


.. py:function:: convert_to_array(v)

   If necessary convert v to a jax.Array.
   Passes through Prior.

   :param v: array-like or scalar

   Returns: jax.Array


.. py:function:: tuple_prod(t)

   Product of shape tuple

   :param t: tuple

   :returns: int


.. py:function:: broadcast_shapes(shape1, shape2)

   Broadcasts two shapes together.

   :param shape1: tuple of int
   :param shape2: tuple of int

   Returns: tuple of int with resulting shape.