Brain cell membrane equivalent neural network training code

import torch
class Cell(torch.nn.Module):
    def __init__(self, in_size=128, out_size=128):
        super(Cell, self