mindspore.mint.outer

查看源文件
mindspore.mint.outer(input, vec2)[源代码]

计算 inputvec2 的外积。如果向量 input 长度为 \(n\)vec2 长度为 \(m\) ,则输出矩阵shape为 \((n, m)\)

警告

这是一个实验性API,后续可能修改或删除。

说明

该函数不支持广播。

参数:
  • input (Tensor) - 输入一维向量。

  • vec2 (Tensor) - 输入一维向量。

返回:

out,两个一维向量的外积,是一个二维矩阵。

异常:
  • TypeError - 如果 inputvec2 不是Tensor。

  • TypeError - inputvec2 隐式转换后的数据类型不是float16、float32、float64、bool、uint8、int8、int16、int32、int64、complex64、complex128、bfloat16之一

  • ValueError - 如果 inputvec2 的维度不是1。

支持平台:

Ascend

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor
>>> from mindspore import mint
>>> input = Tensor(np.array([7, 8, 9]), mindspore.int32)
>>> vec2 = Tensor(np.array([7, 10, 11]), mindspore.int32)
>>> out = mint.outer(input, vec2)
>>> print(out)
[[49 70 77]
 [56 80 88]
 [63 90 99]]