Skip to content

Add support for slicing 2D tensors (i.e. no H,W dims)

Sam Eadie requested to merge sameadie/sadl:slice2d into master

Currently, only slicing of 4D tensors (e.g. from convolutions) is supported. This MR adds support to slice 2D tensors along their non-batch dimension.

Merge request reports