Python PyTorch ScriptModule.buffers用法及代碼示例

本文簡要介紹python語言中 torch.jit.ScriptModule.buffers 的用法。

用法:

buffers(recurse=True)

參數：

recurse(bool) -如果為真，則產生此模塊和所有子模塊的緩衝區。否則，隻產生作為該模塊直接成員的緩衝區。

生成(Yield)：

torch.Tensor - 模塊緩衝區

返回模塊緩衝區的迭代器。

例子：

>>> for buf in model.buffers():
>>>     print(type(buf), buf.size())
<class 'torch.Tensor'> (20L,)
<class 'torch.Tensor'> (20L, 1L, 5L, 5L)