chainer.functions.cast

chainer.functions.cast(x, typ)[source]

Cast an input variable to a given type.

Parameters
Returns

Variable holding a casted array.

Return type

Variable

Example

>>> x = np.arange(0, 3, dtype=np.float64)
>>> x.dtype
dtype('float64')
>>> y = F.cast(x, np.float32)
>>> y.dtype
dtype('float32')
>>> y = F.cast(x, 'float16')
>>> y.dtype
dtype('float16')