wehub-resource-sync
135 lines
2.8 KiB
Python
135 lines
2.8 KiB
Python
# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from paddle import Tensor
|
|
|
|
from .tensor import inverse as inv
|
|
from .tensor.linalg import (
|
|
cholesky,
|
|
cholesky_inverse,
|
|
cholesky_solve,
|
|
cond,
|
|
corrcoef,
|
|
cov,
|
|
det,
|
|
diagonal,
|
|
eig,
|
|
eigh,
|
|
eigvals,
|
|
eigvalsh,
|
|
fp8_fp8_half_gemm_fused,
|
|
householder_product,
|
|
lstsq,
|
|
lu,
|
|
lu_solve,
|
|
lu_unpack,
|
|
matmul,
|
|
matrix_exp,
|
|
matrix_norm,
|
|
matrix_power,
|
|
matrix_rank,
|
|
matrix_transpose,
|
|
multi_dot,
|
|
norm,
|
|
ormqr,
|
|
pca_lowrank,
|
|
pinv,
|
|
qr,
|
|
slogdet,
|
|
solve,
|
|
svd,
|
|
svd_lowrank,
|
|
svdvals,
|
|
triangular_solve,
|
|
vecdot,
|
|
vector_norm,
|
|
)
|
|
|
|
__all__ = [
|
|
'cholesky',
|
|
'cholesky_inverse',
|
|
'norm',
|
|
'matrix_norm',
|
|
'vecdot',
|
|
'vector_norm',
|
|
'cond',
|
|
'cov',
|
|
'corrcoef',
|
|
'cross',
|
|
'inv',
|
|
'eig',
|
|
'eigvals',
|
|
'multi_dot',
|
|
'matrix_rank',
|
|
'matrix_transpose',
|
|
'matmul',
|
|
'svd',
|
|
'svdvals',
|
|
'qr',
|
|
'householder_product',
|
|
'pca_lowrank',
|
|
'svd_lowrank',
|
|
'lu',
|
|
'lu_solve',
|
|
'lu_unpack',
|
|
'matrix_exp',
|
|
'matrix_power',
|
|
'det',
|
|
'slogdet',
|
|
'eigh',
|
|
'eigvalsh',
|
|
'pinv',
|
|
'solve',
|
|
'cholesky_solve',
|
|
'triangular_solve',
|
|
'lstsq',
|
|
'ormqr',
|
|
'fp8_fp8_half_gemm_fused',
|
|
'diagonal',
|
|
]
|
|
|
|
|
|
def cross(
|
|
input: Tensor,
|
|
other: Tensor,
|
|
*,
|
|
dim: int = -1,
|
|
out: Tensor | None = None,
|
|
) -> Tensor:
|
|
"""
|
|
Computes the cross product of two 3-dimensional vectors along the specified dimension.
|
|
|
|
Refer to :ref:`api_paddle_cross` for more detail.
|
|
|
|
Args:
|
|
input (Tensor): The first input tensor.
|
|
other (Tensor): The second input tensor.
|
|
dim (int, optional): The dimension along which to compute the cross product.
|
|
Default: -1, which means the last dimension.
|
|
out (Tensor, optional): The output tensor. Default: None.
|
|
name (str, optional): Name for the operation. For more information, please
|
|
refer to :ref:`api_guide_Name`. Default: None.
|
|
|
|
Returns:
|
|
Tensor: The cross product of ``input`` and ``other``.
|
|
"""
|
|
import paddle
|
|
|
|
return paddle.cross(input, other, axis=dim, out=out)
|