Add utils vector mean function

This commit is contained in:
2024-05-14 21:41:18 +02:00
parent 5c8d3f7e25
commit da8f3167cb
4 changed files with 71 additions and 9 deletions

10
tools/vector_test.py Normal file
View File

@@ -0,0 +1,10 @@
import torch
def gen_vector_mean_test_result():
input = torch.tensor([0.44371, 0.20253, 0.73232, 0.40378, 0.93348, 0.72756, 0.63388, 0.5251, 0.23973, 0.52233])
output = torch.mean(input)
print(output)
if __name__ == "__main__":
gen_vector_mean_test_result()