https://howtostartbloggingformoney.blogspot.com/2023/09/slice-tensor-using-tensor-indices.html