Функция PyTorch torch.full_like
Справочное руководство Pytorch torch
torch.full_like— это функция в PyTorch, используемая для создания заполненного тензора с той же формой, что и входной тензор.
Определение функции
torch.full_like(input, fill_value, dtype=None, device=None, requires_grad=False)
Пример использования
Пример
import torch
# Создайте входной тензор
input = torch.tensor([[1, 2], [3, 4]])
# Создать тензор из всех 7 с той же формой, что и вход
x = torch.full_like(input, 7)
print(x)
# Создайте входной тензор
input = torch.tensor([[1, 2], [3, 4]])
# Создать тензор из всех 7 с той же формой, что и вход
x = torch.full_like(input, 7)
print(x)
Другие расширения