zhangyiss/gguf-tools
1
0
Fork 0
mirror of https://github.com/antirez/gguf-tools.git synced 2025-12-16 00:18:52 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Print tensor with correct strides

Browse Source
This commit is contained in:
Juarez Bochi
2024-01-05 09:59:59 -05:00
parent e48ca317ea
commit 50e79b9ec0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
gguf-tools.c
View File

@@ -368,7 +368,7 @@ void gguf_tools_inspect_weights(const char *filename, const char *tname, uint64_
uint64_t strides[GGUF_TENSOR_MAX_DIM] = {0}; uint64_t strides[GGUF_TENSOR_MAX_DIM] = {0};
strides[tensor.ndim-1] = 1; strides[tensor.ndim-1] = 1;
for (int j = tensor.ndim - 2; j >= 0; j--) { for (int j = tensor.ndim - 2; j >= 0; j--) {
strides[j] = tensor.dim[j + 1] * strides[j + 1]; strides[j] = tensor.dim[tensor.ndim - 2 - j] * strides[j + 1];
} }
const int ident = 4; const int ident = 4;