torch.nn.EmbeddingBagForwardOptions

export interface EmbeddingBagForwardOptions {
  /** Tensor of offsets for each bag (optional if input is 2D) */
  offsets?: Tensor;
  /** Weights for each sample (optional) */
  per_sample_weights?: Tensor;
}

offsets(Tensor)optional: – Tensor of offsets for each bag (optional if input is 2D)
per_sample_weights(Tensor)optional: – Weights for each sample (optional)

Options for EmbeddingBag.forward