Highway network implemented in PyTorch.
highway = Highway(input_size, num_layers, f=torch.nn.functional.relu)
# input is [batch_size, input_size] shaped tensor
out = highway(input)kefirski/pytorch_Highway
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|