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

Add model predict test

Browse Source
This commit is contained in:
Matúš Námešný
2024-03-22 22:31:32 +01:00
parent 9ecd51c63a
commit 9482d7bc43
5 changed files with 145 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
tools/softmax_tes.py
View File

@@ -1,17 +0,0 @@
import torch
from utils import print_cpp_vector
def gen_softmax_test_result():
input = torch.tensor([
0.573, 0.619, 0.732, 0.055, 0.243
])
output = torch.nn.Softmax(dim=0)(input)
print_cpp_vector(output)
if __name__ == "__main__":
print("Generating test results...")
print("Softmax test:")
gen_softmax_test_result()

39
tools/softmax_test.py Normal file
View File

@@ -0,0 +1,39 @@
import torch
from utils import print_cpp_vector
def gen_softmax1_test_result():
input = torch.tensor([
0.573, 0.619, 0.732, 0.055, 0.243
])
output = torch.nn.Softmax(dim=0)(input)
print_cpp_vector(output)
def gen_softmax2_test_result():
input = torch.tensor([
22.496, 36.9006, 30.9904, 28.4213, 26.4541, 31.7887
])
output = torch.nn.Softmax(dim=0)(input)
print_cpp_vector(output)
def gen_softmax_exp():
input = torch.tensor([
22.496, 36.9006, 30.9904, 28.4213, 26.4541, 31.7887
])
output = torch.exp(input)
print_cpp_vector(output)
if __name__ == "__main__":
print("Generating test results...")
print("Softmax 1 test:")
gen_softmax1_test_result()
print("Softmax 2 test:")
gen_softmax2_test_result()
print("Softmax exp test:")
gen_softmax_exp()