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

Run black autoformatter

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-30 13:18:51 +02:00
parent 2f3c34b8b5
commit fcd7d96126
7 changed files with 110 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
tools/vector_test.py
View File

@@ -1,10 +1,25 @@
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])
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()
gen_vector_mean_test_result()