今天写代码的时候发现torch中stack函数和cat函数在处理列表转tensor使用方法不一样，记录一下
首先是list1列表，这种列表里面是tensor([1])，这种是有维度得用cat函数，结果:
《《《《《《《《《《《《《《《《《《《《《》》》》》》》》》》》》》》》》》》》
而对list2这种列表，里面是tensor(1),这种是没有维度的，就使用stack函数就行，结果：
版权声明：本文为qq_46040952原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。