"tests/vscode:/vscode.git/clone" did not exist on "086c7f9ea8f0fde5c62e52289604ec5b178da207"
utils.py 357 Bytes
Newer Older
rusty1s's avatar
to csr  
rusty1s committed
1
2
3
4
5
6
7
8
9
import torch

dtypes = [torch.float, torch.double]

devices = [torch.device('cpu')]
if torch.cuda.is_available():  # pragma: no cover
    devices += [torch.device('cuda:{}'.format(torch.cuda.current_device()))]


rusty1s's avatar
rusty1s committed
10
def tensor(x, dtype, device, requires_grad=False):
rusty1s's avatar
rusty1s committed
11
    return torch.tensor(
rusty1s's avatar
rusty1s committed
12
        x, dtype=dtype, device=device, requires_grad=requires_grad)