load_torch

deepinv.utils.load_torch(fname, device=None, **kwargs)

Load torch tensor from file.

Parameters:

• fname (str, pathlib.Path) – file to load.

• device (torch.device, str) – device to load onto.

Returns:

torch.Tensor containing loaded torch tensor.

Return type:

Tensor