TFSamepaddingLayer

class TFSamepaddingLayer(ksize, stride)[source]

To align with tensorflow same padding.

Putting this before any conv layer that needs padding. Here, we assume kernel has same height and width for simplicity.

Methods

forward

Logic for using layers defined in init.

Attributes

Parameters

  • ksize (int) –

  • stride (int) –

forward(x)[source]

Logic for using layers defined in init.

Parameters

x (torch.Tensor) –