![]() |
IgANet
IgANets - Isogeometric Analysis Networks
|
Tuple utility functions. More...
#include <vector>
#include <tuple>
Go to the source code of this file.
Namespaces | |
namespace | iganet |
namespace | iganet::utils |
Functions | |
template<typename... Tensors, typename Func > | |
torch::Tensor | iganet::utils::cat_tuple_into_tensor (const std::tuple< Tensors... > &tensors, Func &&func, int64_t dim=0) |
Concatenates the entries of an std::tuple object into a single Torch tensor along the given dimension after applying the callback function. | |
template<typename... Tensors> | |
torch::Tensor | iganet::utils::cat_tuple_into_tensor (const std::tuple< Tensors... > &tensors, int64_t dim=0) |
Concatenates the entries of an std::tuple object into a single Torch tensor along the given dimension. | |
template<std::size_t N, typename T > | |
constexpr auto | iganet::utils::repeat_tuple (const T &value) |
Returns an std::tuple object with N replications of the given value. | |
Tuple utility functions.
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/.