Is_bias ( bool, optional) – if this is a bias parameter. Default: None.ĭtype ( str, optional) – Data type of this parameter. any () create_parameter ( shape,ĭefault_initializer = None ) create_parameter ¶Īttr ( ParamAttr, optional) – Parameter attribute of weight. to_tensor ( value1 ) out1 = linear ( in1 ) # hook change the linear's input to input * 2, so out0 is equal to out1. remove () value1 = value0 * 2 in1 = paddle. to_tensor ( value0 ) out0 = linear ( in0 ) # remove the hook forward_pre_hook_handle. register_forward_pre_hook ( forward_pre_hook ) value0 = np. Linear ( 13, 5 ) # register the hook forward_pre_hook_handle = linear. Import paddle import numpy as np # the forward_pre_hook change the input of the layer: input = input * 2 def forward_pre_hook ( layer, input ): # user can use layer and input for information statistis tasks # change the input input_return = ( input * 2 ) return input_return linear = paddle. Hook ( function) – a function registered as a forward pre-hook Hook(Layer, input) -> None or modified input User can use forward pre-hook to change the input of the Layer or perform information statistics tasks on the Layer. We will wrap the value into a tuple if a single value is returned(unless that value is already a tuple). It should have the following form, input of the hook is input of the Layer, hook can either return a tuple or a single modified value in the hook. The hook will be called before forward function has been computed. any () register_forward_pre_hook ( hook ) register_forward_pre_hook ¶ remove () out1 = linear ( in1 ) # hook change the linear's output to output * 2, so out0 is equal to out1 * 2. to_tensor ( value1 ) out0 = linear ( in1 ) # remove the hook forward_post_hook_handle. register_forward_post_hook ( forward_post_hook ) value1 = np. Linear ( 13, 5 ) # register the hook forward_post_hook_handle = linear. Import paddle import numpy as np # the forward_post_hook change the output of the layer: output = output * 2 def forward_post_hook ( layer, input, output ): # user can use layer, input and output for information statistis tasks # change the output return output * 2 linear = paddle. HookRemoveHelper, a HookRemoveHelper object that can be used to remove the added hook by calling hook_remove_helper.remove(). Hook ( function) – a function registered as a forward post-hook Hook(Layer, input, output) -> None or modified output User can use forward post-hook to change the output of the Layer or perform information statistics tasks on the Layer. It should have the following form, input and output of the hook is input and output of the Layer respectively. The hook will be called after forward function has been computed. full_name ()) # demo_linear_net_0 register_forward_post_hook ( hook ) register_forward_post_hook ¶
_linear ( x ) linear_net = LinearNet () print ( linear_net. Linear ( 1, 1 ) def forward ( self, x ): return self. _init_ ( name_scope = "demo_linear_net" ) self. Layer ): def _init_ ( self ): super ( LinearNet, self ).
0 kommentar(er)
