我正在尝试并行化我的tensorflow代码。有一次，我需要运行它来分割张量。
for variable_name, variable in kwargs.items():
input_variables_split[variable_name] = tf.split(variable, number_of_devices)
这段代码的问题是tf.split期望所有变量的第0轴都可以被number_of_devices整除，在我的情况下有时可能不正确。解决这个问题的最佳方法是什么？