2022年9月21日
PyTorch：stack + reshape与cat之间的异同
摘要： value_query = torch.stack([prev_query, now_query], 1) .reshape(bs*2, num_query, -1).permute(1, 0, 2) value_query1 = torch.stack([prev_query, now_query 阅读全文
posted @ 2022-09-21 20:59 龙雪 阅读(2) 评论(0) 推荐(0) 编辑