Method embedding

cactus/graph/graph_builder.cpp:1440–1458

}

size_t CactusGraph::embedding(size_t embedding_tensor, size_t indices) {
    const auto& emb_buffer = get_output_buffer(embedding_tensor);
    const auto& idx_shape = get_output_buffer(indices).shape;

    if (emb_buffer.shape.size() != 2) {
        std::cerr << "Error: Embedding tensor " << embedding_tensor << " has invalid shape: [";
        for(auto d : emb_buffer.shape) std::cerr << d << ",";
        std::cerr << "]. OpType=" << (int)nodes_[node_index_map_[embedding_tensor]]->op_type
                  << " ExtData=" << emb_buffer.external_data << std::endl;
        throw std::runtime_error("Embedding tensor must be 2D [vocab_size, hidden_dim]");
    }

    std::vector<size_t> output_shape = idx_shape;
    output_shape.push_back(emb_buffer.shape[1]);

    OpParams params;
    params.output_precision = Precision::FP16;
    return add_node(OpType::EMBEDDING, {embedding_tensor, indices}, output_shape, params);
}

size_t CactusGraph::bilinear_interpolation(size_t pos_embeds, size_t dst_height, size_t dst_width, bool align_corners) {
    const auto& pos_embeds_buffer = get_output_buffer(pos_embeds);