torch.tensor生成随机数
时间: 2024-08-13 08:06:31 浏览: 52
Pytorch生成随机数Tensor的方法汇总
`torch.tensor` 是 PyTorch 库中的一个核心组件,用于创建张量(Tensor),它是类似于 NumPy 的数组,但更适合于深度学习和 GPU 计算。为了生成随机数,你可以使用 `torch.rand()` 或 `torch.randn()` 函数。以下是两种方法的简要介绍:
1. **torch.rand()**: 这个函数会生成一个给定形状的张量,元素的值均匀分布在 [0, 1) 区间内。例如,创建一个形状为 (2, 3) 的随机张量:
```python
import torch
random_tensor = torch.rand((2, 3))
```
2. **torch.randn()**: 这个函数生成的张量元素值服从标准正态分布(均值为 0,标准差为 1)。如果你想生成同样形状的正态分布随机数,可以这样做:
```python
normal_tensor = torch.randn((2, 3))
```
如果你需要特定范围的随机数,可以先将 `torch.rand()` 或 `torch.randn()` 的结果乘以范围的跨度,再加上范围的起始点。
阅读全文