mindspore.ops.ger

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

计算输入一维Tensor inputvec2 的外积。如果 input shape为 \((m,)\)vec2 shape为 \((n,)\) ,则输出的shape为 \((m, n)\)

说明

Ascend不支持float64数据格式的输入。

参数:
  • input (Tensor) - 输入1-D Tensor,数据类型为float16、float32或float64。

  • vec2 (Tensor) - 输入1-D Tensor,数据类型为float16、float32或float64,输入数据类型需和 input 保持一致。

返回:

Tensor,与 input 数据类型相同的输出Tensor。如果 input shape为 \((m,)\)vec2 shape为 \((n,)\) ,则输出的shape为 \((m, n)\)

异常:
  • TypeError - inputvec2 不是一维Tensor。

  • TypeError - 输入的 inputvec2 数据类型不是float16、float32或float64。

  • TypeError - 输入的 inputvec2 数据类型不一致。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> from mindspore import Tensor, ops
>>> input = Tensor([1., 2., 3., 4.], mindspore.float32)
>>> vec2 = Tensor([1., 2., 3.], mindspore.float32)
>>> output = ops.ger(input, vec2)
>>> print(output)
[[ 1.  2.  3.]
 [ 2.  4.  6.]
 [ 3.  6.  9.]
 [ 4.  8. 12.]]