pytorch中的嵌套函数

lcqin111

已于 2023-05-31 20:37:38 修改

阅读量1.4k

点赞数 2

分类专栏： python Pytorch 文章标签： pytorch python 人工智能

于 2021-06-04 15:52:50 首次发布

本文链接：https://blog.csdn.net/lcqin111/article/details/117563755

版权

python 同时被 2 个专栏收录

78 篇文章

订阅专栏

Pytorch

9 篇文章

订阅专栏

这段PyTorch源码定义了一个_ntuple函数，用于根据参数n生成一个将输入重复n次的元组。_single、_pair、_triple和_quadruple是其特例，分别对应生成1、2、3和4次重复的元组。这些辅助函数在处理迭代器时非常有用，特别是在构建张量形状或处理尺寸时。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

from torch._six import container_abcs

from itertools import repeat
from typing import List


def _ntuple(n):
    def parse(x):
        if isinstance(x, container_abcs.Iterable):
            return x
        return tuple(repeat(x, n))
    return parse

_single = _ntuple(1)
_pair = _ntuple(2)
_triple = _ntuple(3)
_quadruple = _ntuple(4)

这是pytorch源码，用于生成迭代器。

_single本身是一个函数，在调用的时候会将输入的int或者迭代器重复1次。