/ / skonstruuj tensor diagonalny 3d używając tensora 2d - python, numpy, matrix, tensorflow, broadcast

skonstruuj 3d tensor diagonalny za pomocą tensora 2d - python, numpy, matrix, tensorflow, broadcast

Biorąc pod uwagę A = [[1,2], [3,4], [5,6]]. Jak korzystać z tf.diag (), aby skonstruować tensor 3d, w którym każdy stos jest dwuwymiarową macierzą diagonalną, używając wartości z A? Tak więc wyjściem powinno być B = [[[1,0], [0,2]], [[3,0], [0,4]], [[5,0], [0,6]]] . Chcę użyć tego jako moich relacji kowariancji Gaussa.

Odpowiedzi:

0 dla odpowiedzi № 1

Ok, rozumiem. tf.matrix_diag () wykonuje sztuczkę ...