pylibraft.common.outputs

2 min read time

Applies to Linux

Functions

auto_convert_output(f)	Decorator to automatically convert an output device_ndarray
conv(ret)
convert_to_cai_type(device_ndarray)
convert_to_cupy(device_ndarray)
convert_to_torch(device_ndarray)
import_warn_(lib)
no_conversion(device_ndarray)

Module Contents

pylibraft.common.outputs.auto_convert_output(f)

Decorator to automatically convert an output device_ndarray

(or list or tuple of device_ndarray) into the configured __cuda_array_interface__ compliant type.

pylibraft.common.outputs.conv(ret)

pylibraft.common.outputs.convert_to_cai_type(device_ndarray)

pylibraft.common.outputs.convert_to_cupy(device_ndarray)

pylibraft.common.outputs.convert_to_torch(device_ndarray)

pylibraft.common.outputs.import_warn_(lib)

pylibraft.common.outputs.no_conversion(device_ndarray)