Class MatrixSetDiag.Inputs<T extends TType>

java.lang.Object

org.tensorflow.op.RawOpInputs<MatrixSetDiag<T>>

org.tensorflow.op.linalg.MatrixSetDiag.Inputs<T>

Enclosing class:

MatrixSetDiag<T extends TType>

public static class MatrixSetDiag.Inputs<T extends TType>
extends RawOpInputs<MatrixSetDiag<T>>

Field Summary

Fields

Modifier and Type	Field	Description
final String	align	Some diagonals are shorter than max_diag_len and need to be padded.
final Operand<T>	diagonal	Rank r when k is an integer or k[0] == k[1].
final Operand<T>	input	Rank r+1, where r >= 1.
final Operand<TInt32>	k	Diagonal offset(s).
final DataType	T	The T attribute

Constructor Summary

Constructors

Constructor	Description
Inputs(GraphOperation op)

Method Summary

Methods inherited from class RawOpInputs

attributeMetadata, attributeNames, attributes, attributeValue, attributeValues, equals, getOutputs, hashCode, toString

Methods inherited from class Object

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

Field Details

input

public final Operand<T extends TType> input

Rank r+1, where r >= 1.

diagonal

public final Operand<T extends TType> diagonal

Rank r when k is an integer or k[0] == k[1]. Otherwise, it has rank r+1. k >= 1.

k

public final Operand<TInt32> k

Diagonal offset(s). Positive value means superdiagonal, 0 refers to the main diagonal, and negative value means subdiagonals. k can be a single integer (for a single diagonal) or a pair of integers specifying the low and high ends of a matrix band. k[0] must not be larger than k[1].

T

public final DataType T

The T attribute

align

public final String align

Some diagonals are shorter than max_diag_len and need to be padded. align is a string specifying how superdiagonals and subdiagonals should be aligned, respectively. There are four possible alignments: "RIGHT_LEFT" (default), "LEFT_RIGHT", "LEFT_LEFT", and "RIGHT_RIGHT". "RIGHT_LEFT" aligns superdiagonals to the right (left-pads the row) and subdiagonals to the left (right-pads the row). It is the packing format LAPACK uses. cuSPARSE uses "LEFT_RIGHT", which is the opposite alignment.

Constructor Details

Inputs

public Inputs(GraphOperation op)