写一段pytorch画random state图的代码
时间: 2024-02-10 09:10:44 浏览: 140
pytorch测试代码
好的,下面是一段使用 PyTorch 生成随机数据并绘制散点图的代码:
```python
import torch
import matplotlib.pyplot as plt
# 生成随机数据
x = torch.randn(100)
y = torch.randn(100)
# 绘制散点图
plt.scatter(x, y)
plt.show()
```
这段代码中,我们首先使用 PyTorch 的 `randn` 函数生成了两个长度为 100 的随机张量 `x` 和 `y`,然后使用 Matplotlib 的 `scatter` 函数绘制了它们的散点图,并调用 `show` 函数显示图像。你可以根据自己的需求修改数据的维度、形状和范围等参数。
阅读全文