symjax.nn.layers.Conv2D

class symjax.nn.layers.Conv2D(input, n_filters, filter_shape, padding='VALID', strides=1, W=<function glorot_uniform>, b=<built-in function zeros>, trainable_W=True, trainable_b=True, inplace_W=False, inplace_b=False, input_dilations=None, filter_dilations=None, W_preprocessor=None, b_preprocessor=None)[source]

2-D (spatial) convolution

__init__(input, n_filters, filter_shape, padding='VALID', strides=1, W=<function glorot_uniform>, b=<built-in function zeros>, trainable_W=True, trainable_b=True, inplace_W=False, inplace_b=False, input_dilations=None, filter_dilations=None, W_preprocessor=None, b_preprocessor=None)[source]

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

Methods

__init__(input, n_filters, filter_shape[, …]) Initialize self.
Conv2D.add_update
Conv2D.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.
Conv2D.create_tensor
Conv2D.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()
imag() Return the imaginary part of the complex argument.
Conv2D.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.
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
Conv2D.fn_name
name
ndim
scope
shape
Conv2D.updates