class TorchVision::Transforms::RandomHorizontalFlip
Public Class Methods
new(p: 0.5)
click to toggle source
Calls superclass method
# File lib/torchvision/transforms/random_horizontal_flip.rb, line 4 def initialize(p: 0.5) super() @p = p end
Public Instance Methods
forward(img)
click to toggle source
# File lib/torchvision/transforms/random_horizontal_flip.rb, line 9 def forward(img) if Torch.rand(1).item < @p F.hflip(img) else img end end