Source code for src.algorithms.deep.models.FCNet

import torch
from torch import nn
from torch.nn import functional as F




class FCNet(nn.Module):
	"""Fully connected three-layer neural network

	:param in_channels: Number of input channels
	:type in_channels: int
	:param out_channels: Number of output channels
	:type out_channels: int
	"""
	def __init__(self, in_channels: int, out_channels: int):
		super(FCNet, self).__init__()
		self.fc1: nn.Linear = nn.Linear(in_channels, out_channels * 4)
		self.fc2: nn.Linear = nn.Linear(out_channels * 4, out_channels * 2)
		self.fc3: nn.Linear = nn.Linear(out_channels * 2, out_channels)



	def forward(self, x: torch.tensor) -> torch.tensor:
		return torch.sigmoid(self.fc3(F.relu(self.fc2(F.relu(self.fc1(x))))))