pyiqa.matlab_utils.math_util ============================ .. py:module:: pyiqa.matlab_utils.math_util .. autoapi-nested-parse:: Mathematical utilities Created by: https://github.com/tomrunia/PyTorchSteerablePyramid/blob/master/steerable/math_utils.py Modified by: Jiadi Mo (https://github.com/JiadiMo) Module Contents --------------- .. py:function:: abs(x) .. py:function:: roll_n(X, axis, n) .. py:function:: batch_fftshift2d(x) Args: x: An complex tensor. Shape :math:`(N, C, H, W)`. Pytroch version >= 1.8.0 .. py:function:: batch_ifftshift2d(x) :param x: An input tensor. Shape :math:`(N, C, H, W, 2)`. :returns: An complex tensor. Shape :math:`(N, C, H, W)`. .. py:function:: prepare_grid(m, n) .. py:function:: rcosFn(width, position) .. py:function:: pointOp(im, Y, X) .. py:function:: getlist(coeff)