docs
0.0-2dfa388-SNAPSHOT
API
Generated with
docs/torchvision/torchvision.transforms/functional

functional

torchvision.transforms.functional
object functional

Attributes

Source
functional.scala
Graph
Supertypes
class Object
trait Matchable
class Any
Self type
functional.type

Members list

Value members

Concrete methods

def normalize[D <: FloatNN](tensor: Tensor[D], mean: Seq[Float], std: Seq[Float]): Tensor[D]

Attributes

Source
functional.scala
def toImmutableImage[D <: FloatNN](pic: Tensor[D]): ImmutableImage

Attributes

Source
functional.scala
def toTensor(pic: ImmutableImage): Tensor[Float32]

Convert an ImmutableImage (H x W x C) to a [[Tensor[Float32]] of shape (C x H x W) in the range [0.0, 1.0].

Convert an ImmutableImage (H x W x C) to a [[Tensor[Float32]] of shape (C x H x W) in the range [0.0, 1.0].

Attributes

Source
functional.scala
Generated with