Функция PyTorch torch.randint_like

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

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

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

torch.randint_like(input, low=0, high, dtype=None, device=None, requires_grad=False)

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

Пример

import torch

# Создайте входной тензор
input = torch.tensor([[1, 2], [3, 4]])

# Создать случайный целочисленный тензор той же формы, что и вход, диапазон [0, 10)
x = torch.randint_like(input, 10)

print(x)

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

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