search

f-dangel / backpack

BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
https://backpack.pt/
MIT License
549 stars 55 forks source link

[REF] Make output shape of `unfold_by_conv_transpose` consistent with convolution case #289

Closed f-dangel closed 1 year ago

f-dangel commented 1 year ago

Changes the output shape from [N, C, K * X] to [N, C * K, X], like PyTorch's convention of torch.nn.Unfold.