{ "layer_shapes" : { "x" : { "k" : 1, "w" : 15, "n" : 1, "_rank" : 1, "h" : 1 }, "linear_0" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "var_6" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 } } }