mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 01:34:22 +00:00
18 lines
324 B
Python
18 lines
324 B
Python
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:
|
|
print("Usage: python gen_random_vector.py <size>")
|
|
exit(1)
|
|
|
|
vector = gen_random_vector(int(argv[1]))
|
|
utils.print_cpp_vector(vector)
|