( dataset: Dict[str, torch.Tensor], device: str, bs: int, loss_fn: Callable[[t

(model: nn.Module)

(ctx, grad_o)

(self, layers: Tuple[int, int, int], device: str)

( dataset: Dict[str, torch.Tensor], loss_fn: Callable[[torch.Tensor, torch.Tensor], torch.Ten

(self, input)

(self, layers: Tuple[int, int, int], gridsize: int, device: str)

()

()

()

()

(t: Dict[str, Dict[str, float]], out_path: str)

(self, layers: Tuple[int, int, int], gridsize: int, device: str)

(self, layers: Tuple[int, int, int], device: str)

(self, layers: Tuple[int, int, int], gridsize: int, device: str)

(self, layers: Tuple[int, int, int], gridsize: int, device: str)

(self, layers: Tuple[int, int, int], gridsize: int, device: str)

( self, inputdim, outdim, gridsize,addbias=True)

(self, inputdim, outdim, gridsize)

(ctx, grad_o)

(self, x: torch.Tensor)

(self, x: torch.Tensor)

(self, x: torch.Tensor)

(self, x: torch.Tensor)

(self, x: torch.Tensor)

(self, x: torch.Tensor)

(self, x: torch.Tensor)

(self,x)

(ctx, input, fouriercoeffs, bias)

(ctx, input, fouriercoeffs, bias)