softmax.h File Reference

#include <assert.h>
#include "tensor.h"
#include "ops/softmax.h"

Include dependency graph for softmax.h:

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

Go to the source code of this file.

Functions
void	NN_softmax (Tensor *out, const Tensor *tensor, int dim)