Loading

torch.cat(x, dim)使用小技巧

import torch
a = [torch.arange(12).reshape(3, 4)]
print(a)
b = torch.cat(a, dim=0)
print(b)

使用torch.cat去除[]

posted @ 2021-03-09 15:42  Guang'Jun  阅读(812)  评论(0)    收藏  举报