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

8
tools/softmax_test.py
View File

@@ -3,27 +3,33 @@ from utils import print_cpp_vector
def gen_softmax1_test_result():
# fmt: off
input = torch.tensor([
0.573, 0.619, 0.732, 0.055, 0.243
])
# fmt: on
output = torch.nn.Softmax(dim=0)(input)
print_cpp_vector(output)
def gen_softmax2_test_result():
# fmt: off
input = torch.tensor([
22.496, 36.9006, 30.9904, 28.4213, 26.4541, 31.7887
])
# fmt: on
output = torch.nn.Softmax(dim=0)(input)
print_cpp_vector(output)
def gen_softmax_exp():
# fmt: off
input = torch.tensor([
22.496, 36.9006, 30.9904, 28.4213, 26.4541, 31.7887
])
# fmt: on
output = torch.exp(input)
print_cpp_vector(output)
@@ -36,4 +42,4 @@ if __name__ == "__main__":
print("Softmax 2 test:")
gen_softmax2_test_result()
print("Softmax exp test:")
gen_softmax_exp()
gen_softmax_exp()