We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents ababd95 + c6c150b commit d20fe64Copy full SHA for d20fe64
keras/layers/core.py
@@ -1744,7 +1744,7 @@ def set_layer_input(self, head):
1744
def get_output_at(self, head, train=False):
1745
X = self.inputs[head].get_output(train)
1746
mask = self.inputs[head].get_output_mask(train)
1747
- Y = self.layer(X, mask)
+ Y = self.layer(X, mask=mask, train=train)
1748
return Y
1749
1750
def get_output_shape(self, head, train=False):
0 commit comments