eazygrad.relu

eazygrad.relu(input: _Tensor) → _Tensor

Apply the rectified linear unit activation elementwise.

Parameters:

input (_Tensor) – Input tensor.

Returns:

_Tensor – Tensor with max(input, 0) applied elementwise.

See also

torch.nn.functional.relu