Address the uniformity issue of tf.stack() #10
Description
tf.stack() assumes all tensors are the same shape and dtype — no check or fallback. If this is allowed to stay a single different sized tensor object can crash the entire code. Address this issue by devising a solution to handle mutli-size tensor layers.