tch::vision::dataset
pub fn random_flip(t: &Tensor) -> Tensor
Randomly applies horizontal flips This expects a 4 dimension NCHW tensor and returns a tensor with an identical shape.