преобразовать список тензоров в тензоры - PullRequest
1 голос
/ 24 марта 2020

У меня есть список типа

xs = [tensor1, tensor2, tensor3....]

Я хочу изменить его на тензор, чтобы я мог передавать xs и ys в загрузчик данных Pytorch. Я попробовал это,

xs = torch.Tensor(xs)

, но это не работает, так как отдельные элементы не элементы, а тензоры.

1 Ответ

1 голос
/ 24 марта 2020

Вы можете захотеть torch.stack

xs = torch.stack(xs)
...