org.tensorflow.op.linalg

Class Cholesky<T>

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.linalg.Cholesky<T>

Type Parameters:

T - data type for output() output

All Implemented Interfaces:

Op, Operand<T>

@Operator(group="linalg")
public final class Cholesky<T>
extends PrimitiveOp
implements Operand<T>

Computes the Cholesky decomposition of one or more square matrices.

The input is a tensor of shape `[..., M, M]` whose inner-most 2 dimensions form square matrices.

The input has to be symmetric and positive definite. Only the lower-triangular part of the input will be used for this operation. The upper-triangular part will not be read.

The output is a tensor of the same shape as the input containing the Cholesky decompositions for all input submatrices `[..., :, :]`.

Note: The gradient computation on GPU is faster for large matrices but not for large batch dimensions when the submatrices are small. In this case it might be faster to use the CPU.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
Output<T>	asOutput() Returns the symbolic handle of a tensor.
static <T> Cholesky<T>	create(Scope scope, Operand<T> input) Factory method to create a class wrapping a new Cholesky operation.
Output<T>	output() Shape is `[..., M, M]`.

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Method Detail

create

public static <T> Cholesky<T> create(Scope scope,
                                     Operand<T> input)

Factory method to create a class wrapping a new Cholesky operation.

Parameters:

scope - current scope

input - Shape is `[..., M, M]`.

Returns:

a new instance of Cholesky

output

public Output<T> output()

Shape is `[..., M, M]`.

asOutput

public Output<T> asOutput()

Description copied from interface: Operand

Returns the symbolic handle of a tensor.

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.

Specified by:

asOutput in interface Operand<T>

See Also:

OperationBuilder.addInput(Output)