MemSaveReLU

class memsave_torch.nn.MemSaveReLU

MemSaveReLU.

Inits a MemSaveReLU layer with the given params.

forward(x)

Forward pass.

Parameters:

x – Input to the network

Returns:

Output

Return type:

torch.Tensor

classmethod from_nn_ReLU(relu: ReLU)

Converts a nn.ReLU layer to MemSaveReLU.

Parameters:

relu – The nn.ReLU layer

Returns:

The MemSaveReLU object

Return type:

obj

Hint

The usage is the same as torch.nn.ReLU

For usage examples, please refer to the linked torch documentation