摘要: 最近因为vision transformer里的pytorch代码,看到了torch.einsum(np.einsum同理)这个操作,简直是神了; 比如 t = torch.randn(2,4,3) q, k, v = tuple(rearrange(t, 'b t (d k) -> k b t d 阅读全文
posted @ 2021-03-04 13:31 嶙羽 阅读(332) 评论(0) 推荐(0) 编辑