{ "layer_shapes" : { "expanded_b" : { "k" : 1, "w" : 32, "n" : 1, "_rank" : 4, "h" : 5 }, "output1" : { "k" : 1, "w" : 32, "n" : 1, "_rank" : 2, "h" : 5 }, "flattened_b" : { "k" : 160, "w" : 1, "n" : 1, "_rank" : 4, "h" : 1 }, "input_1" : { "k" : 1, "w" : 160, "n" : 1, "_rank" : 1, "h" : 1 } } }