Pytorch cuda tensor in module

import torch
import torch.nn as nn

class Model(nn.Module):
    def __init__(self):
        super(Model, self).__init__()
        self.weight = torch.nn.Parameter(torch.zeros(2, 1))
        self.bias = torch.nn.Parameter(torch.zeros(1))
        self.register_buffer('a_constant_tensor', torch.tensor([0.5]))

    def forward(self, x):
        # linear regression completely from scratch,
        # using parameters created in __init__
        x = torch.mm(x, self.weight) + self.bias + self.a_constant_tensor
        return x


model = Model().cuda()

Source: discuss.pytorch.org

Similar codes

19 votes175 views3 answers
Debugger online for python
Python
0 vote165 views1 answer
Scroll horizontal excel
Python
36 votes168 views6 answers
Remove part of string python
Python
7 votes208 views2 answers
Your models have changes that are not yet reflected in a migration, and so won't be applied. Run 'manage.py makemigrations' to make new migrations, and then re-run 'manage.py migrate' to apply them.
Python
0 vote183 views1 answer
Como eliminar palabras repetidos de una lista python
Python
1 vote174 views1 answer
Biggest of 3 numbers in python
Python
7 votes210 views2 answers
Ckeditor django
Python
0 vote177 views1 answer
Plt clor image histogram
Python
39 votes179 views6 answers
Pyplot python
Python
0 vote227 views1 answer
Indice d'un element dans une liste python
Python