ndarray的宏Einsum

设置

安装python和opt_einsum包。

使用方法

语法类似于numpy。对于每个输入参数，使用(expr).(axes)。然后使用粗箭头=>，对于输出轴使用.(axes)和分号;。分号之后，您需要按照轴的顺序传递每个轴的维度，以便宏在编译时优化路径。请参阅下面的示例

let y: Array<f64, _> = einsum!(a.mi, b.nj, c.ijpl, d.op, e.ql => .mnoq; i 2, j 2, l 10, m 40, n 40, o 8, p 7, q 6);