o #jWA@sddlmZmZddlmZmZddlmZddlm Z gZ ej j j ej j jej j jej j jej j jej j jgZedddZedd d Zedd d Zdd dZedddZedddZedddZeddZdS))_C_opsin_dynamic_mode)core dygraph_only) LayerHelper)castNcC t||S)a Note: This API is only supported from ``CUDA 11.0`` . Applies matrix multiplication of two Tensors. The supported input/output Tensor type are as follows: Note: x[SparseCsrTensor] @ y[SparseCsrTensor] -> out[SparseCsrTensor] x[SparseCsrTensor] @ y[DenseTensor] -> out[DenseTensor] x[SparseCooTensor] @ y[SparseCooTensor] -> out[SparseCooTensor] x[SparseCooTensor] @ y[DenseTensor] -> out[DenseTensor] It supports backward propagation. Dimensions `x` and `y` must be >= 2D. Automatic broadcasting of Tensor is not supported. the shape of `x` should be `[*, M, K]` , and the shape of `y` should be `[*, K, N]` , where `*` is zero or more batch dimensions. Args: x (SparseTensor): The input tensor. It can be SparseCooTensor/SparseCsrTensor. The data type can be float32 or float64. y (SparseTensor|DenseTensor): The input tensor. It can be SparseCooTensor/SparseCsrTensor/DenseTensor. The data type can be float32 or float64. name (str, optional): Name for the operation (optional, default is None). For more information, please refer to :ref:`api_guide_Name`. Returns: SparseTensor|DenseTensor: Determined by `x` and `y` . Examples: .. code-block:: python >>> # doctest: +REQUIRES(env:GPU) >>> import paddle >>> paddle.device.set_device('gpu') >>> # csr @ dense -> dense >>> crows = [0, 1, 2, 3] >>> cols = [1, 2, 0] >>> values = [1., 2., 3.] >>> csr = paddle.sparse.sparse_csr_tensor(crows, cols, values, [3, 3]) >>> print(csr) Tensor(shape=[3, 3], dtype=paddle.float32, place=Place(gpu:0), stop_gradient=True, crows=[0, 1, 2, 3], cols=[1, 2, 0], values=[1., 2., 3.]) >>> dense = paddle.ones([3, 2]) >>> out = paddle.sparse.matmul(csr, dense) >>> print(out) Tensor(shape=[3, 2], dtype=float32, place=Place(gpu:0), stop_gradient=True, [[1., 1.], [2., 2.], [3., 3.]]) >>> # coo @ dense -> dense >>> indices = [[0, 1, 2], [1, 2, 0]] >>> values = [1., 2., 3.] >>> coo = paddle.sparse.sparse_coo_tensor(indices, values, [3, 3]) >>> print(coo) Tensor(shape=[3, 3], dtype=paddle.float32, place=Place(gpu:0), stop_gradient=True, indices=[[0, 1, 2], [1, 2, 0]], values=[1., 2., 3.]) >>> dense = paddle.ones([3, 2]) >>> out = paddle.sparse.matmul(coo, dense) >>> print(out) Tensor(shape=[3, 2], dtype=float32, place=Place(gpu:0), stop_gradient=True, [[1., 1.], [2., 2.], [3., 3.]]) )rZ sparse_matmulxynamerU/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/sparse/binary.pymatmul!s KrcCst|||S)a Note: This API is only supported from ``CUDA 11.3`` . Applies matrix multiplication of two Dense Tensors. The supported input/output Tensor layout are as follows: Note: x[DenseTensor] @ y[DenseTensor] * mask[SparseCooTensor] -> out[SparseCooTensor] x[DenseTensor] @ y[DenseTensor] * mask[SparseCsrTensor] -> out[SparseCsrTensor] It supports backward propagation. Dimensions `x` and `y` must be >= 2D. Automatic broadcasting of Tensor is not supported. the shape of `x` should be `[*, M, K]` , and the shape of `y` should be `[*, K, N]` , and the shape of `mask` should be `[*, M, N]` , where `*` is zero or more batch dimensions. Args: x (DenseTensor): The input tensor. It is DenseTensor. The data type can be float32 or float64. y (DenseTensor): The input tensor. It is DenseTensor. The data type can be float32 or float64. mask (SparseTensor): The mask tensor, which can be SparseCooTensor/SparseCsrTensor. It specify sparse coordinates. The data type can be float32 or float64. name (str, optional): Name for the operation (optional, default is None). For more information, please refer to :ref:`api_guide_Name`. Returns: SparseTensor: SparseCooTensor or SparseCsrTensor, which is same with `mask` . Examples: .. code-block:: python >>> # doctest: +REQUIRES(env:GPU) >>> import paddle >>> paddle.device.set_device('gpu') >>> paddle.seed(100) >>> # dense @ dense * csr_mask -> csr >>> crows = [0, 2, 3, 5] >>> cols = [1, 3, 2, 0, 1] >>> values = [1., 2., 3., 4., 5.] >>> dense_shape = [3, 4] >>> mask = paddle.sparse.sparse_csr_tensor(crows, cols, values, dense_shape) >>> print(mask) Tensor(shape=[3, 4], dtype=paddle.float32, place=Place(gpu:0), stop_gradient=True, crows=[0, 2, 3, 5], cols=[1, 3, 2, 0, 1], values=[1., 2., 3., 4., 5.]) >>> x = paddle.rand([3, 5]) >>> y = paddle.rand([5, 4]) >>> out = paddle.sparse.masked_matmul(x, y, mask) >>> print(out) Tensor(shape=[3, 4], dtype=paddle.float32, place=Place(gpu:0), stop_gradient=True, crows=[0, 2, 3, 5], cols=[1, 3, 2, 0, 1], values=[0.98986477, 0.97800624, 1.14591956, 0.68561077, 0.94714981]) )rZsparse_masked_matmul)r r maskr rrr masked_matmulos=rcCr )a Note: This API is only supported from ``CUDA 11.0`` . Applies matrix-vector product of Sparse Matrix 'x' and Dense vector 'vec' . The supported input/output Tensor layout are as follows: Note: x[SparseCsrTensor] @ vec[DenseTensor] -> out[DenseTensor] x[SparseCooTensor] @ vec[DenseTensor] -> out[DenseTensor] It supports backward propagation. The shape of `x` should be `[M, N]` , and the shape of `vec` should be `[N]` , and the shape of `out` will be `[M]` . Args: x (SparseTensor): The input 2D tensor. It must be SparseCooTensor/SparseCsrTensor. The data type can be float32 or float64. vec (DenseTensor): The input 1D tensor. It must be DenseTensor vector. The data type can be float32 or float64. name (str, optional): Name for the operation (optional, default is None). For more information, please refer to :ref:`api_guide_Name`. Returns: DenseTensor: 1D DenseTensor whose dtype is same with input. Examples: .. code-block:: python >>> # doctest: +REQUIRES(env:GPU) >>> import paddle >>> paddle.device.set_device('gpu') >>> paddle.seed(100) >>> # csr @ dense -> dense >>> crows = [0, 2, 3, 5] >>> cols = [1, 3, 2, 0, 1] >>> values = [1., 2., 3., 4., 5.] >>> dense_shape = [3, 4] >>> csr = paddle.sparse.sparse_csr_tensor(crows, cols, values, dense_shape) >>> print(csr) Tensor(shape=[3, 4], dtype=paddle.float32, place=Place(gpu:0), stop_gradient=True, crows=[0, 2, 3, 5], cols=[1, 3, 2, 0, 1], values=[1., 2., 3., 4., 5.]) >>> vec = paddle.randn([4]) >>> out = paddle.sparse.mv(csr, vec) >>> print(out) Tensor(shape=[3], dtype=float32, place=Place(gpu:0), stop_gradient=True, [-3.85499096, -2.42975140, -1.75087738]) )rZ sparse_mv)r Zvecr rrrmvs 7rcCsh|j|jkr t|d|j}trt||Sd}||d}t|}||j}|j||d|iid|S)uK Add two sparse tensors element-wise. Input x and y's shape should be identical and have same sparse type（SparseCooTensor or SparseCsrTensor）.If input is SparseCooTensor, x and y's sparse_dim should be identical. The equation is: .. math:: out = x + y Args: x (Tensor): the input tensor, it's data type should be float32, float64, int32, int64. y (Tensor): the input tensor, it's data type should be float32, float64, int32, int64. name (str, optional): Name for the operation (optional, default is None). For more information, please refer to :ref:`api_guide_Name`. Returns: Tensor: the result tensor. Examples: .. code-block:: python >>> import paddle >>> paddle.device.set_device("cpu") >>> x = paddle.to_tensor([[0, -1, 0, 2], [0, 0, -3, 0], [4, 5, 0, 0]], 'float32') >>> y = paddle.to_tensor([[0, 0, 0, -2], [0, 2, -3, 0], [2, 3, 4, 8]], 'float32') >>> sparse_x = x.to_sparse_csr() >>> sparse_y = y.to_sparse_csr() >>> sparse_z = paddle.sparse.add(sparse_x, sparse_y) >>> print(sparse_z.to_dense()) Tensor(shape=[3, 4], dtype=float32, place=Place(cpu), stop_gradient=True, [[ 0., -1., 0., 0.], [ 0., 2., -6., 0.], [ 6., 8., 4., 8.]]) N sparse_addr r out)typeinputsZoutputsattrs)dtyperrrrrZ)create_sparse_variable_for_type_inferenceZ append_op)r r r Zop_typerhelperrrrradds %    rcCs(|j|jkrt|d|j}t||S)u Subtract two sparse tensors element-wise. Input x and y's shape should be identical and have same sparse type（SparseCooTensor or SparseCsrTensor）.If input is SparseCooTensor, x and y's sparse_dim should be identical. The equation is: .. math:: out = x - y Args: x (Tensor): the input tensor, it's data type should be float32, float64, int32, int64. y (Tensor): the input tensor, it's data type should be float32, float64, int32, int64. name (str, optional): Name for the operation (optional, default is None). For more information, please refer to :ref:`api_guide_Name`. Returns: Tensor: the result tensor. Examples: .. code-block:: python >>> import paddle >>> paddle.device.set_device("cpu") >>> x = paddle.to_tensor([[0, -1, 0, 2], [0, 0, -3, 0], [4, 5, 0, 0]], 'float32') >>> y = paddle.to_tensor([[0, 0, 0, -2], [0, 2, -3, 0], [2, 3, 4, 8]], 'float32') >>> sparse_x = x.to_sparse_csr() >>> sparse_y = y.to_sparse_csr() >>> sparse_z = paddle.sparse.subtract(sparse_x, sparse_y) >>> print(sparse_z.to_dense()) Tensor(shape=[3, 4], dtype=float32, place=Place(cpu), stop_gradient=True, [[ 0., -1., 0., 4.], [ 0., -2., 0., 0.], [ 2., 2., -4., -8.]]) N)rr sparse_castZsparse_subtractr rrrsubtracts & rcCsJt|ttfrt|t|ddS|j|jkrt|d|j}t||S)u Multiply two sparse tensors element-wise. Input x and y's shape should be identical and have same sparse type（SparseCooTensor or SparseCsrTensor）.If input is SparseCooTensor, x and y's sparse_dim should be identical. The equation is: .. math:: out = x * y Args: x (Tensor): the input tensor, it's data type should be float32, float64, int32, int64. y (Tensor): the input tensor, it's data type should be float32, float64, int32, int64. name (str, optional): Name for the operation (optional, default is None). For more information, please refer to :ref:`api_guide_Name`. Returns: Tensor: the result tensor. Examples: .. code-block:: python >>> import paddle >>> paddle.device.set_device("cpu") >>> x = paddle.to_tensor([[0, -1, 0, 2], [0, 0, -3, 0], [4, 5, 0, 0]], 'float32') >>> y = paddle.to_tensor([[0, 0, 0, -2], [0, 2, -3, 0], [2, 3, 4, 8]], 'float32') >>> sparse_x = x.to_sparse_csr() >>> sparse_y = y.to_sparse_csr() >>> sparse_z = paddle.sparse.multiply(sparse_x, sparse_y) >>> print(sparse_z.to_dense()) Tensor(shape=[3, 4], dtype=float32, place=Place(cpu), stop_gradient=True, [[ 0., -0., 0., -4.], [ 0., 0., 9., 0.], [ 8., 15., 0., 0.]]) gTN) isinstanceintfloatrZ sparse_scalerrZsparse_multiplyr rrrmultiplyIs &  r"cCsd|jtvrt|dtjjj}t|t t frt |t |S|j|jkr,t|d|j}t ||S)u Divide two sparse tensors element-wise. Input x and y's shape should be identical and have same sparse type（SparseCooTensor or SparseCsrTensor）.If input is SparseCooTensor, x and y's sparse_dim should be identical. The equation is: .. math:: out = x / y Args: x (Tensor): the input tensor, it's data type should be float32, float64, int32, int64. y (Tensor): the input tensor, it's data type should be float32, float64, int32, int64. name (str, optional): Name for the operation (optional, default is None). For more information, please refer to :ref:`api_guide_Name`. Returns: Tensor: the result tensor. Examples: .. code-block:: python >>> import paddle >>> paddle.device.set_device("cpu") >>> x = paddle.to_tensor([[0, -1, 0, 2], [0, 0, -3, 0], [4, 5, 0, 0]], 'float32') >>> y = paddle.to_tensor([[0, 0, 0, -2], [0, 2, -3, 0], [2, 3, 4, 8]], 'float32') >>> sparse_x = x.to_sparse_csr() >>> sparse_y = y.to_sparse_csr() >>> sparse_z = paddle.sparse.divide(sparse_x, sparse_y) >>> print(sparse_z.to_dense()) Tensor(shape=[3, 4], dtype=float32, place=Place(cpu), stop_gradient=True, [[ nan , -inf. , nan , -1. ], [ nan , 0. , 1. , nan ], [ 2. , 1.66666663, 0. , 0. ]]) N) r _int_dtype_rrrVarDescVarTypeZFP32rr r!Zsparse_divide_scalarZ sparse_divider rrrdividews &  r&cCs ||S)aV Return the results of shape comparison between two Tensors, check whether x.shape equal to y.shape. Any two type Tensor among DenseTensor/SparseCooTensor/SparseCsrTensor are supported. Args: x (Tensor): The input tensor. It can be DenseTensor/SparseCooTensor/SparseCsrTensor. y (Tensor): The input tensor. It can be DenseTensor/SparseCooTensor/SparseCsrTensor. Returns: bool: True for same shape and False for different shape. Examples: .. code-block:: python >>> import paddle >>> x = paddle.rand([2, 3, 8]) >>> y = paddle.rand([2, 3, 8]) >>> y = y.to_sparse_csr() >>> z = paddle.rand([2, 5]) >>> paddle.sparse.is_same_shape(x, y) True >>> paddle.sparse.is_same_shape(x, z) False ) is_same_shaperrrrr's r')N)ZpaddlerrZpaddle.base.frameworkrrZpaddle.base.layer_helperrZunaryr__all__r$r%ZUINT8ZINT8ZINT16ZINT32ZINT64ZBOOLr#rrrrrr"r&r'rrrrs6   M ?  95 * - 0