Update lambda1d.py

g0lemXIV · web-flow · commit 66df1e606ad0 · 2020-11-23T14:48:35.000+01:00
diff --git a/lambda_layers/lambda1d.py b/lambda_layers/lambda1d.py
@@ -156,7 +156,7 @@ def _validate_call_args(self, inputs):
                 "Given length: {}".format(class_name, len(inputs))
             )
         if len(inputs) > 1:
-            if inputs[0].shape != inputs[1].shape:
+            if inputs[0].shape[1:] != inputs[1].shape[1:]:
                 raise ValueError(
                     "{} layer should have same last dimension of inputs "
                     "and context tensor."

Original file line number	Diff line number	Diff line change
`@@ -156,7 +156,7 @@ def _validate_call_args(self, inputs):`
`156`	`156`	`"Given length: {}".format(class_name, len(inputs))`
`157`	`157`	`)`
`158`	`158`	`if len(inputs) > 1:`
`159`		`- if inputs[0].shape != inputs[1].shape:`
	`159`	`+ if inputs[0].shape[1:] != inputs[1].shape[1:]:`
`160`	`160`	`raise ValueError(`
`161`	`161`	`"{} layer should have same last dimension of inputs "`
`162`	`162`	`"and context tensor."`