如何通过matplotlib或任何其他可视化工具比较两个不同的张量？

import numpy as np
import matplotlib.pyplot as plt

tensor1 = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
tensor2 = np.array([[9, 8, 7], [6, 5, 4], [3, 2, 1]])

plt.figure(figsize=(10, 5))

# 绘制tensor1
plt.subplot(1, 2, 1)
plt.imshow(tensor1, cmap='viridis')
plt.title('Tensor 1')

# 绘制tensor2
plt.subplot(1, 2, 2)
plt.imshow(tensor2, cmap='viridis')
plt.title('Tensor 2')

plt.tight_layout()
plt.show()