Skip to content

Commit 0e545aa

Browse files
committed
Merge pull request #816 from iamalbert/fix-clearstate
prevent SelectTable and NarrowTable clears input
2 parents 0614529 + 2441fc6 commit 0e545aa

File tree

3 files changed

+5
-1
lines changed

3 files changed

+5
-1
lines changed

NarrowTable.lua

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,3 +39,5 @@ function NarrowTable:type(type, tensorCache)
3939
self.gradInput = {}
4040
return parent.type(self, type, tensorCache)
4141
end
42+
43+
NarrowTable.clearState = nn.Identity.clearState

SelectTable.lua

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,3 +58,5 @@ end
5858
function SelectTable:__tostring__()
5959
return torch.type(self) .. '(' .. self.index .. ')'
6060
end
61+
62+
SelectTable.clearState = nn.Identity.clearState

init.lua

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,7 @@ require('nn.VolumetricAveragePooling')
125125
require('nn.VolumetricBatchNormalization')
126126

127127
require('nn.ParallelTable')
128+
require('nn.Identity')
128129
require('nn.ConcatTable')
129130
require('nn.SplitTable')
130131
require('nn.JoinTable')
@@ -133,7 +134,6 @@ require('nn.MixtureTable')
133134
require('nn.CriterionTable')
134135
require('nn.FlattenTable')
135136
require('nn.NarrowTable')
136-
require('nn.Identity')
137137

138138
require('nn.Criterion')
139139
require('nn.MSECriterion')

0 commit comments

Comments
 (0)