Ковнерт - список тензоров к тензору - PullRequest
Новая
       149

Ковнерт - список тензоров к тензору

0 голосов
/ 05 августа 2020

У меня есть список таких тензоров: 

[tensor(-2.9222, grad_fn=<SqueezeBackward1>), tensor(-2.8192, grad_fn=<SqueezeBackward1>), tensor(-3.1894, grad_fn=<SqueezeBackward1>), tensor(-2.9048, grad_fn=<SqueezeBackward1>)]

Я хочу, чтобы он был в такой форме: 

tensor([-0.5575, -0.9004, -0.8491,  ..., -0.7345, -0.6729, -0.7553],
   grad_fn=<SqueezeBackward1>)

Как я могу это сделать? Я ценю помощь.

1 Ответ

1 голос
/ 06 августа 2020

Так как этот тензор является 0-мерным, torch.cat не будет работать, но вы можете использовать torch.stack (который создает новое измерение для объединения): 

a = torch.tensor(1.0, requires_grad=True)
b = torch.tensor(2.0, requires_grad=True)
torch.stack([a,b], dim=0)
>>>tensor([1.,2.], grad_fn=<StackBackward>)
...