docs
0.0-2dfa388-SNAPSHOT
API
Generated with
docs/torch/torch.nn/torch.nn.modules

torch.nn.modules

package torch.nn.modules

Members list

Packages

package torch.nn.modules.activation
package torch.nn.modules.batchnorm
package torch.nn.modules.container
package torch.nn.modules.conv
package torch.nn.modules.flatten
package torch.nn.modules.linear
package torch.nn.modules.normalization
package torch.nn.modules.pooling
package torch.nn.modules.regularization
package torch.nn.modules.sparse

Type members

Classlikes

trait HasParams[ParamType <: FloatNN | ComplexNN] extends Module

Attributes

Source
Module.scala
Supertypes
class Module
class Object
trait Matchable
class Any
Known subtypes
class BatchNorm1d[ParamType]
class BatchNorm2d[ParamType]
class Conv2d[ParamType]
class Linear[ParamType]
class Dropout[ParamType]
class Embedding[ParamType]
Show all
trait HasWeight[ParamType <: FloatNN | ComplexNN]

Attributes

Source
Module.scala
Supertypes
class Object
trait Matchable
class Any
Known subtypes
class BatchNorm1d[ParamType]
class BatchNorm2d[ParamType]
class Linear[ParamType]
class GroupNorm[ParamType]
class LayerNorm[ParamType]
class Embedding[ParamType]
Show all
abstract class Module

Attributes

Source
Module.scala
Supertypes
class Object
trait Matchable
class Any
Known subtypes
class ResNet[D]
class ModuleList[D]
class Sequential[D]
trait HasParams[ParamType]
class BatchNorm1d[ParamType]
class BatchNorm2d[ParamType]
class Conv2d[ParamType]
class Linear[ParamType]
class Dropout[ParamType]
class Embedding[ParamType]
trait TensorModule[D]
class BasicBlock[D]
class Bottleneck[D]
class LogSoftmax[D]
class ReLU[D]
class Softmax[D]
class Tanh[D]
class Flatten[D]
class Identity[D]
class GroupNorm[ParamType]
class LayerNorm[ParamType]
class MaxPool2d[D]
trait TensorModuleBase[D, D2]
Show all
trait TensorModule[D <: DType] extends Module, Tensor[D] => Tensor[D]

Transforms a single tensor into another one of the same type.

Transforms a single tensor into another one of the same type.

Attributes

Source
Module.scala
Supertypes
trait Tensor[D] => Tensor[D]
class Module
class Object
trait Matchable
class Any
Known subtypes
class BasicBlock[D]
class Bottleneck[D]
class LogSoftmax[D]
class ReLU[D]
class Softmax[D]
class Tanh[D]
class BatchNorm1d[ParamType]
class BatchNorm2d[ParamType]
class ModuleList[D]
class Sequential[D]
class Conv2d[ParamType]
class Flatten[D]
class Identity[D]
class Linear[ParamType]
class GroupNorm[ParamType]
class LayerNorm[ParamType]
class MaxPool2d[D]
class Dropout[ParamType]
Show all
trait TensorModuleBase[D <: DType, D2 <: DType] extends Module, Tensor[D] => Tensor[D2]

Attributes

Source
Module.scala
Supertypes
trait Tensor[D] => Tensor[D2]
class Module
class Object
trait Matchable
class Any
Known subtypes
class Embedding[ParamType]
Generated with