Split python scripts

This commit is contained in:
2024-03-22 20:03:22 +01:00
parent 87db47089e
commit 9ecd51c63a
7 changed files with 89 additions and 62 deletions

17
tools/softmax_tes.py Normal file
View File

@@ -0,0 +1,17 @@
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()