• X
    [Einsum] Einsum support repeated labels. (#47290) · 6e1c14e3
    xiongkun 提交于
    * add unittest for einsum-v2-trace and diagonal
    
    * repeat labels.
    
    * einsum support repeated labels.
    
    * forward is ok for diagonal and undiagonalized.
    TODO: check backward is ok by our theorem.
    
    * backward is ok!
    
    * fix by PR suggestions.
    
    * fix ci error
    
    * fix ci error
    
    * fix ci warning
    6e1c14e3
diagonal_grad_kernel.cc 3.2 KB