FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [Original HTTPS Page]

feat: add Nestable and Nest<T>. · chuckbeasley/TensorFlow.NET@4939105 · GitHub

/ TensorFlow.NET Public
forked from SciSharp/TensorFlow.NET

Commit 4939105

Browse files
SanftMonster
committed
feat: add Nestable and Nest<T>.
1 parent 46e190d commit 4939105

12 files changed

Lines changed: 946 additions & 124 deletions

File tree

‎src/TensorFlowNET.Core/Common/Extensions/LinqExtensions.cs‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,5 +22,12 @@ public static Tensors ToTensors(this IEnumerable<Tensor> tensors)
2222
{
2323
return new Tensors(tensors);
2424
}
25+
26+
public static void Deconstruct<T1, T2, T3>(this (T1, T2, T3) values, out T1 first, out T2 second, out T3 third)
27+
{
28+
first = values.Item1;
29+
second = values.Item2;
30+
third = values.Item3;
31+
}
2532
}
2633
}

‎src/TensorFlowNET.Core/Common/Extensions/NestExtensions.cs‎

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
using System;
2+
using System.Collections.Generic;
3+
using System.Text;
4+
using Tensorflow.Common.Types;
5+
6+
namespace Tensorflow.Common.Extensions
7+
{
8+
public static class NestExtensions
9+
{
10+
public static Tensors ToTensors(this INestable<Tensor> tensors)
11+
{
12+
return new Tensors(tensors.AsNest());
13+
}
14+
15+
public static Tensors? ToTensors(this Nest<Tensor> tensors)
16+
{
17+
return Tensors.FromNest(tensors);
18+
}
19+
20+
/// <summary>
21+
/// If the nested object is already a nested type, this function could reduce it.
22+
/// For example, `Nest[Nest[T]]` can be reduced to `Nest[T]`.
23+
/// </summary>
24+
/// <typeparam name="TIn"></typeparam>
25+
/// <typeparam name="TOut"></typeparam>
26+
/// <param name="input"></param>
27+
/// <returns></returns>
28+
public static Nest<TOut> ReduceTo<TIn, TOut>(this INestStructure<TIn> input) where TIn: INestStructure<TOut>
29+
{
30+
return Nest<TOut>.ReduceFrom(input);
31+
}
32+
}
33+
}

‎src/TensorFlowNET.Core/Common/Types/GeneralizedTensorShape.cs‎

Lines changed: 52 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
namespace Tensorflow.Common.Types
77
{
8-
public class GeneralizedTensorShape: IEnumerable<long?[]>
8+
public class GeneralizedTensorShape: IEnumerable<long?[]>, INestStructure<long?>, INestable<long?>
99
{
1010
public TensorShapeConfig[] Shapes { get; set; }
1111
/// <summary>
@@ -63,6 +63,57 @@ public Shape[] ToShapeArray()
6363
return Shapes.Select(x => new Shape(x.Items.Select(y => y is null ? -1 : y.Value).ToArray())).ToArray();
6464
}
6565

66+
public IEnumerable<long?> Flatten()
67+
{
68+
List<long?> result = new List<long?>();
69+
foreach(var shapeConfig in Shapes)
70+
{
71+
result.AddRange(shapeConfig.Items);
72+
}
73+
return result;
74+
}
75+
public INestStructure<TOut> MapStructure<TOut>(Func<long?, TOut> func)
76+
{
77+
List<Nest<TOut>> lists = new();
78+
foreach(var shapeConfig in Shapes)
79+
{
80+
lists.Add(new Nest<TOut>(shapeConfig.Items.Select(x => new Nest<TOut>(func(x)))));
81+
}
82+
return new Nest<TOut>(lists);
83+
}
84+
85+
public Nest<long?> AsNest()
86+
{
87+
Nest<long?> DealWithSingleShape(TensorShapeConfig config)
88+
{
89+
if (config.Items.Length == 0)
90+
{
91+
return Nest<long?>.Empty;
92+
}
93+
else if (config.Items.Length == 1)
94+
{
95+
return new Nest<long?>(config.Items[0]);
96+
}
97+
else
98+
{
99+
return new Nest<long?>(config.Items.Select(x => new Nest<long?>(x)));
100+
}
101+
}
102+
103+
if(Shapes.Length == 0)
104+
{
105+
return Nest<long?>.Empty;
106+
}
107+
else if(Shapes.Length == 1)
108+
{
109+
return DealWithSingleShape(Shapes[0]);
110+
}
111+
else
112+
{
113+
return new Nest<long?>(Shapes.Select(s => DealWithSingleShape(s)));
114+
}
115+
}
116+
66117
public IEnumerator<long?[]> GetEnumerator()
67118
{
68119
foreach (var shape in Shapes)

‎src/TensorFlowNET.Core/Common/Types/INest.cs‎

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
1+
using System;
2+
using System.Collections.Generic;
3+
using System.Text;
4+
5+
namespace Tensorflow.Common.Types
6+
{
7+
/// <summary>
8+
/// This interface indicates that a class may have a nested structure and provide
9+
/// methods to manipulate with the structure.
10+
/// </summary>
11+
public interface INestStructure<T>: INestable<T>
12+
{
13+
/// <summary>
14+
/// Flatten the Nestable object. Node that if the object contains only one value,
15+
/// it will be flattened to an enumerable with one element.
16+
/// </summary>
17+
/// <returns></returns>
18+
IEnumerable<T> Flatten();
19+
/// <summary>
20+
/// Construct a new object with the same nested structure.
21+
/// </summary>
22+
/// <typeparam name="TOut"></typeparam>
23+
/// <param name="func"></param>
24+
/// <returns></returns>
25+
INestStructure<TOut> MapStructure<TOut>(Func<T, TOut> func);
26+
}
27+
}

‎src/TensorFlowNET.Core/Common/Types/INestable.cs‎

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
using System;
2+
using System.Collections.Generic;
3+
using System.Text;
4+
5+
namespace Tensorflow.Common.Types
6+
{
7+
public interface INestable<T>
8+
{
9+
Nest<T> AsNest();
10+
}
11+
}

‎src/TensorFlowNET.Core/Common/Types/Nest.Static.cs‎

Lines changed: 62 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,62 @@
1+
using System;
2+
using System.Collections.Generic;
3+
using System.Text;
4+
5+
namespace Tensorflow.Common.Types
6+
{
7+
public static class Nest
8+
{
9+
/// <summary>
10+
/// Pack the flat items to a nested sequence by the template.
11+
/// </summary>
12+
/// <typeparam name="T"></typeparam>
13+
/// <param name="template"></param>
14+
/// <param name="flatItems"></param>
15+
/// <returns></returns>
16+
public static Nest<T> PackSequenceAs<T>(INestable<T> template, T[] flatItems)
17+
{
18+
return template.AsNest().PackSequence(flatItems);
19+
}
20+
21+
/// <summary>
22+
/// Pack the flat items to a nested sequence by the template.
23+
/// </summary>
24+
/// <typeparam name="T"></typeparam>
25+
/// <param name="template"></param>
26+
/// <param name="flatItems"></param>
27+
/// <returns></returns>
28+
public static Nest<T> PackSequenceAs<T>(INestable<T> template, List<T> flatItems)
29+
{
30+
return template.AsNest().PackSequence(flatItems.ToArray());
31+
}
32+
33+
/// <summary>
34+
/// Flatten the nested object.
35+
/// </summary>
36+
/// <typeparam name="T"></typeparam>
37+
/// <param name="nestedObject"></param>
38+
/// <returns></returns>
39+
public static IEnumerable<T> Flatten<T>(INestable<T> nestedObject)
40+
{
41+
return nestedObject.AsNest().Flatten();
42+
}
43+
44+
/// <summary>
45+
/// Map the structure with specified function.
46+
/// </summary>
47+
/// <typeparam name="TIn"></typeparam>
48+
/// <typeparam name="TOut"></typeparam>
49+
/// <param name="func"></param>
50+
/// <param name="nestedObject"></param>
51+
/// <returns></returns>
52+
public static INestStructure<TOut> MapStructure<TIn, TOut>(Func<TIn, TOut> func, INestable<TIn> nestedObject)
53+
{
54+
return nestedObject.AsNest().MapStructure(func);
55+
}
56+
57+
public static bool IsNested<T>(INestable<T> obj)
58+
{
59+
return obj.AsNest().IsNested();
60+
}
61+
}
62+
}

0 commit comments

Comments
 (0)
Back | FazBrowse Home | New Git URL