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