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