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 (Tensor) – размер input определит размер выходного тензора.
• low (int, необязательно) – Наименьшее целое число для выборки из распределения. По умолчанию: 0.
• high (int) – Значение, на единицу больше наибольшего целого числа для выборки из распределения.

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

• dtype (torch.dtype, optional) – желаемый тип данных возвращаемого тензора. По умолчанию: если None, по умолчанию равен типу данных 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.