Open shriphani opened 7 years ago
The indexing appears to assume that the input tensor is a 2D image. Would be nice if you could remove this dependency (or at least support 1D, 2D, 3D).
The indexing appears to assume that the input tensor is a 2D image. Would be nice if you could remove this dependency (or at least support 1D, 2D, 3D).