forked from SciSharp/TensorFlow.NET
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathSaveSpec.cs
More file actions
32 lines (27 loc) · 804 Bytes
/
SaveSpec.cs
File metadata and controls
32 lines (27 loc) · 804 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
using System;
using System.Collections.Generic;
using System.Text;
namespace Tensorflow
{
/// <summary>
/// Class used to describe tensor slices that need to be saved.
/// </summary>
public class SaveSpec
{
private Tensor _tensor;
public Tensor tensor => _tensor;
private string _slice_spec;
public string slice_spec => _slice_spec;
private string _name;
public string name => _name;
private TF_DataType _dtype;
public TF_DataType dtype => _dtype;
public SaveSpec(Tensor tensor, string slice_spec, string name, TF_DataType dtype = TF_DataType.DtInvalid)
{
_tensor = tensor;
_slice_spec = slice_spec;
_name = name;
_dtype = dtype;
}
}
}