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

torch.nn

package torch.nn

These are the basic building blocks for graphs.

Attributes

Members list

Packages

package torch.nn.functional

Attributes

package torch.nn.loss
package torch.nn.modules

Type members

Classlikes

object init

No gradients will be recorded for these operations.

No gradients will be recorded for these operations.

Attributes

See also

init.h https://pytorch.org/cppdocs/api/file_torch_csrc_api_include_torch_nn_init.h.html#file-torch-csrc-api-include-torch-nn-init-h

Source
init.scala
Supertypes
class Object
trait Matchable
class Any
Self type
init.type
object utils

Attributes

Source
utils.scala
Supertypes
class Object
trait Matchable
class Any
Self type
utils.type

Exports

Defined exports

final type AdaptiveAvgPool2d = [D <: BFloat16 | Float32 | Float64] =>> AdaptiveAvgPool2d[D]
Exported from pooling

Attributes

Source
package.scala
final type BatchNorm1d = [ParamType <: FloatNN | ComplexNN] =>> BatchNorm1d[ParamType]
Exported from batchnorm

Attributes

Source
package.scala
final type BatchNorm2d = [ParamType <: FloatNN | ComplexNN] =>> BatchNorm2d[ParamType]
Exported from batchnorm

Attributes

Source
package.scala
final val Conv2d: Conv2d.type
Exported from conv
final type Conv2d = [ParamType <: FloatNN | ComplexNN] =>> Conv2d[ParamType]
Exported from conv

Attributes

Source
package.scala
final type CrossEntropyLoss = CrossEntropyLoss
Exported from loss

Attributes

Source
package.scala
final type Dropout = [ParamType <: FloatNN | ComplexNN] =>> Dropout[ParamType]
Exported from regularization

Attributes

Source
package.scala
final type Embedding = [ParamType <: FloatNN | ComplexNN] =>> Embedding[ParamType]
Exported from sparse

Attributes

Source
package.scala
final type Flatten = [D <: DType] =>> Flatten[D]
Exported from flatten

Attributes

Source
package.scala
final type GroupNorm = [ParamType <: FloatNN | ComplexNN] =>> GroupNorm[ParamType]
Exported from normalization

Attributes

Source
package.scala
final type Identity = [D <: DType] =>> Identity[D]
Exported from linear

Attributes

Source
package.scala
final type LayerNorm = [ParamType <: FloatNN | ComplexNN] =>> LayerNorm[ParamType]
Exported from normalization

Attributes

Source
package.scala
final type Linear = [ParamType <: FloatNN] =>> Linear[ParamType]
Exported from linear

Attributes

Source
package.scala
final type LogSoftmax = [D <: DType] =>> LogSoftmax[D]
Exported from activation

Attributes

Source
package.scala
final type MaxPool2d = [D <: BFloat16 | Float32 | Float64] =>> MaxPool2d[D]
Exported from pooling

Attributes

Source
package.scala
final type Module = Module
Exported from modules

Attributes

Source
package.scala
final type ModuleList = [D <: DType] =>> ModuleList[D]
Exported from container

Attributes

Source
package.scala
final type ReLU = [D <: DType] =>> ReLU[D]
Exported from activation

Attributes

Source
package.scala
final type Sequential = [D <: DType] =>> Sequential[D]
Exported from container

Attributes

Source
package.scala
final type Softmax = [D <: DType] =>> Softmax[D]
Exported from activation

Attributes

Source
package.scala
final type Tanh = [D <: DType] =>> Tanh[D]
Exported from activation

Attributes

Source
package.scala
Generated with