symjax.nn.layers.GRU

class symjax.nn.layers.GRU(sequence, init_h, units, Wh=<function glorot_uniform>, Uh=<function orthogonal>, bh=<function zeros>, Wz=<function glorot_uniform>, Uz=<function orthogonal>, bz=<function zeros>, Wr=<function glorot_uniform>, Ur=<function orthogonal>, br=<function zeros>, trainable_Wh=True, trainable_Uh=True, trainable_bh=True, trainable_Wz=True, trainable_Uz=True, trainable_bz=True, trainable_Wr=True, trainable_Ur=True, trainable_br=True, activation=<function sigmoid>, phi=<function _one_to_one_unop.<locals>.<lambda>>, only_last=False, gate='minimal')[source]
__init__(sequence, init_h, units, Wh=<function glorot_uniform>, Uh=<function orthogonal>, bh=<function zeros>, Wz=<function glorot_uniform>, Uz=<function orthogonal>, bz=<function zeros>, Wr=<function glorot_uniform>, Ur=<function orthogonal>, br=<function zeros>, trainable_Wh=True, trainable_Uh=True, trainable_bh=True, trainable_Wz=True, trainable_Uz=True, trainable_bz=True, trainable_Wr=True, trainable_Ur=True, trainable_br=True, activation=<function sigmoid>, phi=<function _one_to_one_unop.<locals>.<lambda>>, only_last=False, gate='minimal')[source]

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(sequence, init_h, units[, Wh, Uh, …]) Initialize self.
GRU.add_update
GRU.add_variable
argmax([axis, out]) Returns the indices of the maximum values along an axis.
argmin([axis, out]) Returns the indices of the minimum values along an axis.
astype(new_dtype) Elementwise cast.
cast(new_dtype) Elementwise cast.
clone(givens)
conj() Return the complex conjugate, element-wise.
conjugate() Return the complex conjugate, element-wise.
GRU.create_tensor
GRU.create_variable
dot(b, *[, precision]) Dot product of two arrays.
expand_dims(axis, Tuple[int, …]]) Expand the shape of an array.
flatten() reshape the input into a vector
forward()
full_gate(h, x, Wh, Uh, bh, Wz, Uz, bz, Wr, …)
imag() Return the imaginary part of the complex argument.
GRU.init_input
matmul(b, *[, precision]) Matrix product of two arrays.
max([axis, out, keepdims, initial, where]) Return the maximum of an array or maximum along an axis.
mean([axis, dtype, out, keepdims]) Compute the arithmetic mean along the specified axis.
min([axis, out, keepdims, initial, where]) Return the minimum of an array or minimum along an axis.
minimal_gate(h, x, Wh, Uh, bh, Wz, Uz, bz, …)
prod([axis, dtype, out, keepdims, initial, …]) Return the product of array elements over a given axis.
real() Return the real part of the complex argument.
repeat(repeats[, axis, total_repeat_length]) Repeat elements of an array.
reshape(newshape[, order]) Gives a new shape to an array without changing its data.
round([decimals, out]) Round an array to the given number of decimals.
squeeze(axis, Tuple[int, …]] = None) Remove single-dimensional entries from the shape of an array.
std([axis, dtype, out, ddof, keepdims]) Compute the standard deviation along the specified axis.
sum([axis, dtype, out, keepdims, initial, where]) Sum of array elements over a given axis.
transpose([axes]) Reverse or permute the axes of an array; returns the modified array.
var([axis, dtype, out, ddof, keepdims]) Compute the variance along the specified axis.
variables([trainable])

Attributes

dtype
GRU.fn_name
name
ndim
scope
shape
GRU.updates