candle
candle copied to clipboard
Use squeeze in tensor.get()
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.