Operation.AtGet the value at the given indices: At.((ctx, t).@{idx}).
Set the value at the given indices: At.((ctx, t).@{idx} <- v); returns the updated context.
Get the value at the given index from a single-axis shape tensor.
Set the value at the given index for a single-axis shape tensor; returns the updated context.