Chain.cs

using System.Collections.Generic;
using System.Linq;

namespace chainer
{
    public class Chain : Link
    {
        protected Dictionary<string, Link> Children = new Dictionary<string, Link>();

        public Chain(Dictionary<string, Link> @children)
        {
            Children = @children;
        }

        public override IEnumerable<Variable> GetParams()
        {
            var selfParams = _Params.Values.AsEnumerable();
            var childrenParams = Children.Values.SelectMany(child => child.GetParams());
            return selfParams.Concat(childrenParams);
        }

        public override void Serialize(serializers.Serializer serializer)
        {
            base.Serialize(serializer);
            foreach (var kv in Children)
            {
                kv.Value.Serialize(serializer.Traverse(kv.Key));
            }
        }
    }
}