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

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

用法:

parameters(recurse=True)

參數：

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

生成(Yield)：

參數- 模塊參數

返回模塊參數的迭代器。

這通常被傳遞給優化器。

例子：

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