mirror of
https://github.com/asg017/sqlite-vec.git
synced 2026-04-25 16:56:27 +02:00
23 lines
462 B
Python
23 lines
462 B
Python
|
import numpy as np
|
||
|
|
from io import BytesIO
|
||
|
|
|
||
|
|
|
||
|
|
def to_npy(arr):
|
||
|
|
buf = BytesIO()
|
||
|
|
np.save(buf, arr)
|
||
|
|
buf.seek(0)
|
||
|
|
return buf.read()
|
||
|
|
|
||
|
|
|
||
|
|
to_npy(np.array([[1.0, 2.0, 3.0], [2.0, 3.0, 4.0]], dtype=np.float32))
|
||
|
|
|
||
|
|
print(to_npy(np.array([[1.0, 2.0]], dtype=np.float32)))
|
||
|
|
print(to_npy(np.array([1.0, 2.0], dtype=np.float32)))
|
||
|
|
|
||
|
|
to_npy(
|
||
|
|
np.array(
|
||
|
|
[np.zeros(10), np.zeros(10), np.zeros(10), np.zeros(10), np.zeros(10)],
|
||
|
|
dtype=np.float32,
|
||
|
|
)
|
||
|
|
)
|