source: src/ai/tensor.h @ 4cb5496

/*
  Copyright (C) 2018 Paul Brossier <piem@aubio.org>

  This file is part of aubio.

  aubio is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  aubio is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with aubio.  If not, see <http://www.gnu.org/licenses/>.

*/

#ifndef AUBIO_TENSOR_H
#define AUBIO_TENSOR_H

#ifdef __cplusplus
extern "C" {
#endif

typedef struct {
  uint_t n_dims;
  uint_t dims[8];
  smpl_t **data;
  uint_t n_items;
  uint_t items_per_row;
} aubio_tensor_t;

aubio_tensor_t *new_aubio_tensor(uint_t n_dims, uint_t *dims);

void del_aubio_tensor(aubio_tensor_t *c);

uint_t aubio_tensor_as_fvec(aubio_tensor_t *c, fvec_t *o);
uint_t aubio_fvec_as_tensor(fvec_t *o, aubio_tensor_t *c);

uint_t aubio_tensor_as_fmat(aubio_tensor_t *c, fmat_t *o);
uint_t aubio_fmat_as_tensor(fmat_t *o, aubio_tensor_t *c);

smpl_t aubio_tensor_max(aubio_tensor_t *t);

#define AUBIO_ASSERT_EQUAL_SHAPE(t1, t2) { \
    AUBIO_ASSERT(t1 && t2); \
    AUBIO_ASSERT(t1->n_dims == t2->n_dims); \
    uint_t nn; \
    for (nn = 0; nn < t1->n_dims; nn++) \
      AUBIO_ASSERT(t1->dims[nn] == t2->dims[nn]); \
    }

#ifdef __cplusplus
}
#endif

#endif /* AUBIO_TENSOR_H */