Python/OpenCL
< Python
Jump to navigation
Jump to search
numpy
Data Types
import numpy as np
# Default int type is int64
a = np.array([1, 2, 3])
print(type(a[0]).__name__)
# Default float type is float64
b = np.array([1.0, 2.0, 3.0])
print(type(b[0]).__name__)
# Force int32
c = np.array([1, 2, 3]).astype(np.int32)
print(type(c[0]).__name__)
# Force float32
d = np.array([1.0, 2.0, 3.0]).astype(np.float32)
print(type(d[0]).__name__)