mlx.core.addmm

目录

mlx.core.addmm#

addmm(c: array, a: array, b: array, /, alpha: float = 1.0, beta: float = 1.0, *, stream: None | Stream | Device = None) array#

矩阵乘法并相加,可选择缩放。

执行两个数组的(可能带批次的)矩阵乘法,并将其与结果相加,可选择缩放因子。

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

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

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

  • alpha (float, 可选) – 数组 ab 矩阵乘积的缩放因子(默认值:1

  • beta (float, 可选) – c 的缩放因子(默认值:1

返回:

alpha * (a @ b)  + beta * c

返回类型:

array