Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Creating torch predict function

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-30 13:12:40 +02:00
parent 479c1119e7
commit 2f3c34b8b5
2 changed files with 43 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tools/gen_random_vector.py
View File

@@ -2,9 +2,11 @@ import numpy as np
import utils
from sys import argv
def gen_random_vector(size):
return np.random.rand(size)
if __name__ == "__main__":
if len(argv) < 2:
@@ -12,4 +14,4 @@ if __name__ == "__main__":
exit(1)
vector = gen_random_vector(int(argv[1]))
utils.print_cpp_vector(vector)
utils.print_cpp_vector(vector)