pytorch_pfn_extras.map#

pytorch_pfn_extras.map(func, iterable, out_keys=None, device='cpu')#
Parameters:

  • func (Callable[[Any], Any]) –

  • iterable (Sequence[Any]) –

  • out_keys (Optional[Set[str]]) –

  • device (Any) –

Return type:

Sequence[Any]