2021年3月4日
einsum爱因斯坦求和
摘要： 最近因为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 嶙羽 阅读(314) 评论(0) 推荐(0) 编辑