result_type

Contents

result_type#

class saiunit.math.result_type(*args)#

Determine the result dtype from a set of input arrays or dtypes.

Parameters:

*args (array_like or dtype) – Input arrays or dtypes.

Returns:

out – The result dtype that would arise from operating on the inputs.

Return type:

dtype

Examples

>>> import jax.numpy as jnp
>>> import saiunit.math as sumath
>>> sumath.result_type(jnp.float32, jnp.int32)
dtype('float32')
Contents