argmax.h File Reference

#include "tensor/tensor.h"

Include dependency graph for argmax.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
template<typename T >
void	magmadnn::math::argmax (Tensor< T > *x, int axis, Tensor< T > *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