Hi!
I have huge 2D tensor A and I want to convolve each row by same filter, for this purpose I use torch.nn.functional.conv2D, but there is a problem, sometimes I A is such that I can’t place 2 As in my GPUs memory, so I am interested is there any way that I can make in-place convolution to use less memory?