Как преобразовать список или массив numpy в тензор 1D torch?

У меня есть список (или массив NumPy) значений float. Я хочу создать тензор Факела 1d, который будет содержать все эти значения. Я могу создать тензор Факела и запустить цикл для хранения значений.

но я хочу знать, есть ли способ, я могу создать факел тензора с начальными значениями из списка или массива? Также предложите мне, если есть какой-либо питонический способ достичь этого, поскольку я работаю в pytorch.

6
pytorch torch

1 ответов

Это общие операции в pytorch и доступны в документации. PyTorch позволяет легко взаимодействовать с numpy. Существует метод под названием from_numpy и документация здесь

import numpy as np 
import torch 
array = np.arange(1, 11)
tensor = torch.from_numpy(array)
12