使用torch.rand()生成 [-1, 1)之间的张量_torch 生成-1, 1

这段代码主要涉及到Python的torch库（PyTorch），用来生成随机数据，并且展示了如何用函数列表来调用函数。让我们详细解释一下每一部分。

torch.rand(N, 4) * 2 - 1 的解释

• torch.rand(N, 4): 这个函数调用生成一个形状为 (N, 4) 的张量，元素均为 [0, 1) 之间的均匀分布的随机数。在这里，N 是 10,000，所以生成的张量大小为 10,000 x 4。
• * 2: 将所有元素乘以 2，这会将原来的 [0, 1) 范围扩展到 [0, 2)。
• - 1: 最后，将所有元素减去 1，这会将范围从 [0, 2) 转换到 [-1, 1)。

所以，torch.rand(N, 4) * 2 - 1 生成了一个包含 10,000 行和 4 列的张量，所有元素都是在 [-1, 1) 范围内的均匀分布的随机数。

代码解释

import torch

N = 10_000
x_train = torch.rand(N, 4) * 2 - 1
1
2
3
4

1. import torch: 导入 PyTorch 库。
2. N = 10_000: 定义一个整数 N，表示生成的数据量，具体值为 10,000。
3. x_train = torch.rand(N, 4) * 2 - 1: 使用 torch.rand 函数生成一个 10,000 x 4 的张量，元素范围在 [-1, 1) 之间，并将其赋值给 x_train。