A set of Pytorch compatible modules for use with 4D tensors

4D Layers and Functions Compatible with PyTorch
· Report Bug · Request Feature

Python package for 4d layers to be used on tensors of shape (batch_size, channel, k, d, h, w). I am new to packaging and distributing software in this manner--especially ML packages--so please go easy on me.

Using this package should be fairly simple!

   $ pip install torch4d

Once you've installed the package and once in Python, make sure to import:

   from torch4d.torch4d import MaxPool4d, DropBlock4d, SpatialDropout4d

Alternatively, you can use the functions defined in torch4d.functional by importing

   from torch4d.functional import max_pool4d, drop_block4d, spatial_dropout4d

See the open issues for a list of proposed features (and known issues).

Distributed under the GNU General Public License. See LICENSE for more information.

firasbdarwish - [email protected], dhiyaalajorf - [email protected]

Project Link: https://github.com/FirasBDarwish/PyTorch4D