inp = torch.tensor([1.0,2.0])
时间: 2023-12-10 09:42:32 浏览: 75
torch-1.0.0
5星 · 资源好评率100%
这是一个创建 PyTorch 张量(tensor)的示例代码,其中包含两个元素(1.0和2.0)。可以通过以下代码实现:
```
import torch
inp = torch.tensor([1.0, 2.0])
print(inp)
```
输出:
```
tensor([1., 2.])
```
阅读全文