Skip to main content
Log in

Mojo function

pack_filter_shape

pack_filter_shape(filter: NDBuffer[type, rank, origin, shape, strides, alignment=alignment, address_space=address_space, exclusive=exclusive], num_groups: Int) -> Index[(rank + 1)]

Compute the output shape of transposed convolution filter packing.

Args:

  • filter (NDBuffer[type, rank, origin, shape, strides, alignment=alignment, address_space=address_space, exclusive=exclusive]): The filter to be packed.
  • num_groups (Int): The number of groups in the convolution.

Returns:

The output shape.