lbann
lbann copied to clipboard
Add implementation of periodic padding to Python module
Adds a helper function to the Python module that implements PyTorch's circular padding mode for 2D and 3D tensors.