This crate isn't used by any other public crates. Go back to tch-tensor-like.