pRC
multi-purpose Tensor Train library for C++
Loading...
Searching...
No Matches
squeeze.hpp
Go to the documentation of this file.
1// SPDX-License-Identifier: BSD-2-Clause
2
3#ifndef pRC_CORE_FUNCTORS_SQUEEZE_H
4#define pRC_CORE_FUNCTORS_SQUEEZE_H
5
7
8namespace pRC
9{
10 struct Squeeze
11 {
13 constexpr decltype(auto) operator()(X &&a) const
14 {
15 return squeeze(forward<X>(a));
16 }
17 };
18}
19#endif // pRC_CORE_FUNCTORS_SQUEEZE_H
Definition cholesky.hpp:18
static constexpr Conditional< IsSatisfied< C >, RemoveConstReference< X >, X > copy(X &&a)
Definition copy.hpp:13
static constexpr auto squeeze(X &&a)
Remove axes with mode size 1 from a Tensor.
Definition squeeze.hpp:23
Definition squeeze.hpp:11