sub.h

Go to the documentation of this file.

#ifndef __NN_SUB_H
#define __NN_SUB_H
 
#include <assert.h>
#include <math.h>
 
#include "tensor.h"
#include "print.h"
#include "ops/sub.h"
 
 
void NN_sub(Tensor *out, const Tensor *a, const Tensor *b);
 
void NN_sub_1d(Tensor *out, const Tensor *a, const Tensor *b);
 
void NN_sub_2d(Tensor *out, const Tensor *a, const Tensor *b);
 
void NN_sub_3d(Tensor *out, const Tensor *a, const Tensor *b);
 
void NN_sub_4d(Tensor *out, const Tensor *a, const Tensor *b);
 
 
#endif // __NN_SUB_H