Einsum算子使用爱因斯坦求和约定评估张量序列上的代数张量运算。公式字符串包含逗号分隔的小写字母序列。每个项对应于操作数张量,项中的字符对应于操作数维度。
【输入】
1 ~ ∞
x:一个tensor,数据类型:int8、int16、int32、int64、uint8、uint16、uint32、uint64、float16、float32
【输出】
一个输出
y:一个tensor,数据类型:int8、int16、int32、int64、uint8、uint16、uint32、uint64、float16、float32
【属性】
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版本Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18