define cut of tf.slice via tf.math.argmin

Question

I want a neural network that takes as input a bunch of lists of positive integers of varying lengths. For each list there is also some infos given by an 8 number list.

I