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