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