mlx.core.divmod

目录

mlx.core.divmod#

divmod(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) array#

逐元素的商和余数。

函数 divmod(a, b) 等价于 (a // b, a % b) 但速度更快。该函数使用 numpy 风格的广播语义。输入数组可以是一个或两个都是标量。

参数:
  • a (array) – 输入数组或标量。

  • b (array) – 输入数组或标量。

返回:

a // b 和余数 a % b

返回类型:

tuple(array, array)