python - 如何用python生成身份张量？

Question

我知道 np.eye生成单位矩阵。我的意思是单位矩阵为

In linear algebra, the identity matrix, or sometimes ambiguously called a unit matrix, of size n is the n × n square matrix with ones on the main diagonal and zeros elsewhere.

我知道我们可以在 Numpy 中创建它与 np.identity(3) .

But, I would like to know how can I have an identity Tensor in python.

我想在张量乘法中使用恒等张量。像下面这样:

哪里G = Er ×1 U1 ×2 U2 ...×M UM是一个变换张量，而 Er ∈ R r×r×...×r是一个单位张量(对角线元素为 1，所有其他条目为 0)。我需要有生成 identity tensor 的代码.

先感谢您。

Answer 1

而不是 np.identity使用 tf.eye :

tf.eye(2)
# [[1., 0.],
#  [0., 1.]]