lenskit.torch

PyTorch utility functions.

Functions

inference_mode(func)	Function decorator that puts PyTorch in inference mode.
safe_tensor(array)	Safely convert an array into a NumPy tensor.
sparse_row(mat, row)	Get a row of a sparse (CSR) tensor.

lenskit.torch.inference_mode(func)

Function decorator that puts PyTorch in inference mode.

lenskit.torch.sparse_row(mat, row)

Get a row of a sparse (CSR) tensor. This is needed because indexing a tensor does not work in inference mode.

Parameters:

• mat (Tensor)

• row (int)

Return type:

Tensor

lenskit.torch.safe_tensor(array)

Safely convert an array into a NumPy tensor. This includes copying it to writable memory if necessary.

Return type:

Tensor