torch.embedding_bag_error_requires_2d_input

export type embedding_bag_error_requires_2d_input<ActualRank extends number> =
  ErrorShape<`embedding_bag_error_requires_2d_input<actual_rank=${ActualRank}>`>;

ActualRankextends number

Embedding bag requires 2D input error. Occurs when embedding_bag is called with 1D input and offsets. Use 2D input where each row is a bag instead.

Examples

const indices = torch.tensor([0, 1, 2, 3]);  // 1D
const result = F.embedding_bag(indices, weight, offsets);  // ERROR
// Use 2D input: torch.tensor([[0, 1], [2, 3]])