def dynamic_partition(data, partitions, num_partitions, name=None):
return tf.map_fn(
lambda args: tf.dynamic_partition(*args, num_partitions, name=name),
[data, partitions])
微信
公众号
扫码关注公众号