Python mxnet.ndarray.op.linalg_syrk用法及代码示例

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

参数：

A：(NDArray) - 输入矩阵的张量
transpose：(boolean, optional, default=0) - 使用输入矩阵的转置。
alpha：(double, optional, default=1) - 应用于结果的标量因子。
out：(NDArray, optional) - 输出 NDArray 来保存结果。

out：- 此函数的输出。

返回类型：

NDArray 或 NDArray 列表

矩阵与其转置的乘法。输入是一个张量 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.ndarray.op.linalg_syrk。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。

用法:

参数：

返回：

返回类型：

注意：