本文档会持续更新

1 变量类型转换

tf.cast(x, dtype, name=None)

将一个张量转换新类型,新类型由 dtype 参数指定。

Args:

  • x : 张量或稀疏张量
  • dtype : 目标类型, tf.DType
  • name : 操作名称 (optional)
  • Returns:返回一个类型为 dtype 、大小相同的张量或稀疏张量。
import tensorflow as tf
x = [1.0, 2, 7]
with tf.Session() as sess:
    # 等同于 tf.to_int32(x)
    y = tf.cast(x, tf.int32)
    print(y)
    #输出 Tensor("Cast:0", shape=(3,), dtype=int32)

参考:TensorFlow 数据类型转换