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

fix pow, div, log operations

Browse source
This commit is contained in:
lucasdelimanogueira 2024-05-06 10:18:12 -03:00
parent ca4a0f0dbc
commit 065c6a0f29
15 changed files with 160 additions and 51 deletions
Download patch file Download diff file Expand all files Collapse all files

7
norch/csrc/cpu.cpp
View file

@ -47,6 +47,13 @@ void tensor_div_scalar_cpu(Tensor* tensor, float scalar, float* result_data) {
}
}
void tensor_div_tensor_cpu(Tensor* tensor1, Tensor* tensor2, float* result_data) {
for (int i = 0; i < tensor1->size; i++) {
result_data[i] = tensor1->data[i] / tensor2->data[i];
}
}
void matmul_tensor_cpu(Tensor* tensor1, Tensor* tensor2, float* result_data) {
for (int i = 0; i < tensor1->shape[0]; i++) {