NeuralUtil.h File Reference

#include <Tensor.h>
#include <NeuralNet.h>

Go to the source code of this file.

Namespaces
namespace	NN
	NN Neural Network Library containing Tensor and NeuralNet classes for inference.
namespace	NN::Util
	Namespace for utility functions related to neural networks and tensors.

Functions
template<typename ReduceOp >
auto	NN::Util::bind_reduce (ReduceOp op, int axis)
	Binds a reduction operation on a specified axis for a tensor.
template<typename ReduceIndexOp >
auto	NN::Util::bind_reduceWithIndex (ReduceIndexOp op, int axis)
	Binds a reduction operation with index on a specified axis for a tensor.
template<typename Func >
auto	NN::Util::bind_map (Func f)
	Binds a mapping operation on a tensor.
template<typename Func >
auto	NN::Util::bind_map (Func f, int axis)
	Binds a mapping operation on a tensor that converts a each element to an array of new elements.
template<typename... Args>
auto	NN::Util::bind_reshape (const std::vector< int64_t > &newShape)
	Binds a reshape operation to change the shape of a tensor.
template<typename... Args>
auto	NN::Util::bind_squeeze ()
	Binds a squeeze operation to remove dimensions of size 1 from a tensor.
template<typename... Args>
auto	NN::Util::bind_squeeze (int64_t axis)
	Binds a squeeze operation on a tensor along a specified axis.
template<typename... Args>
auto	NN::Util::bind_unsqueeze (int64_t axis)
	Binds an unsqueeze operation to add a new dimension of size 1 at a specified axis in a tensor.
template<typename... Args>
auto	NN::Util::bind_toCvMat ()
	Binds a conversion operation from a tensor to an OpenCV Mat format.
template<typename T , typename... Args>
auto	NN::Util::bind_toVector ()
	Binds a conversion operation from a tensor to a vector.
template<typename... Args>
auto	NN::Util::bind_inference (NN::NeuralNetPtr model)
	Binds an inference operation on a neural network model.