python创建一个tensor张量
时间: 2023-10-13 10:27:05 浏览: 101
张量 tensor
3星 · 编辑精心推荐
要创建一个张量(tensor)对象,需要使用 PyTorch 库。以下是一个简单的示例代码,用于创建一个形状为 2x3 的张量:
```python
import torch
# 创建一个2x3的零张量
x = torch.zeros(2, 3)
print(x)
# 创建一个2x3的随机张量
y = torch.rand(2, 3)
print(y)
```
输出:
```
tensor([[0., 0., 0.],
[0., 0., 0.]])
tensor([[0.9477, 0.9506, 0.1771],
[0.0935, 0.1209, 0.1645]])
```
在这个示例中,我们首先导入 PyTorch 库,然后使用 `torch.zeros()` 创建一个形状为 2x3 的零张量,并使用 `torch.rand()` 创建一个形状为 2x3 的随机张量。可以看到,输出结果是一个张量对象,它包含了给定形状的零值或随机值。
阅读全文