ok, works for me. Could you tell me please, does I do it right way? I mean about multiplication weights like on my paint:
       output = torch.matmul(x, self.weights_ih) + self.bias_ih

where:
        self.weights_ih = torch.rand((input_features, input_features), device=device)    
        self.bias_ih = torch.rand((input_features), device=device)