Searched defs:LinearOperatorDiag (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/ops/linalg/
H A Dlinear_operator_diag.py31 __all__ = ["LinearOperatorDiag",]
34 @tf_export("linalg.LinearOperatorDiag")
35 class LinearOperatorDiag(linear_operator.LinearOperator): class in inherits:linear_operator.LinearOperator
44 `LinearOperatorDiag` is initialized with a (batch) vector.
49 operator = LinearOperatorDiag(diag)
67 operator = LinearOperatorDiag(diag)
90 Suppose `operator` is a `LinearOperatorDiag` of shape `[N, N]`,
121 name="LinearOperatorDiag"):
122 r"""Initialize a `LinearOperatorDiag`.
160 super(LinearOperatorDiag, sel
[all...]

Completed in 183 milliseconds