From d3e6441c4086c0ba0e3d9a274c94543b0c940f28 Mon Sep 17 00:00:00 2001
From: Connor Olding <cloningdonor@gmail.com>
Date: Sat, 9 Jun 2018 16:20:20 +0200
Subject: [PATCH] reduce tile input to 5 per row using new layers

---
 main.lua | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/main.lua b/main.lua
index 82e1ba0..82b4b64 100644
--- a/main.lua
+++ b/main.lua
@@ -141,14 +141,18 @@ package.loaded['nn'] = nil -- DEBUG
 local nn = require("nn")
 
 local network
-local nn_x, nn_tx, nn_ty, nn_y, nn_z
+local nn_x, nn_tx, nn_ty, nn_tz, nn_y, nn_z
 local function make_network(input_size)
     nn_x = nn.Input({input_size})
     nn_tx = nn.Input({gcfg.tile_count})
     nn_ty = nn_tx:feed(nn.Embed(#game.valid_tiles, 2))
+    nn_tz = nn_ty:feed(nn.Reshape{13, 17 * 2})
+    nn_tz = nn_tz:feed(nn.DenseBroadcast(5))
+    nn_tz = nn_tz:feed(nn.Relu())
+    -- note: due to a quirk in Merge, we don't need to flatten nn_tz.
     nn_y = nn.Merge()
     nn_x:feed(nn_y)
-    nn_ty:feed(nn_y)
+    nn_tz:feed(nn_y)
 
     --[[
     nn_y = nn_y:feed(nn.Dense(128))