この記事は数値計算Advent Calendar 2022の1日目の記事です。 前回の記事では既存実装としてNumPyにおけるnumpy.einsumの仕様を見ていきましたが、今回はRustのndarray crate向けにeinsumを作っていきます。完成形は次のようになります: use ndarray::array; use einsum_derive::einsum; let a = array![ [1.0, 2.0], [3.0, 4.0] ]; let b = array![ [1.0, 2.0], [3.0, 4.0] ]; let c = einsum!("ij,jk->ik", a, b); assert_eq!(c, array![ [6.0, 8.0], [12.0, 16.0] ]); このeinsum_derive crateは下記のリポジトリで開発されています:
![einsum!を作る](https://cdn-ak-scissors.b.st-hatena.com/image/square/69cbd57ce051538e4660c24dc6f0f2aead05a76a/height=288;version=1;width=512/https%3A%2F%2Fres.cloudinary.com%2Fzenn%2Fimage%2Fupload%2Fs--hv-T1hHc--%2Fc_fit%252Cg_north_west%252Cl_text%3Anotosansjp-medium.otf_72%3Aeinsum%252521%2525E3%252582%252592%2525E4%2525BD%25259C%2525E3%252582%25258B%252Cw_1010%252Cx_90%252Cy_100%2Fg_south_west%252Cl_text%3Anotosansjp-medium.otf_37%3Atermoshtt%252Cx_203%252Cy_121%2Fg_south_west%252Ch_90%252Cl_fetch%3AaHR0cHM6Ly9saDMuZ29vZ2xldXNlcmNvbnRlbnQuY29tL2EtL0FPaDE0R2lfdUM4LVBtaVF5cjN4UjBLdWMzN2t3ZDF2cGlsaUZWV25MbzhDS1E9czI1MC1j%252Cr_max%252Cw_90%252Cx_87%252Cy_95%2Fv1627283836%2Fdefault%2Fog-base-w1200-v2.png)