1. tf.splite()
tf.split(
value,
num_or_size_splits,
axis=0,
num=None,
name='split'
)
split0, split1, split2 = tf.split(value, [4, 15, 11], 1)
tf.shape(split0) # [5, 4]
tf.shape(split1) # [5, 15]
tf.shape(split2) # [5, 11]
tf.split(
value,
num_or_size_splits,
axis=0,
num=None,
name='split'
)
split0, split1, split2 = tf.split(value, [4, 15, 11], 1)
tf.shape(split0) # [5, 4]
tf.shape(split1) # [5, 15]
tf.shape(split2) # [5, 11]