TensorFlow 張量投射

tf.cast

cast (  
    x ,  
    dtype ,  
    name = None
  )

定義在：tensorflow/python/ops/math_ops.py

參見(jiàn)指南：張量變換>鑄造

將張量投射到新的類型.

該操作將 x(在張量的情況下)或 x.values(在 SparseTensor 情況下)投射到 dtype.

例如：

＃tensor`a` is [1.8,2.2],dtype = tf.float  
tf.cast(a, tf.int32 ) == >  [ 1 , 2 ]   #dtype = tf.int32

ARGS：

• x：一個(gè)張量或 SparseTensor.
• dtype：目標(biāo)類型.
• name：操作的名稱(可選).

返回：

與 x 具有相同形狀的張量或 SparseTensor.

注意：

• TypeError：當(dāng) x 不能投射到 dtype.