PyTorch随机初始化数据的生成方法

一、PyTorch的Tensor数据结构

PyTorch是基于Python的科学计算库，它的核心是Tensor，它提供了一个强大的多维数组支持。Tensor类似于NumPy的ndarray，不同之处在于Tensor可以在GPU上加速运算。

在PyTorch中，使用随机生成的数据进行模型的训练和调试是非常常见的一种方式。同时，随机化也能够尽可能地使数据的分布更符合实际问题的分布，从而提高训练效果。下面将介绍几种PyTorch随机初始化数据的生成方法。

二、使用torch.rand生成随机数据

在PyTorch中，可以使用torch.rand()函数生成0到1的均匀分布上的随机数，并且可以指定随机数的大小。例如，下面的代码生成大小为(2, 3)的随机数：

import torch
random_data = torch.rand(2, 3)
print(random_data)

上述代码执行后，将生成一个大小为(2, 3)的张量，包含0到1之间的随机数。运行结果如下：

tensor([[0.3350, 0.0242, 0.2949],
        [0.3946, 0.2341, 0.2044]])

上述代码使用了torch.rand()函数生成了一个大小为(2, 3)的张量，其中有两个维度，每个维度的大小分别为2和3，所以张量的形状为(2, 3)。可以使用size()函数获取张量的形状。在生成数据时，可以根据需要修改张量的形状，例如可以使用reshape()函数将张量大小更改为(6,)，表示一个一维张量：

import torch
random_data = torch.rand(2, 3).reshape(6)
print(random_data)

上述代码生成了一个大小为(2, 3)的张量，然后使用reshape()函数将其大小更改为(6,)，表示一个一维张量。运行结果如下：

tensor([0.0786, 0.5189, 0.9957, 0.7286, 0.0571, 0.8879])

三、使用torch.randn生成正态分布的随机数据

除了均匀分布上的随机数据之外，PyTorch还提供了生成正态分布上的随机数据的函数torch.randn()。torch.randn()函数可以生成指定大小的正态分布上的随机数。

例如，下面的代码生成一个大小为(3, 4)的随机数：

import torch
random_data = torch.randn(3, 4)
print(random_data)

上述代码使用了torch.randn()函数生成了一个大小为(3, 4)的张量。运行结果如下：

tensor([[-0.7532,  0.8406,  2.2453,  1.0552],
        [-0.4252,  0.2142, -1.1183,  0.1952],
        [ 0.1936, -0.5773, -0.2420,  0.7166]])

可以看到，生成的数据呈正态分布，均值接近于0，标准差接近于1。

四、使用torch.randint生成随机整数数据

有时候需要生成随机的整数数据，可以使用torch.randint()函数生成随机整数数据。torch.randint()函数需要指定生成数值的下限和上限，并且可以指定生成数据的大小。

例如，下面的代码生成大小为(2, 3)、数值在0到9之间的整数：

import torch
random_data = torch.randint(low=0, high=10, size=(2, 3))
print(random_data)

上述代码使用了torch.randint()函数生成了一个大小为(2, 3)、数值在0到9之间的整数。运行结果如下：

tensor([[3, 2, 6],
        [5, 4, 7]])

五、使用torch.manual_seed实现随机数的可重复性

通常，PyTorch随机初始化数据的生成是根据系统时间生成的，也就是说，每次运行程序都会生成不同的随机数。有时候，需要生成可重复的随机数，这时可以使用torch.manual_seed()函数设置随机种子。使用torch.manual_seed()函数设置的随机种子将会使得生成的随机数可重复。

例如，下面的代码演示了如何使用torch.manual_seed()函数生成可重复的随机数：

import torch
torch.manual_seed(0)
random_data1 = torch.randn(2, 3)
torch.manual_seed(0)
random_data2 = torch.randn(2, 3)
print(random_data1)
print(random_data2)

上述代码使用了torch.manual_seed()函数设置了随机种子，并生成了两个大小为(2, 3)的张量。由于两次生成的随机数使用了相同的随机种子，所以这两个张量的值是相同的。运行结果如下：

tensor([[ 1.5410, -0.2934, -2.1788],
        [ 0.5684, -1.0845, -1.3986]])
tensor([[ 1.5410, -0.2934, -2.1788],
        [ 0.5684, -1.0845, -1.3986]])

六、使用torch.empty_like()函数生成未初始化的随机数

在模型训练过程中，有时候需要使用一些未初始化的随机数，可以使用torch.empty_like()函数生成未初始化的随机数。torch.empty_like()函数会返回一个与输入张量相同大小的新张量，但它并不会初始化该张量，速度比使用torch.randn()等初始化函数更快。

例如，下面的代码生成一个大小为(2, 3)的未初始化的随机张量：

import torch
random_data = torch.empty_like(torch.randn(2, 3))
print(random_data)

上述代码使用torch.randn()函数生成一个大小为(2, 3)的张量作为参数，然后使用torch.empty_like()函数生成一个未初始化的大小相同的随机张量。运行结果如下：

tensor([[ 9.2755e-43,  0.0000e+00, -2.0000e+00],
        [ 1.4013e-45,  0.0000e+00,  0.0000e+00]])

七、总结

本文介绍了PyTorch生成随机数据的几种常用方法，包括使用torch.rand()生成均匀分布的随机数、使用torch.randn()生成正态分布的随机数、使用torch.randint()生成随机整数数据、使用torch.manual_seed()函数实现随机数的可重复性、使用torch.empty_like()生成未初始化的随机数等。在使用这些方法时，可以根据需要自由地指定随机数的大小和分布。

原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/306660.html