Einsum

功能

Einsum算子使用爱因斯坦求和约定评估tensor序列上的代数tensor运算。公式字符串包含逗号分隔的小写字母序列。每个项对应于操作数tensor，项中的字符对应于操作数维度。

输入

1 ~ ∞个输入：

x：一个tensor，数据类型：float16、float、int32。

输出

一个输出：

y：一个tensor，数据类型：float16、float、int32。

属性

equation：爱因斯坦求和的下标，一个和表达式字符串。

约束

目前onnx的einsum只支持双输入的20种情况：

abc,cde->abde,"abc,cde->abde","abcd,aecd->aceb","abcd,adbe->acbe","abcd,cde->abe","abc,cd->abd","abc,dc->abd","abc,abd->dc","abc,dec->abde","abc,abde->dec","abcd,aecd->acbe","abcd,acbe->aecd","abcd,ecd->abe","abcd,abe->ecd","abcd,acbe->adbe","abcd,abde->abce","abcd,abce->abde","abcd,aebd->aebc","abcd,abce->acde","abc,abd->acd","ab,cb->ac"。

支持的ONNX版本

支持的ONNX版本Opset v12/v13/v14/v15/v16/v17/v18