jax.numpy.promote_types

jax.numpy.promote_types#

jax.numpy.promote_types(a, b)[source]#

返回二元运算应将其参数转换为的类型。

有关 JAX 类型提升语义的详细信息，请参阅 类型提升语义。

参数：

  • a (DTypeLike) – a numpy.dtype 或 dtype 说明符。

  • b (DTypeLike) – a numpy.dtype 或 dtype 说明符。

返回值：

A numpy.dtype 对象。

返回类型：

DType

内容