jax.numpy.promote_types

jax.numpy.promote_types#

jax.numpy.promote_types(a, b)[源代码][源代码]#

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

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

参数:

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

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

返回:

一个 numpy.dtype 对象。

返回类型:

DType

目录