Is there any specific reason why torch.smm is only implemented on CPU but not GPU whereas torch.hsmm is implemented on both?
i think it’s just a missing feature. no one implemented it yet.
Is there any specific reason why torch.smm is only implemented on CPU but not GPU whereas torch.hsmm is implemented on both?
i think it’s just a missing feature. no one implemented it yet.