tensorflow.cast(x, dtype, name=None)使用记录

最新推荐文章于 2022-02-09 12:00:30 发布

原创最新推荐文章于 2022-02-09 12:00:30 发布 · 767 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #ml

tensorflow 专栏收录该内容

8 篇文章

订阅专栏

tensorflow 中执行类型转换

将变量x转换为dtype类型的新变量（x为序列，则对序列内元素转换）

example:

import tensorflow as tf

a = tf.Variable([True,False,True])

b = tf.cast(a,dtype=tf.float32)

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print(sess.run(b))
#[1. 0. 1.]

tensorflow help info:

'''
Help on function cast in module tensorflow.python.ops.math_ops:

cast(x, dtype, name=None)
    Casts a tensor to a new type.
    
    The operation casts `x` (in case of `Tensor`) or `x.values`
    (in case of `SparseTensor`) to `dtype`.
    
    For example:
    
    ```python
    x = tf.constant([1.8, 2.2], dtype=tf.float32)
    tf.cast(x, tf.int32)  # [1, 2], dtype=tf.int32
    ```
    
    Args:
      x: A `Tensor` or `SparseTensor`.
      dtype: The destination type.
      name: A name for the operation (optional).
    
    Returns:
      A `Tensor` or `SparseTensor` with same shape as `x`.
    
    Raises:
      TypeError: If `x` cannot be cast to the `dtype`.

None
'''