class TorchVision::Transforms::CenterCrop

Public Class Methods

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

Public Instance Methods

forward(img) click to toggle source
# File lib/torchvision/transforms/center_crop.rb, line 8
def forward(img)
  F.center_crop(img, @size)
end