使用opt_einsum和PyTorch FX图形重写进行Einsum优化。 该库当前支持: 将多个枚举融合为一个使用库优化einsum 用标量常量融合乘法和除法,包括通过与einsum之类的运算 ...
確定! 回上一頁