From db191149442a087e16d23847e488463739af6066 Mon Sep 17 00:00:00 2001 From: Ivan Razumov Date: Wed, 29 Nov 2023 17:21:37 +0100 Subject: [PATCH] Code-format --- PhysicsTools/PythonAnalysis/test/time_serie_prediction.cpp | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/PhysicsTools/PythonAnalysis/test/time_serie_prediction.cpp b/PhysicsTools/PythonAnalysis/test/time_serie_prediction.cpp index 604743c9b5f3d..efe4b7fb6c1ea 100644 --- a/PhysicsTools/PythonAnalysis/test/time_serie_prediction.cpp +++ b/PhysicsTools/PythonAnalysis/test/time_serie_prediction.cpp @@ -2,13 +2,10 @@ #include struct Net : torch::nn::Module { - Net(int64_t N, int64_t M) - : linear(register_module("linear", torch::nn::Linear(N, M))) { + Net(int64_t N, int64_t M) : linear(register_module("linear", torch::nn::Linear(N, M))) { another_bias = register_parameter("b", torch::randn(M)); } - torch::Tensor forward(torch::Tensor input) { - return linear(input) + another_bias; - } + torch::Tensor forward(torch::Tensor input) { return linear(input) + another_bias; } torch::nn::Linear linear; torch::Tensor another_bias; };