argmax.h

Go to the documentation of this file.

#pragma once

#include "tensor/tensor.h"


namespace magmadnn {
namespace math {

template <typename T>
void argmax(Tensor<T> *x, int axis, Tensor<T> *out);

}
}