Matplotlib 绘图支持 tensor 数据类型

今天发现了一个特别方便的东西，原来 Matplotlib 绘图是直接支持 tensor 数据类型的，这样的话以后在机器学习可视化中就无需对 tensor 数据进行 numpy 数据转换了。

示例1------plot()

import torch
import matplotlib.pyplot as plt

a = torch.linspace(0., 2. * torch.pi, steps=25, requires_grad=True)

b = torch.sin(a)
plt.figure(figsize=[8, 5])
plt.plot(a.detach(), b.detach())
plt.show()

得到的图像如下：

示例2------imshow()

import torch
import matplotlib.pyplot as plt

a = torch.linspace(0., 2. * torch.pi, steps=25, requires_grad=True)

b = torch.sin(a)
plt.figure(figsize=[8, 5])
plt.imshow(a.detach().view(1, -1))
plt.show()

得到的图像如下：

细心的小伙伴可能会发现，对于获取 pi 值，我们又多了一种方法，即使用 torch.pi。