我通过以下方式创建了一个pyopencl.Image
对象:
import pyopencl as cl
import numpy as np
ctx = cl.create_some_context()
image = cl.image_from_array(ctx, np.ones((16, 16), dtype=np.uint8))
如何获取图像频道类型信息(此处为cl.channel_type.UNSIGNED_INT8
)?我想实现一个通用下载函数,它返回一个给定Image
的numpy数组。
我尝试了两种方法:
使用Image.element_size
给出了单个元素的字节大小,但是我不能将元素大小映射到数据类型(例如,float32和int32具有相同的元素大小)。
Image.format
返回cdata 'struct _cl_image_format &'
。查询字段image_channel_data_type
和image_channel_order
中的任何一个都会返回0。