Функция torch.ravel в PyTorch
Справочное руководство Pytorch torch
torch.ravelЭто функция в PyTorch, используемая для выравнивания тензора в одно измерение. Она возвращает представление входного тензора (если возможно), а не копию.
Определение функции
torch.ravel(input)
Описание параметра:
input: входной тензор
Пример использования
Пример
import torch
# Создать матрицу 2x3
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# Уплощение до одного измерения
y = torch.ravel(x)
print(y)
# Создать матрицу 2x3
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# Уплощение до одного измерения
y = torch.ravel(x)
print(y)
Результат вывода:
tensor([1, 2, 3, 4, 5, 6])
Пример
import torch
# Создать трёхмерный тензор
x = torch.arange(12).reshape(2, 3, 2)
# Уплощение до одного измерения
y = torch.ravel(x)
print(y)
# Создать трёхмерный тензор
x = torch.arange(12).reshape(2, 3, 2)
# Уплощение до одного измерения
y = torch.ravel(x)
print(y)
Результат вывода:
tensor([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11])
Другие расширения