python - 如何绘制 pytorch 张量？

Question

我想绘制 pytorch gpu 张量：

input= torch.randn(100).to(device)
output = torch.where(input>=0, input, -input)

input = input.('cpu').detach().numpy().copy()
output = output.('cpu').detach().numpy().copy()

plt.plot(input,out)

但是我尝试将这些张量转换为 cpu，numpy，它不起作用。如何绘制张量？

标签： pythonpytorchtensor

Answer 1

这行得通吗？

plt.plot(input.cpu().numpy(),output.cpu().numpy())

或者你可以试试，

plt.plot(input.to('cpu').numpy(),output.to('cpu').numpy())