[ 组合替代实现 ]torch.nn.modules.utils._ntuple

torch.nn.modules.utils._ntuple

torch.nn.modules.utils._ntuple(n, name="parse")

Paddle 无此 API,需要组合实现。

转写示例

# PyTorch 写法
torch.nn.modules.utils._ntuple(2, name="parse")

# Paddle 写法
def _ntuple(n, name="parse"):
    def parse(x):
        if isinstance(x, collections.abc.Iterable):
            return tuple(x)
        return tuple(repeat(x, n))

    parse.__name__ = name
    return parse

_ntuple(2, name="parse")