pytorch_pfn_extras.handler.forward

pytorch_pfn_extras.handler.forward(block)

Returns a CodeBlock that performs the forward pass for the given torch.nn.Module or another CodeBlock.

Parameters:

block (Callable) – torch.nn.Module or CodeBlock to update the parameters.

Return type:

CodeBlock

Returns: A CodeBlock object.