TypeError: Input 'split_dim' of 'Split' Op has type float32 that does not match expected type of int

TypeError: Input 'split_dim' of 'Split' Op has type float32 that does not match expected type of int32.

1. TypeError: Input 'split_dim' of 'Split' Op has type float32 that does not match expected type of int32.

TensorFlow 版本不同，使用函数接口修改导致的问题。

API r1.3

tf.split

split(    value,    num_or_size_splits,    axis=0,    num=None,    name='split')

http://blog.csdn.net/chengyq116/article/details/78504173

API r0.12

tf.split

tf.split(split_dim, num_split, value, name='split')

http://blog.csdn.net/chengyq116/article/details/78543244