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, optional) -- ab 矩陣乘積的縮放係數(預設:1

  • beta (float, optional) -- c 的縮放係數(預設:1

回傳:

alpha * (a @ b)  + beta * c

回傳型別:

array