Function tch::vision::dataset::augmentation

pub fn augmentation(t: &Tensor, flip: bool, crop: i64, cutout: i64) -> Tensor