Функция torch.randint в PyTorch

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.randintЭто функция в PyTorch, используемая для создания случайных целочисленных тензоров.

Определение функции

torch.randint(low, high, size, dtype, device, requires_grad)

Примеры использования

Пример

import torch

# Создать случайный целочисленный тензор от 0 до 9
x = torch.randint(0, 10, (3, 4))

print(x)

Результат вывода:

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

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения