Fork me on GitHub

src/arraymancer/nn_primitives/private/p_nnp_types

Search:
Group by:
  Source Edit

Types

Size2D = tuple[height: int, width: int]
Tuple of height and width This is notably used to specify padding and stride parameters for Convolution2D.   Source Edit

Procs

proc nchw_channels[T](input: Tensor[T]): int {.inline.}
Return number of channels of a Tensor in NCWH layout   Source Edit
proc nchw_height[T](input: Tensor[T]): int {.inline.}
Return height of a Tensor in NCWH layout   Source Edit
proc nchw_width[T](input: Tensor[T]): int {.inline.}
Return width of a Tensor in NCWH layout   Source Edit
Arraymancer Technical reference Tutorial Spellbook (How-To's) Under the hood