cross

paddle.fluid.layers.cross(input, other, dim=None)[源代码]

该OP返回在 dim 维度上，两个张量 input 和 other 的向量积（叉积）。 input 和 other 必须有相同的形状， 且指定的 dim 维上 size 必须为3，如果 dim 未指定，默认选取第一个 size 等于3的维度。

参数

  • input （Variable）– 第一个输入张量。
  • other （Variable）– 第二个输入张量。
  • dim (int, optional) – 沿着此维进行叉积操作，若未指定，则默认选取第一个 size 等于3的维度

返回

  • Variable ，数据类型同输入。

代码示例

  1. import paddle
  2. import paddle.fluid as fluid
  3. import numpy as np
  5. data_x = np.array([[1.0, 1.0, 1.0],
  6.                    [2.0, 2.0, 2.0],
  7.                    [3.0, 3.0, 3.0]])
  8. data_y = np.array([[1.0, 1.0, 1.0],
  9.                    [1.0, 1.0, 1.0],
  10.                    [1.0, 1.0, 1.0]])
  12. with fluid.dygraph.guard():
  13.     x = fluid.dygraph.to_variable(data_x)
  14.     y = fluid.dygraph.to_variable(data_y)
  15.     out_z1 = fluid.layers.cross(x, y)
  16.     print(out_z1.numpy())
  17.     #[[-1. -1. -1.]
  18.     # [ 2.  2.  2.]
  19.     # [-1. -1. -1.]]
  20.     out_z2 = fluid.layers.cross(x, y, dim=1)
  21.     print(out_z2.numpy())
  22.     #[[0. 0. 0.]
  23.     # [0. 0. 0.]
  24.     # [0. 0. 0.]]