chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,134 @@
|
||||
# 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)
|
||||
Reference in New Issue
Block a user