T
- data type for output()
output@Operator(group="linalg") public final class CholeskyGrad<T extends Number> extends PrimitiveOp implements Operand<T>
For an explanation see "Differentiation of the Cholesky algorithm" by Iain Murray http://arxiv.org/abs/1602.07527.
operation
Modifier and Type | Method and Description |
---|---|
Output<T> |
asOutput()
Returns the symbolic handle of a tensor.
|
static <T extends Number> |
create(Scope scope,
Operand<T> l,
Operand<T> grad)
Factory method to create a class wrapping a new CholeskyGrad operation.
|
Output<T> |
output()
Symmetrized version of df/dA .
|
equals, hashCode, op, toString
public static <T extends Number> CholeskyGrad<T> create(Scope scope, Operand<T> l, Operand<T> grad)
scope
- current scopel
- Output of batch Cholesky algorithm l = cholesky(A). Shape is `[..., M, M]`.
Algorithm depends only on lower triangular part of the innermost matrices of
this tensor.grad
- df/dl where f is some scalar function. Shape is `[..., M, M]`.
Algorithm depends only on lower triangular part of the innermost matrices of
this tensor.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 extends Number>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.