Tensorflow 使用大小不可被输出张量数整除的张量拆分
我正在尝试并行化我的tensorflow代码。在某一点上，我需要运行这个来拆分张量Tensorflow 使用大小不可被输出张量数整除的张量拆分,tensorflow,tensorflow-gpu,Tensorflow,Tensorflow Gpu,我正在尝试并行化我的tensorflow代码。在某一点上，我需要运行这个来拆分张量 for variable_name, variable in kwargs.items(): input_variables_split[variable_name] = tf.split(variable, number_of_devices) 这段代码的问题是tf.split期望所有变量的第0轴都能被设备的数量除以，在我的例子中，这有时可能不是真的。解决这个问题的最好办法是什
for variable_name, variable in kwargs.items():
input_variables_split[variable_name] = tf.split(variable, number_of_devices)