12 ms·Iirc pytorch's einsum used to be very slowby savant_penguin 5y agoIirc pytorch's einsum used to be very slow