|TensorFlow 2 version|
Defined in generated file:
Returns a diagonal tensor with a given diagonal values.
tf.linalg.tensor_diag( diagonal, name=None )
diagonal, this operation returns a tensor with the
everything else padded with zeros. The diagonal is computed as follows:
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.
# '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]]
Tensor. Must be one of the following types:
complex128. Rank k tensor where k is at most 1.
name: A name for the operation (optional).
Tensor. Has the same type as