1 | /* |
---|

2 | Copyright (C) 2018 Paul Brossier <piem@aubio.org> |
---|

3 | |
---|

4 | This file is part of aubio. |
---|

5 | |
---|

6 | aubio is free software: you can redistribute it and/or modify |
---|

7 | it under the terms of the GNU General Public License as published by |
---|

8 | the Free Software Foundation, either version 3 of the License, or |
---|

9 | (at your option) any later version. |
---|

10 | |
---|

11 | aubio is distributed in the hope that it will be useful, |
---|

12 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
---|

13 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
---|

14 | GNU General Public License for more details. |
---|

15 | |
---|

16 | You should have received a copy of the GNU General Public License |
---|

17 | along with aubio. If not, see <http://www.gnu.org/licenses/>. |
---|

18 | |
---|

19 | */ |
---|

20 | |
---|

21 | #ifndef AUBIO_CONV1D_H |
---|

22 | #define AUBIO_CONV1D_H |
---|

23 | |
---|

24 | #ifdef __cplusplus |
---|

25 | extern "C" { |
---|

26 | #endif |
---|

27 | |
---|

28 | typedef struct _aubio_conv1d_t aubio_conv1d_t; |
---|

29 | |
---|

30 | /** create a new conv1d layer */ |
---|

31 | aubio_conv1d_t *new_aubio_conv1d(uint_t filters, uint_t kernel_shape[1]); |
---|

32 | |
---|

33 | /** perform forward 1D convolution */ |
---|

34 | void aubio_conv1d_do(aubio_conv1d_t *t, aubio_tensor_t *input_tensor, |
---|

35 | aubio_tensor_t *activations); |
---|

36 | |
---|

37 | /** TODO: implement */ |
---|

38 | void aubio_conv1d_train(aubio_conv1d_t *t, aubio_tensor_t *input_tensor); |
---|

39 | |
---|

40 | /** get conv1d weights */ |
---|

41 | aubio_tensor_t *aubio_conv1d_get_kernel(aubio_conv1d_t *t); |
---|

42 | |
---|

43 | /** get conv1d biases */ |
---|

44 | fvec_t *aubio_conv1d_get_bias(aubio_conv1d_t *t); |
---|

45 | |
---|

46 | /** set conv1d stride */ |
---|

47 | uint_t aubio_conv1d_set_stride(aubio_conv1d_t *c, uint_t stride[1]); |
---|

48 | |
---|

49 | uint_t aubio_conv1d_set_padding_mode(aubio_conv1d_t *c, |
---|

50 | const char_t *padding_mode); |
---|

51 | |
---|

52 | uint_t aubio_conv1d_get_output_shape(aubio_conv1d_t *t, |
---|

53 | aubio_tensor_t *input_tensor, uint_t *shape); |
---|

54 | |
---|

55 | void del_aubio_conv1d(aubio_conv1d_t *t); |
---|

56 | |
---|

57 | #ifdef __cplusplus |
---|

58 | } |
---|

59 | #endif |
---|

60 | |
---|

61 | #endif /* AUBIO_CONV1D_H */ |
---|