mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 09:44:28 +00:00
10 lines
276 B
Python
10 lines
276 B
Python
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])
|
|
output = torch.mean(input)
|
|
|
|
print(output)
|
|
|
|
if __name__ == "__main__":
|
|
gen_vector_mean_test_result() |