god-graph 0.6.0-alpha

A graph-based LLM white-box optimization toolbox: topology validation, Lie group orthogonalization, tensor ring compression
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

//! 网格图生成器

use crate::graph::builders::GraphBuilder;
use crate::graph::Graph;

/// 生成 2D 网格图
///
/// # 参数
/// * `rows` - 行数
/// * `cols` - 列数
/// * `diagonal` - 是否包含对角线连接
///
/// # 返回
/// 生成的网格图
pub fn grid_graph<T>(rows: usize, cols: usize, diagonal: bool) -> Graph<T, f64>
where
    T: Clone + Default,
{
    let n = rows * cols;
    let mut builder = GraphBuilder::undirected().with_nodes((0..n).map(|_| T::default()));

    for r in 0..rows {
        for c in 0..cols {
            let idx = r * cols + c;

            // 向右连接
            if c + 1 < cols {
                builder = builder.with_edge(idx, idx + 1, 1.0);
            }

            // 向下连接
            if r + 1 < rows {
                builder = builder.with_edge(idx, idx + cols, 1.0);
            }

            // 对角线连接
            if diagonal {
                if r + 1 < rows && c + 1 < cols {
                    builder = builder.with_edge(idx, idx + cols + 1, 1.0);
                }
                if r + 1 < rows && c > 0 {
                    builder = builder.with_edge(idx, idx + cols - 1, 1.0);
                }
            }
        }
    }

    builder.build().unwrap_or_else(|_| Graph::undirected())
}