Files
greptimedb/tests/cases/standalone/common/function/vector/vector.sql
Sicong Hu 09dacc8e9b feat: add vec_subvector function (#5683)
* feat: add vec_subvector function

* change datatype of arg1 and arg2 from u64 to i64

* add sqlness test

* improve description comments
2025-03-16 10:43:53 +00:00

125 lines
3.3 KiB
SQL

SELECT vec_to_string(parse_vec('[1.0, 2.0]'));
SELECT vec_to_string(parse_vec('[1.0, 2.0, 3.0]'));
SELECT vec_to_string(parse_vec('[]'));
SELECT vec_to_string(vec_add('[1.0, 2.0]', '[3.0, 4.0]'));
SELECT vec_to_string(vec_add(parse_vec('[1.0, 2.0]'), '[3.0, 4.0]'));
SELECT vec_to_string(vec_add('[1.0, 2.0]', parse_vec('[3.0, 4.0]')));
SELECT vec_to_string(vec_mul('[1.0, 2.0]', '[3.0, 4.0]'));
SELECT vec_to_string(vec_mul(parse_vec('[1.0, 2.0]'), '[3.0, 4.0]'));
SELECT vec_to_string(vec_mul('[1.0, 2.0]', parse_vec('[3.0, 4.0]')));
SELECT vec_to_string(vec_sub('[1.0, 1.0]', '[1.0, 2.0]'));
SELECT vec_to_string(vec_sub('[-1.0, -1.0]', '[1.0, 2.0]'));
SELECT vec_to_string(vec_sub('[1.0, 1.0]', parse_vec('[1.0, 2.0]')));
SELECT vec_to_string(vec_sub('[-1.0, -1.0]', parse_vec('[1.0, 2.0]')));
SELECT vec_to_string(vec_sub(parse_vec('[1.0, 1.0]'), '[1.0, 2.0]'));
SELECT vec_to_string(vec_sub(parse_vec('[-1.0, -1.0]'), '[1.0, 2.0]'));
SELECT vec_elem_sum('[1.0, 2.0, 3.0]');
SELECT vec_elem_sum('[-1.0, -2.0, -3.0]');
SELECT vec_elem_sum(parse_vec('[1.0, 2.0, 3.0]'));
SELECT vec_elem_sum(parse_vec('[-1.0, -2.0, -3.0]'));
SELECT vec_to_string(vec_div('[1.0, 2.0]', '[3.0, 4.0]'));
SELECT vec_to_string(vec_div(parse_vec('[1.0, 2.0]'), '[3.0, 4.0]'));
SELECT vec_to_string(vec_div('[1.0, 2.0]', parse_vec('[3.0, 4.0]')));
SELECT vec_to_string(vec_div('[1.0, -2.0]', parse_vec('[0.0, 0.0]')));
SELECT vec_elem_product('[1.0, 2.0, 3.0, 4.0]');
SELECT vec_elem_product('[-1.0, -2.0, -3.0, 4.0]');
SELECT vec_elem_product(parse_vec('[1.0, 2.0, 3.0, 4.0]'));
SELECT vec_elem_product(parse_vec('[-1.0, -2.0, -3.0, 4.0]'));
SELECT vec_to_string(vec_norm('[0.0, 2.0, 3.0]'));
SELECT vec_to_string(vec_norm('[1.0, 2.0, 3.0]'));
SELECT vec_to_string(vec_norm('[7.0, 8.0, 9.0]'));
SELECT vec_to_string(vec_norm('[7.0, -8.0, 9.0]'));
SELECT vec_to_string(vec_norm(parse_vec('[7.0, -8.0, 9.0]')));
SELECT vec_to_string(vec_sum(v))
FROM (
SELECT '[1.0, 2.0, 3.0]' AS v
UNION ALL
SELECT '[-1.0, -2.0, -3.0]' AS v
UNION ALL
SELECT '[4.0, 5.0, 6.0]' AS v
);
SELECT vec_to_string(vec_product(v))
FROM (
SELECT '[1.0, 2.0, 3.0]' AS v
UNION ALL
SELECT '[-1.0, -2.0, -3.0]' AS v
UNION ALL
SELECT '[4.0, 5.0, 6.0]' AS v
);
SELECT vec_dim('[7.0, 8.0, 9.0, 10.0]');
SELECT v, vec_dim(v)
FROM (
SELECT '[1.0, 2.0, 3.0]' AS v
UNION ALL
SELECT '[-1.0]' AS v
UNION ALL
SELECT '[4.0, 5.0, 6.0]' AS v
) Order By vec_dim(v) ASC;
SELECT v, vec_dim(v)
FROM (
SELECT '[1.0, 2.0, 3.0]' AS v
UNION ALL
SELECT '[-1.0]' AS v
UNION ALL
SELECT '[7.0, 8.0, 9.0, 10.0]' AS v
) Order By vec_dim(v) ASC;
SELECT vec_to_string(vec_subvector('[1.0,2.0,3.0,4.0,5.0]', 0, 3));
SELECT vec_to_string(vec_subvector('[1.0,2.0,3.0,4.0,5.0]', 5, 5));
SELECT v, vec_to_string(vec_subvector(v, 3, 5))
FROM (
SELECT '[1.0, 2.0, 3.0, 4.0, 5.0]' AS v
UNION ALL
SELECT '[-1.0, -2.0, -3.0, -4.0, -5.0, -6.0]' AS v
UNION ALL
SELECT '[4.0, 5.0, 6.0, 10, -8, 100]' AS v
) ORDER BY v;
SELECT vec_to_string(vec_subvector(v, 0, 5))
FROM (
SELECT '[1.1, 2.2, 3.3, 4.4, 5.5]' AS v
UNION ALL
SELECT '[-1.1, -2.1, -3.1, -4.1, -5.1, -6.1]' AS v
UNION ALL
SELECT '[4.0, 5.0, 6.0, 10, -8, 100]' AS v
) ORDER BY v;