zennit-crp
zennit-crp copied to clipboard
MPS fix for torch.take
A simple solution for the current problem that torch.take is not implemented for MPS.