documentation_v0.1/html/functors_2broadcast_8hpp_source.html

// SPDX-License-Identifier: BSD-2-Clause


#ifndef pRC_CORE_FUNCTORS_BROADCAST_H

#define pRC_CORE_FUNCTORS_BROADCAST_H


#include <prc/core/basic/type_traits.hpp>

#include <prc/core/tensor/functions/broadcast.hpp>


namespace pRC

{

    template<Size... Bs>


    struct Broadcast

    {

        template<class X,

            If<True<decltype(broadcast<Bs...>(declval<X>()))>> = 0>


        constexpr decltype(auto) operator()(X &&a) const

        {

            return broadcast<Bs...>(forward<X>(a));

        }


    };


}

#endif // pRC_CORE_FUNCTORS_BROADCAST_H

type_traits.hpp

pRC
Definition cholesky.hpp:18

pRC::If
std::enable_if_t< B{}, int > If
Definition type_traits.hpp:68

pRC::Size
std::size_t Size
Definition type_traits.hpp:20

pRC::broadcast
static constexpr auto broadcast(X &&a)
Broadcasts a Tensor into one whose mode sizes are a multiple of the mode sizes of the original Tensor...
Definition broadcast.hpp:30

pRC::copy
static constexpr Conditional< IsSatisfied< C >, RemoveConstReference< X >, X > copy(X &&a)
Definition copy.hpp:13

pRC::Broadcast
Definition broadcast.hpp:13

pRC::True
Definition type_traits.hpp:38

broadcast.hpp