Merge pull request #2781 from jmoenig/lists.js-stride-transpose

Update lists.js
pull/95/head
Jens Mönig 2021-02-14 11:38:37 +01:00 zatwierdzone przez GitHub
commit fd18966112
Nie znaleziono w bazie danych klucza dla tego podpisu
ID klucza GPG: 4AEE18F83AFDEB23
1 zmienionych plików z 4 dodań i 4 usunięć

Wyświetl plik

@ -726,18 +726,18 @@ List.prototype.strideTranspose = function () {
}
for (i = oldShape.length(); i > 1; i -= 1) {
product *= oldShape.at(1);
product *= oldShape.at(i);
newSizes.add(product, 1);
}
product = 1;
for (i = 1; i <= oldShape.length() - 1; i += 1) {
product *= oldShape.at(1);
product *= oldShape.at(i);
oldSizes.add(product);
}
for (i = 1; i <= oldFlat.length(); i += 1) {
newFlat.add(
newFlat.put(
oldFlat.at(i),
newIndex(i, oldSizes, newSizes)
newIndex(i-1, oldSizes, newSizes)+1
);
}
return newFlat.reshape(newShape);