Tensor.shrink()

Returns a tensor with the specified dimensions removed.

Usage

from tinygrad.tensor import Tensor
 
a1 = Tensor([[1, 2, 3], [3, 4, 5], [5, 6, 7]])
a1 = a1.shrink((((0, 2), (1, 2))))
 
print(a1.numpy())

[[2.]
 [4.]]

;