仅作为记录,大佬请跳过。
直接上代码
import torch
tensor1=torch.tensor([1,2,3])
tensor2=torch.tensor([4,5,6])
tensor_list=[]
tensor_list.append(tensor1)
tensor_list.append(tensor2)
final_tensor=torch.stack(tensor_list,0)
print(tensor_list)
print(type(tensor_list))
print(final_tensor)
print(type(final_tensor))
展示
参考
感谢大佬博主文章——传送门