Changed the signatures of NamedTensorBase._broadcast_order and NamedTensorBase.._mask_broadcast_order to take in names and not NamedTensors.
This is needed since NTorch.index_copy_ needs to know the dimension names and size of self[index]. Getting them using NTorch._index_base might be more efficient than getting the actual tensor itself.
Changed the signatures of
NamedTensorBase._broadcast_orderandNamedTensorBase.._mask_broadcast_orderto take in names and not NamedTensors.This is needed since
NTorch.index_copy_needs to know the dimension names and size ofself[index]. Getting them usingNTorch._index_basemight be more efficient than getting the actual tensor itself.