zennit-crp icon indicating copy to clipboard operation
zennit-crp copied to clipboard

MPS fix for torch.take

Open Dekazu opened this issue 5 months ago • 0 comments

A simple solution for the current problem that torch.take is not implemented for MPS.

Screenshot

Dekazu avatar Sep 04 '24 09:09 Dekazu