candle icon indicating copy to clipboard operation
candle copied to clipboard

Use squeeze in tensor.get()

Open janimo opened this issue 3 weeks ago • 1 comments

Make get() use squeeze() instead of reshape(), just like get_on_dim() does. It should avoid a copy when the tensor is not contiguous. Also use the same name for the index argument.

janimo avatar Nov 05 '25 13:11 janimo