torch.tensor()、torch.rand()、torch.zeros()
时间: 2024-05-01 17:20:35 浏览: 121
PyTorch | (3)Tensor及其基本操作
这三个函数都是 PyTorch 中创建张量(tensor)的函数。
- `torch.tensor()` 可以将一个 Python 的 list 或者 numpy 数组转换成 PyTorch 的 tensor。
- `torch.rand()` 用于创建指定大小的随机数张量,张量中的每个元素都是从 [0, 1) 的均匀分布中随机采样得到的。
- `torch.zeros()` 用于创建指定大小的全零张量。
示例代码:
```python
import torch
# torch.tensor()
a = [1, 2, 3]
tensor_a = torch.tensor(a)
print(tensor_a)
# torch.rand()
rand_tensor = torch.rand(2, 3)
print(rand_tensor)
# torch.zeros()
zero_tensor = torch.zeros(2, 3)
print(zero_tensor)
```
输出结果:
```
tensor([1, 2, 3])
tensor([[0.6932, 0.3795, 0.5899],
[0.1538, 0.8040, 0.6843]])
tensor([[0., 0., 0.],
[0., 0., 0.]])
```
阅读全文