aparler/PyNorch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fixed sum axis with keepdim

Browse source
This commit is contained in:
lucasdelimanogueira 2024-05-17 19:29:16 -03:00
parent 5dd3f3d202
commit 2df6c36d99
6 changed files with 20 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

5
norch/csrc/cpu.cpp
View file

@ -205,7 +205,7 @@ void log_tensor_cpu(Tensor* tensor, float* result_data) {
}
}
void sum_tensor_cpu(Tensor* tensor, float* result_data, int size, int* result_shape, int axis, bool keepdim) {
void sum_tensor_cpu(Tensor* tensor, float* result_data, int size, int* result_shape, int axis) {
if (axis == -1) {
// Sum over all elements
float sum = 0.0;
@ -232,9 +232,6 @@ void sum_tensor_cpu(Tensor* tensor, float* result_data, int size, int* result_sh
result_data[j] += tensor->data[index + i * axis_stride];
}
}
for (int j = 0; j < size; j++) {
printf("%f", result_data[j]);
}
}
}