torch.randint_like

torch.randint_like(input, low=0, high, \*, dtype=None, layout=torch.strided, device=None, requires_grad=False, memory_format=torch.preserve_format) → Tensor

Возвращает тензор с такой же формой, что и тензор input, заполненный случайными целыми числами, сгенерированными равномерно между low (включительно) и high (исключительно).

Параметры:

• input (Тензор) – размер input определит размер выходного тензора.
• low (int, необязательно) – Наименьшее целое число, которое должно быть извлечено из распределения. Значение по умолчанию: 0.
• high (int) – Число, на единицу больше наибольшего целого числа, которое должно быть извлечено из распределения.

Ключевые аргументы:

• dtype (torch.dtype, optional) – желаемый тип данных возвращаемого тензора. Значение по умолчанию: если None, по умолчанию используется dtype тензора input.
• layout (torch.layout, optional) – желаемая структура возвращаемого тензора. Значение по умолчанию: если None, по умолчанию используется структура тензора input.
• device (torch.device, optional) – желаемое устройство возвращаемого тензора. Значение по умолчанию: если None, по умолчанию используется устройство тензора input.
• requires_grad (bool, необязательно) – Если автограду следует записывать операции над возвращаемым тензором. Значение по умолчанию: False.
• memory_format (torch.memory_format, optional) – желаемый формат памяти возвращаемого тензора. Значение по умолчанию: torch.preserve_format.