[ Web Proxy ]
URL:
Viewing: https://diffsharp.github.io/reference/diffsharp-model.html [Back]  [Original]

DiffSharp.Model

DiffSharp.Model Namespace

Contains types and functionality related to describing models.

Type/Module Description

BatchNorm1d

github.png [github.png] github-hover.png [github-hover.png]

Applies Batch Normalization over a 2D or 3D input (a mini-batch of 1D inputs with optional additional channel dimension)

BatchNorm2d

github.png [github.png] github-hover.png [github-hover.png]

Applies Batch Normalization over a 4D input (a mini-batch of 2D inputs with optional additional channel dimension)

BatchNorm3d

github.png [github.png] github-hover.png [github-hover.png]

Applies Batch Normalization over a 5D input (a mini-batch of 3D inputs with optional additional channel dimension)

Conv1d

github.png [github.png] github-hover.png [github-hover.png]

A model that applies a 1D convolution over an input signal composed of several input planes

Conv2d

github.png [github.png] github-hover.png [github-hover.png]

A model that applies a 2D convolution over an input signal composed of several input planes

Conv3d

github.png [github.png] github-hover.png [github-hover.png]

A model that applies a 3D convolution over an input signal composed of several input planes

ConvTranspose1d

github.png [github.png] github-hover.png [github-hover.png]

A model that applies a 1D transposed convolution operator over an input image composed of several input planes.

ConvTranspose2d

github.png [github.png] github-hover.png [github-hover.png]

A model that applies a 2D transposed convolution operator over an input image composed of several input planes.

ConvTranspose3d

github.png [github.png] github-hover.png [github-hover.png]

A model that applies a 3D transposed convolution operator over an input image composed of several input planes.

Dropout

github.png [github.png] github-hover.png [github-hover.png]

A model which during training, randomly zeroes some of the elements of the input tensor with probability p using samples from a Bernoulli distribution.

Dropout2d

github.png [github.png] github-hover.png [github-hover.png]

A model which during training, randomly zero out entire channels. Each channel will be zeroed out independently on every forward call with probability p using samples from a Bernoulli distribution.

Dropout3d

github.png [github.png] github-hover.png [github-hover.png]

A model which during training, randomly zero out entire channels. Each channel will be zeroed out independently on every forward call with probability p using samples from a Bernoulli distribution.

Linear

github.png [github.png] github-hover.png [github-hover.png]

A model that applies a linear transformation to the incoming data: \(y = xA^T + b\)

LSTM

LSTMCell

github.png [github.png] github-hover.png [github-hover.png]

Unit cell of a long short-term memory (LSTM) recurrent neural network. Prefer using the RNN class instead, which can combine RNNCells in multiple layers.

Mode

Model

"> Model

github.png [github.png] github-hover.png [github-hover.png]

Represents a model, primarily a collection of named parameters and sub-models and a function governed by them.

ModelBase

Parameter

ParameterDict

RecurrentShape

RNN

RNNCell

github.png [github.png] github-hover.png [github-hover.png]

Unit cell of a recurrent neural network. Prefer using the RNN class instead, which can combine RNNCells in multiple layers.

Sequential

VAE

VAEBase

VAEMLP

github.png [github.png] github-hover.png [github-hover.png]

Variational auto-encoder with multilayer perceptron (MLP) encoder and decoder.

Weight

github.png [github.png] github-hover.png [github-hover.png]

Contains functionality related to generating initial parameter weights for models.


Copyright 2021, DiffSharp Contributors.



Web Proxy Viewer  |  New URL  |  Original Page