shiftresnet-cifar icon indicating copy to clipboard operation
shiftresnet-cifar copied to clipboard

Published 20 hours ago • verified publisher icon alvinwan

Can the shift be implemented in Pytorch without custom cuda function?

Open yxchng opened this issue 4 years ago • 0 comments

Not sure if I understand correctly, I just need to shift the tensor in 8 different directions (plus not shifting) and that can be done in Pytorch using simple slicing?

Is that what the cuda operation is doing?

yxchng avatar Jan 20 '21 16:01 yxchng