T
- data type for output()
output@Operator(group="linalg") public final class Sqrtm<T> extends PrimitiveOp implements Operand<T>
matmul(sqrtm(A), sqrtm(A)) = A
The input matrix should be invertible. If the input matrix is real, it should have no eigenvalues which are real and negative (pairs of complex conjugate eigenvalues are allowed).
The matrix square root is computed by first reducing the matrix to quasi-triangular form with the real Schur decomposition. The square root of the quasi-triangular matrix is then computed directly. Details of the algorithm can be found in: Nicholas J. Higham, "Computing real square roots of a real matrix", Linear Algebra Appl., 1987.
The input is a tensor of shape `[..., M, M]` whose inner-most 2 dimensions form square matrices. The output is a tensor of the same shape as the input containing the matrix square root for all input submatrices `[..., :, :]`.
operation
Modifier and Type | Method and Description |
---|---|
Output<T> |
asOutput()
Returns the symbolic handle of a tensor.
|
static <T> Sqrtm<T> |
create(Scope scope,
Operand<T> input)
Factory method to create a class wrapping a new Sqrtm operation.
|
Output<T> |
output()
Shape is `[..., M, M]`.
|
equals, hashCode, op, toString
public static <T> Sqrtm<T> create(Scope scope, Operand<T> input)
scope
- current scopeinput
- Shape is `[..., M, M]`.public Output<T> asOutput()
Operand
Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.
asOutput
in interface Operand<T>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.