Make tensorfunction field pub

Signed-off-by: Matthias Wahl <[email protected]>
mfelsche · Oct 8, 2020 · 999ebd8 · 999ebd8
1 parent bc60232
commit 999ebd8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/common/activations.rs b/src/common/activations.rs
@@ -26,7 +26,7 @@ pub fn _tanh(x: &Tensor) -> Tensor {
     x.tanh()
 }
 
-pub struct TensorFunction(Box<fn(&Tensor) -> Tensor>);
+pub struct TensorFunction(pub Box<fn(&Tensor) -> Tensor>);
 
 impl std::fmt::Debug for TensorFunction {
     fn fmt(&self, f: &mut std::fmt::Formatter) -> Result<(), std::fmt::Error> {