如何构建pytorch张量
时间: 2023-08-25 16:07:55 浏览: 92
如何搭建pytorch环境
要构建一个 PyTorch 张量,可以使用 `torch.Tensor()` 函数。例如,创建一个大小为 3x3 的张量,可以使用以下代码:
```python
import torch
tensor = torch.Tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(tensor)
```
输出结果为:
```
tensor([[1., 2., 3.],
[4., 5., 6.],
[7., 8., 9.]])
```
在上面的示例中,我们使用 `torch.Tensor()` 构造函数创建了一个大小为 3x3 的张量,并将其存储在变量 `tensor` 中。注意,我们传递给 `torch.Tensor()` 的参数是一个列表,其中包含每个行的值。在这个例子中,我们创建了一个包含数字 1 到 9 的 3x3 矩阵。
阅读全文