Tensor.cast()

Casts a tensor to a new type.

Usage

from tinygrad.tensor import Tensor
from tinygrad.helpers import dtypes
 
a1 = Tensor([[1, 2], [3, 4]])
a1 = a1.cast(dtypes.float32)
 
print(a1.numpy())

Return value

[[1. 2.]
 [3. 4.]]

bitcast cat

;