class TorchVision::Transforms::Compose

Public Class Methods

new(transforms) click to toggle source
# File lib/torchvision/transforms/compose.rb, line 4
def initialize(transforms)
  @transforms = transforms
end

Public Instance Methods

forward(img) click to toggle source
# File lib/torchvision/transforms/compose.rb, line 8
def forward(img)
  @transforms.each do |t|
    img = t.call(img)
  end
  img
end