tensorflow::ops::Diag
#include <array_ops.h>
Returns a diagonal tensor with a given diagonal values.
Summary
Given a diagonal
, this operation returns a tensor with the diagonal
and everything else padded with zeros. The diagonal is computed as follows:
Assume diagonal
has dimensions [D1,..., Dk], then the output is a tensor of rank 2k with dimensions [D1,..., Dk, D1,..., Dk] where:
output[i1,..., ik, i1,..., ik] = diagonal[i1, ..., ik]
and 0 everywhere else.
For example:
# 'diagonal' is [1, 2, 3, 4] tf.diag(diagonal) ==> [[1, 0, 0, 0] [0, 2, 0, 0] [0, 0, 3, 0] [0, 0, 0, 4]]
Arguments:
- scope: A Scope object
- diagonal: Rank k tensor where k is at most 1.
Returns:
-
Output
: The output tensor.
Constructors and Destructors | |
---|---|
Diag(const ::tensorflow::Scope & scope, ::tensorflow::Input diagonal) |
Public attributes | |
---|---|
operation | |
output |
Public functions | |
---|---|
node() const | ::tensorflow::Node * |
operator::tensorflow::Input() const | |
operator::tensorflow::Output() const |
Public attributes
operation
Operation operation
output
::tensorflow::Output output
Public functions
Diag
Diag( const ::tensorflow::Scope & scope, ::tensorflow::Input diagonal )
node
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/cc/class/tensorflow/ops/diag