Open
Description
When running the nested tensors tutorial on google colab builds from master instead of main. The master branch version is non-functional, main branch version appears to work correctly:
- nested tensors created with
torch.nested_tensor
instead oftorch.nested.nested_tensor
- the
mha_netsed
function handle batch size inference incorrectly.