45e1418d553f26de3bf0d58c62bb954d17268c7e,torchsample/functions/affine.py,,F_bilinear_interp2d,#,40

Before Change

    y0 = y.floor()
    y1 = y0 + 1

    stride = torch.LongTensor(input.stride()[1:])
    x0_ix = x0.mul(stride[0]).long()
    x1_ix = x1.mul(stride[0]).long()
    y0_ix = y0.mul(stride[1]).long()

After Change

    y0_ix = y0.mul(stride[2]).long()
    y1_ix = y1.mul(stride[2]).long()

    input_flat = input.view(input.size(0),-1).contiguous()

    vals_00 = input_flat.gather(1, x0_ix.add(y0_ix).detach())
    vals_10 = input_flat.gather(1, x1_ix.add(y0_ix).detach())
    vals_01 = input_flat.gather(1, x0_ix.add(y1_ix).detach())

Instances