Function graph_downsample

src/framework/audio/zipenhancer.cpp:727–750

}

core::TensorValue graph_downsample(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & input,
    const Param & bias) {
    const int64_t ds = static_cast<int64_t>(bias.values.size());
    if (ds <= 0) {
        throw std::runtime_error("ZipEnhancer downsample bias is empty");
    }
    const auto weights = softmax_values(bias);
    const int64_t out_s = (input.shape.dims[0] + ds - 1) / ds;
    std::vector<core::TensorValue> pieces;
    pieces.reserve(static_cast<size_t>(out_s));
    for (int64_t s = 0; s < out_s; ++s) {
        core::TensorValue acc;
        for (int64_t k = 0; k < ds; ++k) {
            const int64_t src_s = std::min(input.shape.dims[0] - 1, s * ds + k);
            auto slice = modules::SliceModule({0, src_s, 1}).build(ctx, input);
            slice = graph_scale(ctx, slice, weights[static_cast<size_t>(k)]);
            acc = acc.valid() ? graph_add(ctx, acc, slice) : slice;
        }
        pieces.push_back(acc);
    }
    return graph_concat_all(ctx, pieces, 0);
}

core::TensorValue graph_upsample_slice(
    core::ModuleBuildContext & ctx,