mindspore.ops.outer
- mindspore.ops.outer(input, vec2)[源代码]
计算 input 和 vec2 的外积。如果向量 input 长度为 \(n\) , vec2 长度为 \(m\) ,则输出矩阵shape为 \((n, m)\) 。
说明
该函数不支持广播。
- 参数:
input (Tensor) - 输入一维向量。
vec2 (Tensor) - 输入一维向量。
- 返回:
out (Tensor, optional),两个一维向量的外积,是一个2维矩阵。
- 异常:
TypeError - 如果 input 或 vec2 不是Tensor。
ValueError - 如果 input 或 vec2 不是一维Tensor。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor >>> from mindspore import ops >>> input = Tensor(np.array([1, 2, 3]), mindspore.int32) >>> vec2 = Tensor(np.array([1, 2, 3]), mindspore.int32) >>> out = ops.outer(input, vec2) >>> print(out) [[1 2 3] [2 4 6] [3 6 9]]