获取MatMul计算所需空间