torch.full_like

torch.full_like(input, fill_value, \*, dtype=None, layout=torch.strided, device=None, requires_grad=False, memory_format=torch.preserve_format) → Tensor

Возвращает тензор с такими же размерами, как input, заполненный значением fill_value. torch.full_like(input, fill_value) эквивалентно torch.full(input.size(), fill_value, dtype=input.dtype, layout=input.layout, device=input.device).

Параметры:

• input (Тензор) – размер input определит размер выходного тензора.
• fill_value – число, которым будет заполнен выходной тензор.

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

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