bnmf-algs
tensor_ops.hpp
Go to the documentation of this file.
1 #pragma once
2 
3 #include "cuda/memory.hpp"
4 #include "defs.hpp"
5 #include <array>
6 
7 namespace bnmf_algs {
12 namespace cuda {
13 
40 template <typename T>
41 void tensor_sums(const DeviceMemory3D<T>& tensor,
42  std::array<DeviceMemory2D<T>, 3>& result_arr);
43 
44 } // namespace cuda
45 } // namespace bnmf_algs
void tensor_sums(const DeviceMemory3D< T > &tensor, std::array< DeviceMemory2D< T >, 3 > &result_arr)
Sum the given 3D input tensor along each of its axes and return all 2D sum tensors.
Main namespace for bnmf-algs library.
Definition: alloc_model_funcs.hpp:12