Add mergesort switch. · 689dbeed20 - decagon-pytorch - ADARED:Code

IF YOU WOULD LIKE TO GET AN ACCOUNT, please write an email to s dot adaszewski at gmail dot com. User accounts are meant only to report issues and/or generate pull requests. This is a purpose-specific Git hosting for ADARED projects. Thank you for your understanding!

Kaynağa Gözat

Add mergesort switch.

master

Stanislaw Adaszewski 4 yıl önce

ebeveyn

670237c3f8

işleme

689dbeed20

1 değiştirilmiş dosya ile 1 ekleme ve 1 silme

Görünümü Böl

							
								+ 1
								
								- 1
							
src/triacontagon/cumcount.pyDosyayı Görüntüle
								
			@@ -24,7 +24,7 @@ def argunsort(s):
		
			def cumcount(a):

			    n = torch.numel(a)

			    s = np.argsort(a.detach().cpu().numpy())

			    s = np.argsort(a.detach().cpu().numpy(), kind='mergesort')

			    s = torch.tensor(s, device=a.device)

			    i = argunsort(s)

			    b = a[s]