HasWeight

torch.nn.modules.HasWeight

Attributes

Source
Module.scala
Graph
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

Members list

Value members

Abstract methods

def weight: Tensor[ParamType]

Attributes

Source
Module.scala