Answers for "pytorch concatenate list of tensors"

5

torch concat matrix

third_tensor = torch.cat((first_tensor, second_tensor), 0) # keep column width append in rows

third_tensor = torch.cat((first_tensor, second_tensor), 1) # keep row height and append in columns
Posted by: Guest on September-28-2020
-1

how can I covert a list of tensor into tensor?

l = list(torch.tensor([1,2,3]))
print(l)
>>>[tensor(1), tensor(2), tensor(3)]
k = torch.stack(l)
print(k)
>>>tensor([1, 2, 3])
Posted by: Guest on November-12-2020

Code answers related to "pytorch concatenate list of tensors"

Python Answers by Framework

Browse Popular Code Answers by Language