tensor_ops.hpp File Reference

Go to the source code of this file.

Namespaces
	bnmf_algs
	Main namespace for bnmf-algs library.
	bnmf_algs::cuda
	cuda namespace contains functions that operate on nvidia GPUs using CUDA routines.

Functions
template<typename T >
void	bnmf_algs::cuda::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. More...