Classes
BandedTriangularSolve<T extends TType> | |
BandedTriangularSolve.Options | Optional attributes for BandedTriangularSolve
|
BandPart<T extends TType> | Copy a tensor setting everything outside a central band in each innermost matrix to zero. |
BatchCholesky<T extends TNumber> | |
BatchCholeskyGrad<T extends TNumber> | |
BatchMatrixBandPart<T extends TType> | |
BatchMatrixDeterminant<T extends TType> | |
BatchMatrixDiag<T extends TType> | |
BatchMatrixDiagPart<T extends TType> | |
BatchMatrixInverse<T extends TNumber> | |
BatchMatrixInverse.Options | Optional attributes for BatchMatrixInverse
|
BatchMatrixSetDiag<T extends TType> | |
BatchMatrixSolve<T extends TNumber> | |
BatchMatrixSolve.Options | Optional attributes for BatchMatrixSolve
|
BatchMatrixSolveLs<T extends TNumber> | |
BatchMatrixSolveLs.Options | Optional attributes for BatchMatrixSolveLs
|
BatchMatrixTriangularSolve<T extends TNumber> | |
BatchMatrixTriangularSolve.Options | Optional attributes for BatchMatrixTriangularSolve
|
BatchSelfAdjointEig<T extends TNumber> | |
BatchSelfAdjointEig.Options | Optional attributes for BatchSelfAdjointEig
|
BatchSvd<T extends TType> | |
BatchSvd.Options | Optional attributes for BatchSvd
|
Cholesky<T extends TType> | Computes the Cholesky decomposition of one or more square matrices. |
CholeskyGrad<T extends TNumber> | Computes the reverse mode backpropagated gradient of the Cholesky algorithm. |
ConjugateTranspose<T extends TType> | Shuffle dimensions of x according to a permutation and conjugate the result. |
Cross<T extends TNumber> | Compute the pairwise cross product. |
Det<T extends TType> | Computes the determinant of one or more square matrices. |
Eig<U extends TType> | Computes the eigen decomposition of one or more square matrices. |
Eig.Options | Optional attributes for Eig
|
Einsum<T extends TType> | Tensor contraction according to Einstein summation convention. |
EuclideanNorm<T extends TType> | Computes the euclidean norm of elements across dimensions of a tensor. |
EuclideanNorm.Options | Optional attributes for EuclideanNorm
|
Inv<T extends TType> | Computes the inverse of one or more square invertible matrices or their adjoints (conjugate transposes). |
Inv.Options | Optional attributes for Inv
|
LoadAndRemapMatrix | Loads a 2-D (matrix) `Tensor` with name `old_tensor_name` from the checkpoint
at `ckpt_path` and potentially reorders its rows and columns using the specified remappings. |
LoadAndRemapMatrix.Options | Optional attributes for LoadAndRemapMatrix
|
LogMatrixDeterminant<T extends TType> | Computes the sign and the log of the absolute value of the determinant of
one or more square matrices. |
Lu<T extends TType, U extends TNumber> | Computes the LU decomposition of one or more square matrices. |
MatMul<T extends TType> | Multiply the matrix "a" by the matrix "b". |
MatMul.Options | Optional attributes for MatMul
|
MatrixDiag<T extends TType> | Returns a batched diagonal tensor with given batched diagonal values. |
MatrixDiagPart<T extends TType> | Returns the batched diagonal part of a batched tensor. |
MatrixDiagPartV3<T extends TType> | Returns the batched diagonal part of a batched tensor. |
MatrixDiagPartV3.Options | Optional attributes for MatrixDiagPartV3
|
MatrixDiagV3<T extends TType> | Returns a batched diagonal tensor with given batched diagonal values. |
MatrixDiagV3.Options | Optional attributes for MatrixDiagV3
|
MatrixLogarithm<T extends TType> | Computes the matrix logarithm of one or more square matrices:
\\(log(exp(A)) = A\\) This op is only defined for complex matrices. |
MatrixSetDiag<T extends TType> | Returns a batched matrix tensor with new batched diagonal values. |
MatrixSetDiag.Options | Optional attributes for MatrixSetDiag
|
MatrixSolveLs<T extends TType> | Solves one or more linear least-squares problems. |
MatrixSolveLs.Options | Optional attributes for MatrixSolveLs
|
Qr<T extends TType> | Computes the QR decompositions of one or more matrices. |
Qr.Options | Optional attributes for Qr
|
QuantizedMatMul<V extends TType> | Perform a quantized matrix multiplication of `a` by the matrix `b`. |
QuantizedMatMul.Options | Optional attributes for QuantizedMatMul
|
QuantizedMatMulWithBias<W extends TType> | Performs a quantized matrix multiplication of `a` by the matrix `b` with bias add. |
QuantizedMatMulWithBias.Options | Optional attributes for QuantizedMatMulWithBias
|
QuantizedMatMulWithBiasAndRelu<V extends TType> | Perform a quantized matrix multiplication of `a` by the matrix `b` with bias add and relu fusion. |
QuantizedMatMulWithBiasAndRelu.Options | Optional attributes for QuantizedMatMulWithBiasAndRelu
|
QuantizedMatMulWithBiasAndReluAndRequantize<W extends TType> | Perform a quantized matrix multiplication of `a` by the matrix `b` with bias add and relu and requantize fusion. |
QuantizedMatMulWithBiasAndReluAndRequantize.Options | Optional attributes for QuantizedMatMulWithBiasAndReluAndRequantize
|
SelfAdjointEig<T extends TType> | Computes the eigen decomposition of one or more square self-adjoint matrices. |
SelfAdjointEig.Options | Optional attributes for SelfAdjointEig
|
Solve<T extends TType> | Solves systems of linear equations. |
Solve.Options | Optional attributes for Solve
|
Sqrtm<T extends TType> | Computes the matrix square root of one or more square matrices:
matmul(sqrtm(A), sqrtm(A)) = A The input matrix should be invertible. |
Svd<T extends TType> | Computes the singular value decompositions of one or more matrices. |
Svd.Options | Optional attributes for Svd
|
TensorDiag<T extends TType> | Returns a diagonal tensor with a given diagonal values. |
TensorDiagPart<T extends TType> | Returns the diagonal part of the tensor. |
Transpose<T extends TType> | Shuffle dimensions of x according to a permutation. |
TriangularSolve<T extends TType> | Solves systems of linear equations with upper or lower triangular matrices by backsubstitution. |
TriangularSolve.Options | Optional attributes for TriangularSolve
|
TridiagonalMatMul<T extends TType> | Calculate product with tridiagonal matrix. |
TridiagonalSolve<T extends TType> | Solves tridiagonal systems of equations. |
TridiagonalSolve.Options | Optional attributes for TridiagonalSolve
|