geoffsmith

import onnx
import torch
from caffe2.python.onnx import backend
from torch import nn
from torch.autograd import Variable
from torch.nn import PixelShuffle

class MyPixelShuffle(nn.Module):
    """
    Alternative version of pixel shuffle that was copied from the C++ version