class TorchVision::Transforms::Normalize

Public Class Methods

new(mean, std, inplace: false) click to toggle source
# File lib/torchvision/transforms/normalize.rb, line 4
def initialize(mean, std, inplace: false)
  @mean = mean
  @std = std
  @inplace = inplace
end

Public Instance Methods

forward(tensor) click to toggle source
# File lib/torchvision/transforms/normalize.rb, line 10
def forward(tensor)
  F.normalize(tensor, @mean, @std, inplace: @inplace)
end