fynance.models.neural_network.BaseNeuralNet.named_buffers¶
-
BaseNeuralNet.
named_buffers
(prefix='', recurse=True)¶ Returns an iterator over module buffers, yielding both the name of the buffer as well as the buffer itself.
- Args:
prefix (str): prefix to prepend to all buffer names. recurse (bool): if True, then yields buffers of this module
and all submodules. Otherwise, yields only buffers that are direct members of this module.- Yields:
- (string, torch.Tensor): Tuple containing the name and buffer
Example:
>>> for name, buf in self.named_buffers(): >>> if name in ['running_var']: >>> print(buf.size())