o #j#,@sddlZddlZddlmZmZddlmZddlmZm Z m Z m Z ddl m Z ddlmZmZddgZd d Zd d Zd dZddZ dddZe dddZdS)N)_C_opsin_dynamic_mode) convert_dtype)_current_expected_place_get_paddle_placecore dygraph_only) LayerHelper)max to_tensorsparse_coo_tensorsparse_csr_tensorcCs(|rt|t|jkr|t|S|S)N)rdtypeZastype)datarrW/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/sparse/creation.py _handle_dtype#srcCsZt|jdks Jt|dd}|d}|}t|jdkr)t||jdd}t|S)N)Zaxis)lenshaper numpynpappendlist)indicesvaluesZlensrrr_infer_dense_shape*s rcCs@t|}|dur t}|St|tjtjtjtjfstd|S)Nz^'place' must be any of paddle.Place, paddle.CPUPlace, paddle.CUDAPinnedPlace, paddle.CUDAPlace) rr isinstancerZPlaceZCPUPlaceZCUDAPinnedPlaceZ CUDAPlace ValueError)placerrr _get_place4sr!cCs$|tjtjtjtjfvrtddS)NzDthe dtype of indices must be 'int8' or 'int16' or 'int32' or 'int64')paddleZint8Zint16Zint32Zint64 TypeError)rrrr_check_indices_dtypeAs r$TcCstrt|}t|tjjst|d|dd}t|tjjs$t||||}t|jdkr/t d|jd}|jd}t |j ||jdkrPt d ||jdt|jd}|j |sc||d }|j |so||d }t||}||_t||} |dur| }n%t|}|| krt d | d |t|||krt d ||t|t|||Sd } ||d} |ddurd|d<d|i} t| } | |}| j| | d|i| d|S)a& Constructs a sparse ``paddle.Tensor`` in coordinate format according to the indices and values of the specified non-zero elements. Args: indices(list|tuple|ndarray|Tensor): the indices of non-zero elements. Can be a list, tuple, numpy\.ndarray, paddle\.Tensor. The indices must be 2-D. values(list|tuple|ndarray|Tensor): Initial values for the tensor. Can be a scalar, list, tuple, numpy\.ndarray, paddle\.Tensor. shape(list|tuple, optional): The shape of the sparse tensor also represents the shape of original dense tensor. If not provided the smallest shape will be inferred to hold all elements. dtype(str|np.dtype, optional): The desired data type of returned tensor. Can be 'bool' , 'float16' , 'float32' , 'float64' , 'int8' , 'int16' , 'int32' , 'int64' , 'uint8', 'complex64' , 'complex128'. Default: None, infers dtype from ``data`` except for python float number which gets dtype from ``get_default_type`` . place(CPUPlace|CUDAPinnedPlace|CUDAPlace|str, optional): The place to allocate Tensor. Can be CPUPlace, CUDAPinnedPlace, CUDAPlace. Default: None, means global place. If ``place`` is string, It can be ``cpu``, ``gpu:x`` and ``gpu_pinned``, where ``x`` is the index of the GPUs. stop_gradient(bool, optional): Whether to block the gradient propagation of Autograd. Default: True. Returns: Tensor: A Tensor constructed from ``indices`` and ``values`` . Examples: .. code-block:: python >>> import paddle >>> indices = [[0, 1, 2], [1, 2, 0]] >>> values = [1.0, 2.0, 3.0] >>> dense_shape = [3, 3] >>> coo = paddle.sparse.sparse_coo_tensor(indices, values, dense_shape) >>> print(coo) Tensor(shape=[3, 3], dtype=paddle.float32, place=Place(cpu), stop_gradient=True, indices=[[0, 1, 2], [1, 2, 0]], values=[1., 2., 3.]) NTrr stop_gradientrz'indices' must be 2-D.rrzKthe indices and values must have same number of non-zero, but get {} and {}Fzthe minimun shape required is z , but get zVthe number of dimensions(len(shape) must be sparse_dim({}) + dense_dim({}), but get {}sparse_sparse_coo_tensor)rrrout)typeinputsZoutputsattrs)rr!rreagerTensorr rrrr$rformatr _equals_copy_torr&rrrr'r Z)create_sparse_variable_for_type_inferenceZ append_op)rrrrr r&ZnnzZ sparse_dimZ dense_dimZ min_shapeZop_typer+r,helperr)rrrr Hsh,               cCst|}t|tjjst|d|dd}t|tjjs"t|d|dd}t|tjjs0t||||}t|jt|jt|dkrMt|dkrMt d||t|d}|j |sa| |d}|j |sm| |d}|j |sy| |d}t ||}||_t|jdkst|jdkst|jdkrt d t|t|krt d t|dkr|jd |dkrt d |jd ||d d krt d |d|jd krt dn|jd |dd krt d|jd |tj|||||S)a< Constructs a sparse ``paddle.Tensor`` in CSR(Compressed Sparse Row) format according to the ``crows``, ``cols`` and ``values``. Currently, the crows and cols of each batch must be incrementd. Args: crows(list|tuple|ndarray|Tensor): 1-D array, each element in the rows represents the starting position of the first non-zero element of each row in values. Can be a list, tuple, numpy\.ndarray, paddle\.Tensor. cols(list|tuple|ndarray|Tensor): 1-D array, the column of non-zero elements. Can be a list, tuple, numpy\.ndarray, paddle\.Tensor. values(list|tuple|ndarray|Tensor): 1-D array, the non-zero elements. Can be a scalar, list, tuple, numpy\.ndarray, paddle\.Tensor. shape(list|tuple, optional): The shape of the sparse tensor also represents the shape of original dense tensor. hold all elements. dtype(str|np.dtype, optional): The desired data type of returned tensor. Can be 'bool' , 'float16' , 'float32' , 'float64' , 'int8' , 'int16' , 'int32' , 'int64' , 'uint8', 'complex64' , 'complex128'. Default: None, infers dtype from ``data`` except for python float number which gets dtype from ``get_default_type`` . place(CPUPlace|CUDAPinnedPlace|CUDAPlace|str, optional): The place to allocate Tensor. Can be CPUPlace, CUDAPinnedPlace, CUDAPlace. Default: None, means global place. If ``place`` is string, It can be ``cpu``, ``gpu:x`` and ``gpu_pinned``, where ``x`` is the index of the GPUs. stop_gradient(bool, optional): Whether to block the gradient propagation of Autograd. Default: True. Returns: Tensor: A Tensor constructed from ``crows``, ``cols`` and ``values`` . Examples: .. code-block:: python >>> import paddle >>> 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.int64, place=Place(cpu), stop_gradient=True, crows=[0, 2, 3, 5], cols=[1, 3, 2, 0, 1], values=[1, 2, 3, 4, 5]) NTr%rz>SparseCsrTensor only support 2-D or 3-D matrix. but get shape Frz-The 'crows', 'cols' and 'values' must be 1-D.z3the length of cols must be same as length of valuesrzBThe length({}) of crows must be equal to the rows({})+1 of matrix.z the 0th value of crows must be 0r(zs&     p