argmax.cpp File Reference

#include "math/argmax.h"

Include dependency graph for argmax.cpp:

Functions
template<typename T >
void	magmadnn::math::argmax (Tensor< T > *x, int axis, Tensor< T > *out)
template void	magmadnn::math::argmax (Tensor< int > *x, int axis, Tensor< int > *out)
template void	magmadnn::math::argmax (Tensor< float > *x, int axis, Tensor< float > *out)
template void	magmadnn::math::argmax (Tensor< double > *x, int axis, Tensor< double > *out)

Detailed Description

Author

Daniel Nichols

Version

0.1

Date

2019-06-11

Copyright

Copyright (c) 2019

Function Documentation

argmax()

template<typename T >

void magmadnn::math::argmax	(	Tensor< T > *	x,
		int	axis,
		Tensor< T > *	out
	)

Returns the argmax along the specified axis.

Template Parameters

T

Parameters

x
axis
out