torch.nn.modules.container

Members list

Type members

Classlikes

final class ModuleList[D <: DType](val modules: TensorModule[D]*) extends Module, Seq[TensorModule[D]]

Attributes

Source
ModuleList.scala
Supertypes
trait Seq[TensorModule[D]]
trait Seq[TensorModule[D]]
trait Equals
trait Int => TensorModule[D]
class Module
class Object
trait Matchable
class Any
Show all
final class Sequential[D <: DType](val modules: TensorModule[D]*) extends Module, TensorModule[D]

Attributes

Source
Sequential.scala
Supertypes
trait TensorModule[D]
trait Tensor[D] => Tensor[D]
class Module
class Object
trait Matchable
class Any
Show all