BaseFunction.h

class MLFunction : public exec::VectorFunction {
public:
    virtual ~MLFunction() = default;

    virtual float* getTensor() const = 0;

    virtual std::vector<int> getDims() {
        return dims;
    }

    virtual std::string getFuncName() {
        return "";
    }

    virtual int getNumDims() {
        return dims.size();
    }

    virtual CostEstimate getCost(std::vector<int> inputDims) {
        return CostEstimate(0, inputDims[0], inputDims[1]);
    }
 
protected:
    std::vector<int> dims; 

    double getWeightedCost(std::string name, float cost) {
        std::vector<double> coefficient =
            UdfCostCoefficient::getInstance().getCoefficient(name);
        // FIXME: Implement weighted cost calculation.
        return 0;
    }

    std::vector<double> getCoefficientVector(std::string name) {
        return UdfCostCoefficient::getInstance().getCoefficient(name);
    }
};