torch.nn.utils.WeightNormOptions

export interface WeightNormOptions {
  /** Name of the weight parameter (default: 'weight') */
  name?: string;
  /** Dimension over which to compute the norm (default: 0) */
  dim?: number;
}

name(string)optional: – Name of the weight parameter (default: 'weight')
dim(number)optional: – Dimension over which to compute the norm (default: 0)

Options for weight_norm