![]() |
IgANet
IgANets - Isogeometric Analysis Networks
|
Linear algebra utility functions. More...
#include <torch/torch.h>
Go to the source code of this file.
Namespaces | |
namespace | iganet |
namespace | iganet::utils |
Functions | |
template<short_t dim = 0, typename T0 , typename T1 > | |
auto | iganet::utils::dotproduct (T0 &&t0, T1 &&t1) |
Computes the directional dot-product between two tensors with summation along the given dimension. | |
template<short_t dim = 0, typename T , typename... Ts> | |
auto | iganet::utils::kronproduct (T &&t, Ts &&...ts) |
Computes the directional Kronecker-product between two or more tensors along the given dimension. | |
template<short_t dim = 0, typename T0 , typename T1 > | |
auto | iganet::utils::kronproduct (T0 &&t0, T1 &&t1) |
Computes the directional Kronecker-product between two tensors along the given dimension. | |
template<typename T , std::size_t N> | |
T | iganet::utils::prod (std::array< T, N > array, std::size_t start_index=0, std::size_t stop_index=N - 1) |
Computes the (partial) product of all std::array entries. | |
template<typename T , std::size_t N> | |
T | iganet::utils::sum (std::array< T, N > array, std::size_t start_index=0, std::size_t stop_index=N - 1) |
Computes the (partial) sum of all std::array entries. | |
Linear algebra 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/.