How can I implement this function of torch.where and torch.sqrt in a better way so that autograd works?

Eklenme Tarih 1 hour ago
Active 19
Görüntülenme 89
S

How can I implement this function of torch.where and torch.sqrt in a better way so that autograd works?