Python PyTorch pop用法及代碼示例

本文簡要介紹python語言中 torch.distributed.pipeline.sync.skip.skippable.pop 的用法。

用法:

class torch.distributed.pipeline.sync.skip.skippable.pop(name)

參數：

name(str) -跳躍張量的名稱

返回：

先前由另一個層以相同名稱隱藏的跳過張量

彈出跳過張量的命令。

def forward(self, input):
    skip = yield pop('name')
    return f(input) + skip