当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python mxnet.ndarray.linalg.syrk用法及代码示例


用法:

mxnet.ndarray.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 * AT

如果 transpose=False ,或

out = alpha * AT * A

如果 transpose=True

如果 n>2syrk 对所有输入的尾随两个维度分别执行(批处理模式)。

注意

该运算符仅支持 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.linalg.syrk。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。