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.