Функция torch.full в PyTorch

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.full— это функция в PyTorch, используемая для создания тензора, заполненного указанным значением.

Определение функции

torch.full(size, fill_value, dtype=None, device=None, requires_grad=False)

Пример использования

Пример

import torch

# Создать тензор 3x4 со значением заполнения 7
x = torch.full((3, 4), 7)

print(x)

Результат вывода:

tensor([[7., 7., 7., 7.],
        [7., 7., 7., 7.],
        [7., 7., 7., 7.]])

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения