TensorFlow

API

 tf.linalg / diag_part


Computes the eigenvalues of one or more self-adjoint matrices.

tensor Tensor of shape [..., N, N].
name string, optional name of the operation.

e Eigenvalues. Shape is [..., N]. The vector e[..., :] contains the N eigenvalues of tensor[..., :, :].

此页内容是否对您有帮助