diff --git a/doc/fluid/api/tensor.rst b/doc/fluid/api/tensor.rst index 84b1501a1ed928960447bee4886d4c32be80b45d..73ce3e9d698fc54bd2ee5ba6591de1e6c8d695c4 100644 --- a/doc/fluid/api/tensor.rst +++ b/doc/fluid/api/tensor.rst @@ -20,6 +20,7 @@ paddle.tensor tensor/cos.rst tensor/create_tensor.rst tensor/crop_tensor.rst + tensor/cross.rst tensor/cumsum.rst tensor/diag.rst tensor/div.rst diff --git a/doc/fluid/api/tensor/cross.rst b/doc/fluid/api/tensor/cross.rst new file mode 100644 index 0000000000000000000000000000000000000000..3bb049f74d7232bd42020fee1b702c313395ba85 --- /dev/null +++ b/doc/fluid/api/tensor/cross.rst @@ -0,0 +1,7 @@ +.. _api_tensor_cn_cos: + +cross +------------------------------- +:doc_source: paddle.tensor.cross + + diff --git a/doc/fluid/api_cn/tensor_cn/cross_cn.rst b/doc/fluid/api_cn/tensor_cn/cross_cn.rst index 9541c6ef690a97e2f637a5cfc5d09be4990859ff..4bfb4ed980490ece02796281afd9a7b527367ac7 100644 --- a/doc/fluid/api_cn/tensor_cn/cross_cn.rst +++ b/doc/fluid/api_cn/tensor_cn/cross_cn.rst @@ -3,51 +3,55 @@ cross ------------------------------- -.. py:function:: paddle.cross(input, other, dim=None) +.. py:function:: paddle.cross(x, y, axis=None, name=None) :alias_main: paddle.cross :alias: paddle.cross,paddle.tensor.cross,paddle.tensor.linalg.cross -该OP返回在 ``dim`` 维度上,两个张量 ``input`` 和 ``other`` 的向量积(叉积)。 ``input`` 和 ``other`` 必须有相同的形状, -且指定的 ``dim`` 维上 ``size`` 必须为3,如果 ``dim`` 未指定,默认选取第一个 ``size`` 等于3的维度。 +计算张量 ``x`` 和 ``y`` 在 ``axis`` 维度上的向量积(叉积)。 ``x`` 和 ``y`` 必须有相同的形状, +且指定的 ``axis`` 的长度必须为3. 如果未指定 ``axis`` ,默认选取第一个长度为3的 ``axis`` . **参数**: - - **input** (Variable)– 第一个输入张量。 - - **other** (Variable)– 第二个输入张量。 - - **dim** (int, optional) – 沿着此维进行叉积操作,若未指定,则默认选取第一个 ``size`` 等于3的维度 + - **x** (Variable)– 第一个输入张量。 + - **y** (Variable)– 第二个输入张量。 + - **axis** (int, optional) – 沿着此维进行向量积操作。默认选取第一个长度为3的 ``axis`` . + - **name** (str,可选)- 输出的名字。默认值为None。该参数供开发人员打印调试信息时使用,具体用法请参见 :ref:`api_guide_Name` 。 + +**返回**:向量积的结果。 + +**返回类型**:Variable -**返回**: - - **Variable** ,数据类型同输入。 - **代码示例**: .. code-block:: python import paddle - import paddle.fluid as fluid + from paddle.imperative import to_variable import numpy as np + paddle.enable_imperative() + data_x = np.array([[1.0, 1.0, 1.0], [2.0, 2.0, 2.0], [3.0, 3.0, 3.0]]) data_y = np.array([[1.0, 1.0, 1.0], [1.0, 1.0, 1.0], [1.0, 1.0, 1.0]]) - - with fluid.dygraph.guard(): - x = fluid.dygraph.to_variable(data_x) - y = fluid.dygraph.to_variable(data_y) - out_z1 = paddle.cross(x, y) - print(out_z1.numpy()) - #[[-1. -1. -1.] - # [ 2. 2. 2.] - # [-1. -1. -1.]] - out_z2 = paddle.cross(x, y, dim=1) - print(out_z2.numpy()) - #[[0. 0. 0.] - # [0. 0. 0.] - # [0. 0. 0.]] + x = to_variable(data_x) + y = to_variable(data_y) + + z1 = paddle.cross(x, y) + print(z1.numpy()) + # [[-1. -1. -1.] + # [ 2. 2. 2.] + # [-1. -1. -1.]] + + z2 = paddle.cross(x, y, axis=1) + print(z2.numpy()) + # [[0. 0. 0.] + # [0. 0. 0.] + # [0. 0. 0.]]