Returns the number of trainable parameters in a module.

Not equal to

* \Brief Transposes the scalar type. * The default implementation does nothing. */

Indirection

Postfix increment

Equal to

Draws samples from an interpolated cosine timestep distribution (from simple diffusion).

Draws samples from an optionally truncated log-logistic distribution.

Draws samples from an lognormal distribution.

Draws samples from an log-uniform distribution.

Draws samples from a split lognormal distribution.

Draws samples from a truncated v-diffusion training timestep distribution.

Generate random rotations as 3x3 rotation matrices. Args: n: Number of rotation matrices in a batch to return. dtype: Type t

* \brief Extracts the real part of the complex matrix. * This method is only available for complex matrices. * This is the non-const lvalue ve

Converts 6D rotation representation by Zhou et al. [1] to rotation matrix using Gram--Schmidt orthogonalization per Section B of [1]. Arg