|
- using System;
- using System.Collections.Generic;
- using System.Text;
-
- namespace Tensorflow
- {
- public class Session : BaseSession, IPython
- {
- private IntPtr _handle;
- public Status Status = new Status();
- public SessionOptions Options { get; }
- public Graph graph;
-
- public Session(string target = "", Graph graph = null)
- {
- if(graph == null)
- {
- graph = tf.get_default_graph();
- }
- this.graph = graph;
- Options = new SessionOptions();
- _handle = c_api.TF_NewSession(graph, Options, Status);
- Status.Check();
- }
-
- public Session(IntPtr handle)
- {
- _handle = handle;
- }
-
- public Session(Graph graph, SessionOptions opts, Status s = null)
- {
- if (s == null)
- s = Status;
- _handle = c_api.TF_NewSession(graph, opts, s);
- Status.Check(true);
- }
-
- public static implicit operator IntPtr(Session session) => session._handle;
- public static implicit operator Session(IntPtr handle) => new Session(handle);
-
- public void close()
- {
- Dispose();
- }
-
- public void Dispose()
- {
- Options.Dispose();
- Status.Dispose();
- c_api.TF_DeleteSession(_handle, Status);
- }
-
- public void __enter__()
- {
-
- }
-
- public void __exit__()
- {
-
- }
- }
- }
|