Функция randn

Функция randn создает тензор заданной формы, заполненный случайными числами из стандартного нормального распределения (среднее значение 0, дисперсия 1). Первым параметром функция принимает размерность тензора (переменное количество аргументов или кортеж). Вторым параметром можно передать тип данных элементов, а также другие опциональные аргументы.

Синтаксис

torch.randn(*size, dtype=None, device=None, requires_grad=False)

Пример

Давайте создадим одномерный тензор из 5 случайных чисел:

import torch

torch.manual_seed(0)
t = torch.randn(5)
print(t)

Результат выполнения кода:

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

Пример

Давайте создадим двумерный тензор размером 2x3:

import torch

torch.manual_seed(0)
t = torch.randn(2, 3)
print(t)

Результат выполнения кода:

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

Пример

Давайте зададим тип данных float64 для элементов тензора:

import torch

torch.manual_seed(0)
t = torch.randn(3, dtype=torch.float64)
print(t)

Результат выполнения кода:

tensor([ 1.5410, -0.2934, -2.1788], dtype=torch.float64)

Пример

Давайте создадим тензор с включенным вычислением градиента:

import torch

torch.manual_seed(0)
t = torch.randn(4, requires_grad=True)
print(t)

Результат выполнения кода:

tensor([ 1.5410, -0.2934, -2.1788, 0.5684], requires_grad=True)

Смотрите также

функцию rand,
которая создает тензор со случайными числами из равномерного распределения
функцию randn_like,
которая создает тензор из нормального распределения на основе формы другого тензора
функцию randint,
которая создает тензор со случайными целыми числами в заданном диапазоне
функцию normal,
которая создает тензор с числами из нормального распределения с заданными параметрами