{ "layer_shapes" : { "x" : { "k" : 1, "w" : 128, "n" : 1, "_rank" : 2, "h" : 1 }, "linear_0" : { "k" : 1, "w" : 1167, "n" : 1, "_rank" : 2, "h" : 1 }, "var_8" : { "k" : 1, "w" : 1167, "n" : 1, "_rank" : 2, "h" : 1 } } }