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