function is_torch_kernel

deeptime.kernels.is_torch_kernel(kernel: Kernel) → bool

Tests whether given kernel supports PyTorch. Note that even if the kernel does not support PyTorch it (depending on the case) might still be usable in conjunction with PyTorch code at a performance penalty, e.g., when no gradients are required for the kernel evaluation.

Parameters:

kernel (Kernel) – A kernel instance.

Returns:

is_torch_kernel – True if it supports PyTorch, False otherwise.

Return type:

bool