jax.numpy.result_type#
- jax.numpy.result_type(*args)[source]#
返回应用 NumPy
LAX 后端实现的
numpy.result_type()的类型。原始文档字符串如下。
类型提升规则到参数。
NumPy 中的类型提升与 C++ 等语言中的规则类似，但也有一些细微差别。当同时使用标量和数组时，数组的类型优先，并且会考虑标量的实际值。
例如，计算 3*a，其中 a 是一个 32 位浮点数数组，直观上应该得到一个 32 位浮点数输出。如果 3 是一个 32 位整数，则 NumPy 规则表明它不能无损转换为 32 位浮点数，因此结果类型应该为 64 位浮点数。通过检查常量“3”的值，我们发现它适合 8 位整数，可以无损转换为 32 位浮点数。
- 返回:
out – 结果类型。
- 返回类型:
- 参数:
args (Any)