python - 如何用python生成恒等张量？

Question

标签： pythonnumpytensorflowidentitytensor

Answer 1

Instead of np.identity use tf.eye:

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