ttnn.to_dtype

ttnn.to_dtype() → None

Converts a tensor to the desired dtype

:param * tensor: the ttnn.Tensor :param * dtype: ttnn data type.

Example

>>> tensor = ttnn.from_torch(torch.randn((10, 64, 32), dtype=torch.bfloat16))
>>> tensor = ttnn.to_dtype(tensor, dtype=ttnn.uint16)