tf.cast()用法对应torch