construya el tensor diagonal 3d utilizando el tensor 2d - python, numpy, matrix, tensorflow, broadcast

Dado A = [[1,2], [3,4], [5,6]]. Cómo utilizar tf.diag () para construir un tensor 3d donde cada pila es una matriz diagonal 2d usando los valores de A? Por lo tanto, la salida debe ser B = [[[1,0], [0,2]], [[3,0], [0,4]], [[5,0], [0,6]]] . Quiero usar esto como mis matrices de covarianza gaussianas.

Respuestas

Ok me lo imagino. tf.matrix_diag () hace el truco ...