From 5afee8e79238ac4ea450d7ff42e5ac6ccede1b43 Mon Sep 17 00:00:00 2001
From: Leon Chen <leon@md.ai>
Date: Wed, 5 Oct 2016 18:52:01 -0400
Subject: [PATCH] fix Deconvolution2D layer outputShape

---
 src/layers/convolutional/Deconvolution2D.js | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/layers/convolutional/Deconvolution2D.js b/src/layers/convolutional/Deconvolution2D.js
index a0e0148..ed191b7 100644
--- a/src/layers/convolutional/Deconvolution2D.js
+++ b/src/layers/convolutional/Deconvolution2D.js
@@ -37,7 +37,11 @@ export default class Deconvolution2D extends Layer {
 
     this.kernelShape = [nbFilter, nbRow, nbCol]
 
-    this.outputShape = outputShape
+    if (outputShape[0] == null) {
+      this.outputShape = outputShape.slice(1)
+    } else {
+      this.outputShape = outputShape
+    }
 
     this.activation = activations[activation]