摘要: torch.cat的作用是对张量按行或列进行拼接。在使用的过程中我也碰到了一些需要注意的点。 import torch x = torch.rand((4,5)) print(x) tmp = torch.Tensor() for i in range(4): if i%2 == 0: tmp = 阅读全文
posted @ 2020-10-13 11:02 Kayden_Cheung 阅读(554) 评论(0) 推荐(0) 编辑
//目录