mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-05 17:34:21 +00:00
17 lines
356 B
Python
17 lines
356 B
Python
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() |