1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 11:40:18 +01:00

Add Flatten layer to architectures + make flatten optional in embedders (#483)

Flatten layer required for embedders that mix conv and dense
(Cherry picking from #478)
This commit is contained in:
Guy Jacob
2021-05-12 11:11:10 +03:00
committed by GitHub
parent c369984c2e
commit 235a259223
7 changed files with 49 additions and 10 deletions

View File

@@ -76,3 +76,11 @@ class NoisyNetDense(object):
def __str__(self):
return "Noisy Dense (num outputs = {})".format(self.units)
class Flatten(object):
"""
Base class for framework specific flatten layer (used to convert 3D convolution output to 1D dense input)
"""
def __str__(self):
return "Flatten"