Skip to content
Snippets Groups Projects
Commit 09a6e997 authored by Maja Jabłońska's avatar Maja Jabłońska
Browse files

Add Activation function

parent d1ee7772
1 merge request!46Merge COMBO 3.0 into master
import torch
import torch.nn as nn
from overrides import overrides
class Activation(nn.Module):
def __cal__(self, tensor: torch.Tensor) -> torch.Tensor:
def forward(self, x: torch.Tensor) -> torch.Tensor:
raise NotImplementedError
class LinearActivation(Activation):
@overrides
def forward(self, x: torch.Tensor) -> torch.Tensor:
return x
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment