numpy - 是否有 numpy.linalg 函数的行为类似于 tensorflow.linalg 函数在外部昏暗上广播并在两个内部昏暗上运行？

Question

例如，如果A.shape = (100, 5, 5)，tf.linalg.diag(A)将对最后两个暗淡进行操作，并为您提供(100, 5).

这些东西有 numpy 版本吗？

标签： numpytensorflow