cast¶
dragon.
cast
(
inputs,
dtype,
copy=True,
**kwargs
)[source]¶Convert the data type of input.
Examples:
x = dragon.constant([1, 2, 3], dtype='int64') print(dragon.cast(x, dtype='float32'))
- Parameters:
- inputs (dragon.Tensor) – The input tensor.
- dtype (str) – The data type to convert to.
- copy (bool, optional, default=True) – Return a new tensor or call in-place.
- Returns:
dragon.Tensor – The output tensor.