Функция full
Функция full создает тензор заданной формы и заполняет
его указанным значением. Первым параметром функция принимает
размерность тензора (кортеж или список), вторым параметром -
значение для заполнения. Также можно указать тип данных элементов
с помощью параметра dtype.
Синтаксис
torch.full(size, fill_value, [dtype])
Пример
Давайте создадим одномерный тензор из 5 элементов,
заполненный числом 7:
import torch
t = torch.full((5,), 7)
print(t)
Результат выполнения кода:
tensor([7, 7, 7, 7, 7])
Пример
Давайте создадим двумерный тензор размера 2 на 3,
заполненный числом 5:
import torch
t = torch.full((2, 3), 5)
print(t)
Результат выполнения кода:
tensor([
[5, 5, 5],
[5, 5, 5],
])
Пример
Давайте создадим тензор с вещественным типом данных float:
import torch
t = torch.full((2, 3), 3.14, dtype=torch.float32)
print(t)
Результат выполнения кода:
tensor([
[3.1400, 3.1400, 3.1400],
[3.1400, 3.1400, 3.1400],
])
Пример
Давайте создадим трехмерный тензор размера 2 на 2 на 2,
заполненный значением 9:
import torch
t = torch.full((2, 2, 2), 9)
print(t)
Результат выполнения кода:
tensor([
[
[9, 9],
[9, 9],
],
[
[9, 9],
[9, 9],
],
])
Пример
Давайте создадим тензор с логическим типом данных bool:
import torch
t = torch.full((3,), True, dtype=torch.bool)
print(t)
Результат выполнения кода:
tensor([True, True, True])
