Class DiffWaveResidualBlock<T>

Namespace

AiDotNet.Diffusion.Models

Assembly

AiDotNet.dll

Residual block for DiffWave with dilated convolution.

public class DiffWaveResidualBlock<T>

Type Parameters

T

The numeric type.

Inheritance

object

DiffWaveResidualBlock<T>

Inherited Members

object.Equals(object)

object.Equals(object, object)

object.GetHashCode()

object.GetType()

object.MemberwiseClone()

object.ReferenceEquals(object, object)

object.ToString()

Constructors

DiffWaveResidualBlock(int, int, int, int?)

Initializes a new residual block.

public DiffWaveResidualBlock(int channels, int dilation, int conditionChannels = 0, int? seed = null)

Parameters

channels int

dilation int

conditionChannels int

seed int?

Properties

ParameterCount

Gets the number of parameters.

public int ParameterCount { get; }

Property Value

int

Methods

Forward(Tensor<T>, Tensor<T>, Tensor<T>?)

Forward pass returning output and skip connection.

public (Tensor<T> Output, Tensor<T> Skip) Forward(Tensor<T> x, Tensor<T> diffusionEmbed, Tensor<T>? melCondition)

Parameters

x Tensor<T>

diffusionEmbed Tensor<T>

melCondition Tensor<T>

Returns

(Tensor<T> grad1, Tensor<T> grad2)

GetParameters()

Gets all parameters.

public Vector<T> GetParameters()

Returns

Vector<T>

SetParameters(Vector<T>)

Sets all parameters.

public void SetParameters(Vector<T> parameters)

Parameters

parameters Vector<T>