Verwendung einer Variablen für num_splits für tf.split ()

Ist es möglich, eine Platzhaltereingabe für den Parameter num_split zu verwenden, um tf.split () zu verwenden?

Ich möchte im Idealfall etwas machen wie:

num_splits = tf.placeholder(tf.int32)
inputs = tf.placeholder(tf.int32, [5, None])
split_inputs = tf.split(1, num_splits, inputs)


TypeError: Expected int for argument 'num_split' not <tensorflow.python.framework.ops.Tensor object at 0x10b819ad0>.

Der Ansatz stimmt möglicherweise nicht. Ich möchte eine Dimension in einem Tensor mit variabler Form aufzählen. Vielen Dank

Antworten auf die Frage(2)

Ihre Antwort auf die Frage