Loading

2021年3月9日
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) 推荐(0)