search

hikettei / cl-waffe2

[Experimental] Graph and Tensor Abstraction for Deep Learning all in Common Lisp
https://hikettei.github.io/cl-waffe2/
MIT License
122 stars 5 forks source link

[Add] AvgPool2D MaxPool2D #61

Closed hikettei closed 11 months ago

hikettei commented 11 months ago
  1. added reshapers:
;; x                ... ( 3 3 ) Tensor
;; (!sum x :axis 1) ... ( 3 1 ) Tensor

;; broadcast-to will return: (t `(:broadcast 3))
(!mul x (!view (!sum x :axis 1) (broadcast-to x)))
  1. Added AvgPool2D/MaxPool2D

TODO: backpropagate of !min isn't working. backprops of Pooling2D aren't working due to some issues on !permute