softmax.h

Go to the documentation of this file.

#ifndef __NN_SOFTMAX_H
#define __NN_SOFTMAX_H
 
#include <assert.h>
 
#include "tensor.h"
#include "ops/softmax.h"
 
 
void NN_softmax(Tensor *out, const Tensor *tensor, int dim);
 
 
#endif // __NN_SOFTMAX_H