imate.AffineMatrixFunction.get_linear_operator

AffineMatrixFunction.get_linear_operator(gpu=False, num_gpu_devices=0)

Sets the linear operator object on CPU or GPU and returns its object.

Parameters:

gpubool, default=False

If True, sets the object representing the matrix data on GPU.

num_gpu_devicesint, default=0

Number of GPU devices. If 0, it uses maximum available GPU devices.

Returns:

operatorobject

Object representing data on CPU or GPU