T
- data type for output()
output@Operator(group="linalg") public final class Solve<T> extends PrimitiveOp implements Operand<T>
`Matrix` is a tensor of shape `[..., M, M]` whose inner-most 2 dimensions form square matrices. `Rhs` is a tensor of shape `[..., M, K]`. The `output` is a tensor shape `[..., M, K]`. If `adjoint` is `False` then each output matrix satisfies `matrix[..., :, :] * output[..., :, :] = rhs[..., :, :]`. If `adjoint` is `True` then each output matrix satisfies `adjoint(matrix[..., :, :]) * output[..., :, :] = rhs[..., :, :]`.
Modifier and Type | Class and Description |
---|---|
static class |
Solve.Options
Optional attributes for
Solve |
operation
Modifier and Type | Method and Description |
---|---|
static Solve.Options |
adjoint(Boolean adjoint) |
Output<T> |
asOutput()
Returns the symbolic handle of a tensor.
|
static <T> Solve<T> |
create(Scope scope,
Operand<T> matrix,
Operand<T> rhs,
Solve.Options... options)
Factory method to create a class wrapping a new Solve operation.
|
Output<T> |
output()
Shape is `[..., M, K]`.
|
equals, hashCode, op, toString
public static <T> Solve<T> create(Scope scope, Operand<T> matrix, Operand<T> rhs, Solve.Options... options)
scope
- current scopematrix
- Shape is `[..., M, M]`.rhs
- Shape is `[..., M, K]`.options
- carries optional attributes valuespublic static Solve.Options adjoint(Boolean adjoint)
adjoint
- Boolean indicating whether to solve with `matrix` or its (block-wise)
adjoint.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.