forked from SciSharp/TensorFlow.NET
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathvariable_utils.cs
More file actions
33 lines (32 loc) · 985 Bytes
/
variable_utils.cs
File metadata and controls
33 lines (32 loc) · 985 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
33
using System;
using System.Collections.Generic;
using System.Text;
using Tensorflow.Framework;
namespace Tensorflow.Util
{
internal static class variable_utils
{
public static Tensor[] convert_variables_to_tensors(object[] values)
{
return values.Select(x =>
{
if (resource_variable_ops.is_resource_variable(x))
{
return ops.convert_to_tensor(x);
}
else if (x is CompositeTensor)
{
throw new NotImplementedException("The composite tensor has not been fully supported.");
}
else if(x is Tensor tensor)
{
return tensor;
}
else
{
throw new TypeError("Currently the output of function to be traced must be `Tensor`.");
}
}).ToArray();
}
}
}