tf.cast https://blog.csdn.net/luoganttcc/article/details/70315538 https://blog.csdn.net/zj360202/article/details/70260265 tf.count_nonzero