mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 01:34:22 +00:00
Add utils vector mean function
This commit is contained in:
10
tools/vector_test.py
Normal file
10
tools/vector_test.py
Normal 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()
|
||||
Reference in New Issue
Block a user