torch.ScatterShape

export type ScatterShape<S extends Shape, Dim extends number> = number extends Dim
  ? S
  : number extends S['length']
    ? S
    : LessThan<Dim, S['length']> extends true
      ? S // Valid
      : scatter_error_dim_out_of_range<Dim, S['length']>;

Sextends Shape

Dimextends number

Validate scatter dim is within bounds. scatter returns tensor with same shape as input.