fold_shape

fold_shape[dtype: DType, input_dim: DimList](input: NDBuffer[dtype, 3, MutableAnyOrigin, input_dim], output_size: IndexList[2], kernel_size: IndexList[2], stride: IndexList[2], dilation: IndexList[2], padding: IndexList[2]) -> IndexList[4]

Returns the shape of the output tensor of the fold operation.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!