python使用torch 库简单举例
时间: 2023-09-16 11:13:35 浏览: 89
Python-介绍PyTorch的简单示例
好的,以下是一个简单的使用 PyTorch 库的例子:
```
import torch
# 创建一个张量
x = torch.tensor([[1, 2], [3, 4]])
# 打印张量
print(x)
# 创建一个全零张量
y = torch.zeros((2, 2))
# 打印全零张量
print(y)
# 相加
z = x + y
# 打印相加后的结果
print(z)
```
这个例子展示了如何使用 PyTorch 创建一个张量,并且打印出来。另外,我们创建了一个全零张量,然后将两个张量相加,最后打印出相加后的结果。
阅读全文