def METHOD_NAME(self):
    # inp, weights, biases
    import torch
    return torch.randn(self.bs, self.i,
                       self.k).cuda(), torch.randn(self.k,
                                                   self.j).cuda(), torch.randn(self.j).cuda()