Python mxnet.symbol.op.linalg_syrk用法及代碼示例

用法:
mxnet.symbol.op.linalg_syrk(A=None, transpose=_Null, alpha=_Null, name=None, attr=None, out=None, **kwargs)

參數：

A：(Symbol) - 輸入矩陣的張量
transpose：(boolean, optional, default=0) - 使用輸入矩陣的轉置。
alpha：(double, optional, default=1) - 應用於結果的標量因子。
name：(string, optional.) - 結果符號的名稱。

結果符號。

返回類型：

Symbol

矩陣與其轉置的乘法。輸入是一個張量 A 的維度 n >= 2 。

如果 n=2 ，則操作符執行 BLAS3 函數 syrk ：

out = alpha * A * A^T

如果 transpose=False ，或

out = alpha * A^T * A

如果 transpose=True 。

如果 n>2 ， syrk 對所有輸入的尾隨兩個維度分別執行(批處理模式)。

注意：

該運算符僅支持 float32 和 float64 數據類型。

例子：

Single matrix multiply
A = [[1., 2., 3.], [4., 5., 6.]]
syrk(A, alpha=1., transpose=False)
         = [[14., 32.],
            [32., 77.]]
syrk(A, alpha=1., transpose=True)
         = [[17., 22., 27.],
            [22., 29., 36.],
            [27., 36., 45.]]

Batch matrix multiply
A = [[[1., 1.]], [[0.1, 0.1]]]
syrk(A, alpha=2., transpose=False) = [[[4.]], [[0.04]]]

相關用法

注：本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.symbol.op.linalg_syrk。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

返回類型：

注意：