blob: 9571f87afefe59cb6a27f2ae9a75e3a6d9b1c0ac (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
### `tf.cast(x, dtype, name=None)` {#cast}
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
# tensor `a` is [1.8, 2.2], dtype=tf.float
tf.cast(a, tf.int32) ==> [1, 2] # dtype=tf.int32
```
##### Args:
* <b>`x`</b>: A `Tensor` or `SparseTensor`.
* <b>`dtype`</b>: The destination type.
* <b>`name`</b>: A name for the operation (optional).
##### Returns:
A `Tensor` or `SparseTensor` with same shape as `x`.
##### Raises:
* <b>`TypeError`</b>: If `x` cannot be cast to the `dtype`.
|