Files
CUDANet/tools/utils.py
2024-03-22 20:03:22 +01:00

8 lines
244 B
Python

def print_cpp_vector(vector):
print("std::vector<float> expected = {", end="")
for i in range(len(vector)):
if i != 0:
print(", ", end="")
print(str(round(vector[i].item(), 5)) + "f", end="")
print("};")