Class: TorchVision::Transforms::ToTensor

Inherits:

Torch::NN::Module

Object
Torch::NN::Module
TorchVision::Transforms::ToTensor

show all

Defined in:: lib/torchvision/transforms/to_tensor.rb

Instance Method Summary collapse

#forward(pic) ⇒ Object

Instance Method Details

#forward(pic) ⇒ `Object`



4
5
6

# File 'lib/torchvision/transforms/to_tensor.rb', line 4

def forward(pic)
  F.to_tensor(pic)
end