Skip to content

Commit

Permalink
Add a compute_output_shape() in BackwardMerge
Browse files Browse the repository at this point in the history
raise a NotImplementedError for now.
  • Loading branch information
nhuet authored and ducoffeM committed Jan 17, 2024
1 parent 0560b9a commit 58e9242
Showing 1 changed file with 13 additions and 0 deletions.
13 changes: 13 additions & 0 deletions src/decomon/backward_layers/backward_merge.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,6 +100,19 @@ def call(self, inputs: List[BackendTensor], **kwargs: Any) -> List[List[BackendT
"""
pass

def compute_output_shape(self, input_shape: List[Tuple[Optional[int], ...]]) -> List[Tuple[Optional[int], ...]]:
"""Compute expected output shape according to input shape
Will be called by symbolic calls on Keras Tensors.
Args:
input_shape
Returns:
"""
raise NotImplementedError()

def build(self, input_shape: List[Tuple[Optional[int], ...]]) -> None:
"""
Args:
Expand Down

0 comments on commit 58e9242

Please sign in to comment.