Update local_response_norm.py
Browse files- local_response_norm.py +4 -0
local_response_norm.py
CHANGED
|
@@ -1,3 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
class LocalResponseNorm(nn.Module):
|
| 2 |
@nn.compact
|
| 3 |
def __call__(
|
|
|
|
| 1 |
+
from flax import linen as nn
|
| 2 |
+
import jax
|
| 3 |
+
import jax.numpy as jnp
|
| 4 |
+
|
| 5 |
class LocalResponseNorm(nn.Module):
|
| 6 |
@nn.compact
|
| 7 |
def __call__(
|