megenginelite.utils¶

class TensorBatchCollector(shape, dtype=LiteDataType.LITE_INT8, device_type=LiteDeviceType.LITE_CUDA, device_id=0, is_pinned_host=False, tensor=None)[源代码]¶

this is a tensor utils to collect subtensor in batch continuous

collect(array)[源代码]¶

collect_by_ctypes(data, length)[源代码]¶: collect with ctypes data input

collect_id(array, batch_id)[源代码]¶

free(indexes)[源代码]¶

get()[源代码]¶

to_numpy()[源代码]¶

上一页

megenginelite.struct

下一页

与其它框架 API 进行对比