Lines Matching defs:Array
29 Array = Any variable
34 def relu(x: Array) -> Array:
45 def softplus(x: Array) -> Array:
55 def soft_sign(x: Array) -> Array:
65 def sigmoid(x: Array) -> Array:
75 def silu(x: Array) -> Array:
87 def log_sigmoid(x: Array) -> Array:
97 def elu(x: Array, alpha: Array = 1.0) -> Array:
111 def leaky_relu(x: Array, negative_slope: Array = 1e-2) -> Array:
126 def hard_tanh(x: Array) -> Array:
140 def celu(x: Array, alpha: Array = 1.0) -> Array:
156 def selu(x: Array) -> Array:
178 def gelu(x: Array, approximate: bool = True) -> Array:
206 def glu(x: Array, axis: int = -1) -> Array:
218 def log_softmax(x: Array, axis: Optional[Union[int, Tuple[int, ...]]] = -1) -> Array:
235 def softmax(x: Array, axis: Optional[Union[int, Tuple[int, ...]]] = -1) -> Array:
252 def normalize(x: Array,
254 mean: Optional[Array] = None,
255 variance: Optional[Array] = None,
256 epsilon: Array = 1e-5) -> Array:
268 def one_hot(x: Array, num_classes: int, *, dtype: Any = jnp.float64) -> Array:
300 def relu6(x: Array) -> Array:
310 def hard_sigmoid(x: Array) -> Array:
320 def hard_silu(x: Array) -> Array: