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
'''