mindspore.mint.outer
- mindspore.mint.outer(input, vec2)[源代码]
计算 input 和 vec2 的外积。如果向量 input 长度为 \(n\) , vec2 长度为 \(m\) ,则输出矩阵shape为 \((n, m)\) 。
警告
这是一个实验性API,后续可能修改或删除。
说明
该函数不支持广播。
- 参数:
input (Tensor) - 输入一维向量。
vec2 (Tensor) - 输入一维向量。
- 返回:
out,两个一维向量的外积,是一个二维矩阵。
- 异常:
TypeError - 如果 input 或 vec2 不是Tensor。
TypeError - input 和 vec2 隐式转换后的数据类型不是float16、float32、float64、bool、uint8、int8、int16、int32、int64、complex64、complex128、bfloat16之一
ValueError - 如果 input 或 vec2 的维度不是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]]