tensor_creation.h File Reference

#include <stddef.h>
#include <stdint.h>
#include <stdlib.h>
#include <stdio.h>
#include <assert.h>
#include <string.h>
#include "tensor.h"
#include "fill.h"

Include dependency graph for tensor_creation.h:

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

Go to the source code of this file.

Functions
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)