Function replicate_pad1d

src/framework/audio/flashsr.cpp:209–234

}

core::TensorValue replicate_pad1d(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & input,
    int64_t left,
    int64_t right) {
    if (left < 0 || right < 0) {
        throw std::runtime_error("FlashSR replicate padding must be non-negative");
    }
    auto output = input;
    constexpr int axis = 2;
    if (left > 0) {
        auto edge = modules::SliceModule({axis, 0, 1}).build(ctx, input);
        auto pad_shape = edge.shape;
        pad_shape.dims[axis] = left;
        auto pad = modules::RepeatModule({pad_shape}).build(ctx, edge);
        output = modules::ConcatModule({axis}).build(ctx, pad, output);
    }
    if (right > 0) {
        auto edge = modules::SliceModule({axis, input.shape.dims[axis] - 1, 1}).build(ctx, input);
        auto pad_shape = edge.shape;
        pad_shape.dims[axis] = right;
        auto pad = modules::RepeatModule({pad_shape}).build(ctx, edge);
        output = modules::ConcatModule({axis}).build(ctx, output, pad);
    }
    return output;
}

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