torch 生成两条类似正弦曲线的散点图,幅值和频率接近,
时间: 2023-10-01 19:09:07 浏览: 167
测量正弦信号的幅值和频率
4星 · 用户满意度95%
下面是使用PyTorch生成两条类似正弦曲线的散点图的示例代码:
```python
import torch
import matplotlib.pyplot as plt
# 生成正弦曲线
x = torch.linspace(0, 2 * torch.tensor(3.1415926), 1000)
y1 = torch.sin(x)
y2 = torch.sin(x + 0.2)
# 添加随机噪声
noise1 = torch.randn_like(y1) * 0.1
noise2 = torch.randn_like(y2) * 0.1
y1 += noise1
y2 += noise2
# 绘制散点图
fig, ax = plt.subplots()
ax.scatter(x, y1, label='sin(x)', s=10)
ax.scatter(x, y2, label='sin(x+0.2)', s=10)
plt.legend()
plt.show()
```
这段代码首先使用torch.linspace()函数生成了0到2π之间1000个等间距的数据点,然后使用torch.sin()函数计算出对应的正弦值。接着,添加了一些随机噪声以使得两条曲线看起来更加相似,最后使用matplotlib中的scatter()函数绘制出散点图。
阅读全文