Commit d72c0e89 authored by Benjamin Thomas Graham's avatar Benjamin Thomas Graham
Browse files

fix Torch SparseToDense

parent 13bdf051
...@@ -221,7 +221,7 @@ void scn_ARCH_REAL_DIMENSIONMaxPooling_updateGradInput( ...@@ -221,7 +221,7 @@ void scn_ARCH_REAL_DIMENSIONMaxPooling_updateGradInput(
// SparseToDense // SparseToDense
void scn_ARCH_REAL_DIMENSIONSparseToDense_updateOutput( void scn_ARCH_REAL_DIMENSIONSparseToDense_updateOutput(
THLongTensor *inputSize, void **m, THTensor *input_features, THLongTensor *inputSize, void **m, THTensor *input_features,
THTensor *output_features, THITensor *rulesBuffer); THTensor *output_features, THITensor *rulesBuffer, long nPlanes);
void scn_ARCH_REAL_DIMENSIONSparseToDense_updateGradInput( void scn_ARCH_REAL_DIMENSIONSparseToDense_updateGradInput(
THLongTensor *inputSize, void **m, THTensor *input_features, THLongTensor *inputSize, void **m, THTensor *input_features,
THTensor *d_input_features, THTensor *d_output_features, THTensor *d_input_features, THTensor *d_output_features,
......
...@@ -36,7 +36,7 @@ return function(sparseconvnet) ...@@ -36,7 +36,7 @@ return function(sparseconvnet)
input.features:cdata(), input.features:cdata(),
self.output:cdata(), self.output:cdata(),
self.shared.rulesBuffer and self.shared.rulesBuffer:cdata(), self.shared.rulesBuffer and self.shared.rulesBuffer:cdata(),
self.nPlanes or input.features.size(2)) self.nPlanes or input.features:size(2))
return self.output return self.output
end end
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment