Browse Source

Keras layer activations fix

tags/yolov3
MPnoy Haiping 4 years ago
parent
commit
5801566a8d
2 changed files with 6 additions and 2 deletions
  1. +3
    -1
      src/TensorFlowNET.Keras/Activations/Activations.Sigmoid.cs
  2. +3
    -1
      src/TensorFlowNET.Keras/Activations/Activations.Tanh.cs

+ 3
- 1
src/TensorFlowNET.Keras/Activations/Activations.Sigmoid.cs View File

@@ -17,7 +17,9 @@ namespace Tensorflow.Keras
return results[0];
}

throw new NotImplementedException("");
var _op = tf.OpDefLib._apply_op_helper("Sigmoid", name: name, args: new { x = features });

return _op.output;
};
}
}

+ 3
- 1
src/TensorFlowNET.Keras/Activations/Activations.Tanh.cs View File

@@ -17,7 +17,9 @@ namespace Tensorflow.Keras
return results[0];
}

throw new NotImplementedException("");
var _op = tf.OpDefLib._apply_op_helper("Tanh", name: name, args: new { x = features });

return _op.output;
};
}
}

Loading…
Cancel
Save