class TorchVision::Transforms::Resize
Public Class Methods
new(size)
click to toggle source
# File lib/torchvision/transforms/resize.rb, line 4 def initialize(size) @size = size end
Public Instance Methods
forward(img)
click to toggle source
# File lib/torchvision/transforms/resize.rb, line 8 def forward(img) F.resize(img, @size) end