tensor_creation.h

Go to the documentation of this file.

#ifndef __NN_TENSOR_CREATION
#define __NN_TENSOR_CREATION
 
#include <stddef.h>
#include <stdint.h>
#include <stdlib.h>
#include <stdio.h>
#include <assert.h>
#include <string.h>
 
#include "tensor.h"
#include "fill.h"
 
 
void NN_init_tensor(Tensor *tensor, const size_t ndim, const size_t *shape, DataType dtype, void *data);
 
Tensor *NN_tensor(size_t ndim, const size_t *shape, DataType dtype, void *data);
 
Tensor *NN_zeros(size_t ndim, const size_t *shape, DataType dtype);
 
Tensor *NN_ones(size_t ndim, const size_t *shape, DataType dtype);
 
Tensor *NN_rand(size_t ndim, const size_t *shape, DataType dtype);
 
Tensor *NN_arange(size_t ndim, const size_t *shape, DataType dtype, float start, float step);
 
 
#endif // __NN_TENSOR_CREATION