nonlinearity Namespace Reference

Module caffe2.python.helpers.nonlinearity. More...

Functions
def	PRelu (model, blob_in, blob_out, num_channels=1, slope_init=None, kwargs)
def	Relu (model, blob_in, blob_out, use_cudnn=False, order="NCHW", kwargs)

Detailed Description

Module caffe2.python.helpers.nonlinearity.

Function Documentation

PRelu()

def nonlinearity.PRelu	(		model,
			blob_in,
			blob_out,
			num_channels = 1,
			slope_init = None,
			kwargs
	)

PRelu

Definition at line 12 of file nonlinearity.py.

Relu()

def nonlinearity.Relu	(		model,
			blob_in,
			blob_out,
			use_cudnn = False,
			order = "NCHW",
			kwargs
	)

Relu.

Definition at line 32 of file nonlinearity.py.