mindspore.ops.outer

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

计算两个tensor的外积。

如果向量 input 长度为 nvec2 长度为 m ,则输出矩阵shape为 (n,m)

说明

该函数不支持广播。

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

  • vec2 (Tensor) - 一维输入tensor。

返回:

Tensor

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> input = mindspore.tensor([1, 2, 3])
>>> vec2 = mindspore.tensor([4, 5, 6])
>>> mindspore.ops.outer(input, vec2)
Tensor(shape=[3, 3], dtype=Int64, value=
[[ 4,  5,  6],
 [ 8, 10, 12],
 [12, 15, 18]])