Skip to main content

Pytorch Extension Module.

Project description

PYTHON version PyPI version Downloads

(WIP) torchex library

torchex library provides advanced Neural Network Layers. You can easily use them like using original pytorch.


$ pip install torchex


  • Pytorch >= 1.0


How to use

Lazy Style Model Definition

import torch
import torchex.nn as exnn

net = exnn.Linear(10)
# You don't need to give the size of input for this module.
# This network is equivalent to `nn.Linear(100, 10)`.

x = troch.randn(10, 100)

y = net(x)

torchex.nn list

  • torchex.nn.Pass
  • torchex.nn.Flatten
  • torchex.nn.Linear
    • Lazy style
  • torchex.nn.Conv1d
    • Lazy style
  • torchex.nn.Conv2d
    • Lazy style
  • torchex.nn.Conv3d
    • Lazy style
  • torchex.nn.Conv2dLocal
  • torchex.nn.GlobalAvgPool1d
  • torchex.nn.GlobalAvgPool2d
  • torchex.nn.GlobalMaxPool1d
  • torchex.nn.GlobalMaxPool2d
  • torchex.nn.MaxAvgPool2d
  • torch.nn.Crop2d
  • torch.nn.Crop3d
  • torch.nn.MLPConv2d
  • torch.nn.UpsampleConvLayer
  • torch.nn.CordConv2d
  • torch.nn.DFT1d
  • torch.nn.DFT2d
  • torch.nn.PeriodicPad2d
  • torch.nn.PeriodicPad3d
  • torch.nn.Highway
  • torch.nn.Inception
  • torch.nn.InceptionBN
  • torch.nn.IndRNNCell
  • torch.nn.IndRNNTanhCell
  • torch.nn.IndRNNReLuCell
  • torch.nn.IndRNN
  • torch.nn.GraphLinear
  • torch.nn.GraphConv
  • torch.nn.SparseMM
  • torch.nn.GraphBatchNrom


for visualization

  • torchex.attribute.IntegratedGradients

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for torchex, version 0.0.16
Filename, size File type Python version Upload date Hashes
Filename, size torchex-0.0.16-py2.py3-none-any.whl (46.3 kB) File type Wheel Python version py2.py3 Upload date Hashes View
Filename, size torchex-0.0.16.tar.gz (29.8 kB) File type Source Python version None Upload date Hashes View

Supported by

Pingdom Pingdom Monitoring Google Google Object Storage and Download Analytics Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page