timothygebhard / pytorch-conv4d

Rudimentary Conv4D Layer Implementation for PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

pytorch-conv4d

This repository contains a simple PyTorch port of the conv4d for TensorFlow repository by Jan Funke. It consists essentially of a single class, Conv4d, which provides a (still rather rudimentary) PyTorch layer for 4-dimensional convolutions. Like the original, it works by performing and stacking several 3D convolutions (see the original repository for a more detailed explanations).

This implementation is still work in progress (hence it comes with no warranties whatsoever), and pull requests or advice for improvements are very much welcome! :)

About

Rudimentary Conv4D Layer Implementation for PyTorch

License:MIT License


Languages

Language:Python 100.0%