Method InferShape

operator/operator/fully_connected.cpp:29–77

namespace TEngine {

bool FullyConnected::InferShape(const std::vector<TEngine::TShape>& ishape, std::vector<TEngine::TShape>& oshape,
                                int layout)
{
    const TShape& input = ishape[0];
    const TShape& weight = ishape[1];

    std::vector<int> dim;

    int n = weight.Shape(0);
    int k = weight.Shape(1);

    int m = input.Shape(0);
    int input_k = input.Shape(1);

    if(input.GetDim().size() == 2)
    {
        dim = {m, n};
    }
    else if(input.GetDim().size() == 3)
    {
        input_k *= input.Shape(2);
        if(layout == TENGINE_LAYOUT_NHWC)
            dim = {m, 1, n};
        else
            dim = {m, n, 1};
    }
    else if(input.GetDim().size() == 4)
    {
        input_k *= input.Shape(2) * input.Shape(3);
        if(layout == TENGINE_LAYOUT_NHWC)
            dim = {m, 1, 1, n};
        else
            dim = {m, n, 1, 1};
    }
    else
        return false;

    if(k != input_k)
        return false;

    TShape shape;

    shape.SetDim(dim);
    shape.SetDataLayout(layout);

    oshape[0] = shape;

    return true;
}

float FullyConnected::GetFops(const std::vector<TShape>& inputs, const std::vector<TShape>& outputs)
{