Функция torch.rand в PyTorch
Справочное руководство Pytorch torch
torch.randЭто функция в PyTorch, используемая для создания случайных тензоров из равномерного распределения [0, 1).
Обычно используется в глубоком обучении для таких сценариев, как инициализация, генерация случайных масок и т. д.
Определение функции
torch.rand(*size, dtype=None, device=None, requires_grad=False)
Parameters:
*size(int): Форма тензора.dtype(torch.dtype, необязательно): Тип данных.device(torch.device, необязательно): Устройство.requires_grad(bool, необязательно): Нужно ли вычислять градиент.
Возвращаемое значение:
torch.Tensor: Возвращает тензор, содержащий случайные числа в диапазоне [0, 1).
Примеры использования
Пример 1: Создать случайный тензор
Пример
import torch
# Создать случайный тензор 3x4
x = torch.rand(3, 4)
print(x)
print("Диапазон: [{:.4f}, {:.4f})".format(x.min().item(), x.max().item()))
# Создать случайный тензор 3x4
x = torch.rand(3, 4)
print(x)
print("Диапазон: [{:.4f}, {:.4f})".format(x.min().item(), x.max().item()))
Результат вывода:
tensor([[0.6272, 0.1802, 0.8336, 0.3900],
[0.2188, 0.6593, 0.3598, 0.4021],
[0.9247, 0.0144, 0.2587, 0.6634]])
范围: [0.0144, 0.9247)
Пример 2: Создать случайную маску
Пример
import torch
# Создать случайную маску с вероятностью 50%
mask = torch.rand(10) > 0.5
print("Mask:", mask)
print("True count:", mask.sum().item())
# Создать случайную маску с вероятностью 50%
mask = torch.rand(10) > 0.5
print("Mask:", mask)
print("True count:", mask.sum().item())
Другие расширения