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