[ 组合替代实现 ]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")