Skip to content

Commit

Permalink
WIP
Browse files Browse the repository at this point in the history
  • Loading branch information
darioizzo committed Oct 31, 2023
1 parent 0effd03 commit 67b8b44
Showing 1 changed file with 27 additions and 0 deletions.
27 changes: 27 additions & 0 deletions test/model_ffnn.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
// Copyright 2020, 2021, 2022, 2023 Francesco Biscani ([email protected]), Dario Izzo ([email protected])
//
// This file is part of the heyoka library.
//
// This Source Code Form is subject to the terms of the Mozilla
// Public License v. 2.0. If a copy of the MPL was not distributed
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.

#include <fmt/core.h>
#include <fmt/ranges.h>

#include <heyoka/config.hpp>
#include <heyoka/expression.hpp>
#include <heyoka/math.hpp>
#include <heyoka/model/ffnn.hpp>

#include "catch.hpp"

using namespace heyoka;

TEST_CASE("impl")
{
auto [x] = make_vars("x");
auto my_net = model::ffnn_impl({x}, 2, {2, 2}, {heyoka::tanh, heyoka::tanh, [](auto ret) { return ret; }},
{1_dbl, 2_dbl, 3_dbl, 4_dbl, 5_dbl, 6_dbl, 7_dbl, 8_dbl, 9_dbl, 0_dbl, 1_dbl, 2_dbl,
3_dbl, 4_dbl, 5_dbl, 6_dbl});
}

0 comments on commit 67b8b44

Please sign in to comment.