Investigate all the Places that Involve Shape or Dimension Evaluation

[DarrenZhang01]

Since the current shape Info for the Kernel object is stored inside an ndarray in the form like shape1 = ndarray<tf.Tensor([1024 2], shape=(2,), dtype=int64)>, there is currently error like #10, then there is possibility that there are chained problems caused by this shape evaluation that also leads to mis-evaluation on the dimensionality, etc. Shapes like ndarray<tf.Tensor([2], shape=(1,), dtype=int64)> looks more or less like a mistake caused by the nested eval, something like shape1.shape. Overall, need to investigate the places of shape evaluation and make revisions when necessary.

[DarrenZhang01]

Solved in 98be634, together with issues google/neural-tangents#52, #10 and #6