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

autograd add function

Browse source
This commit is contained in:
lucasdelimanogueira 2024-04-30 20:09:08 -03:00
parent d4751341bd
commit 407242bd7e
15 changed files with 229 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

14
norch/csrc/cpu.cpp
View file

@ -61,3 +61,17 @@ void sum_tensor_cpu(Tensor* tensor, float* result_data) {
*result_data = sum;
}
void ones_like_tensor_cpu(Tensor* tensor, float* result_data) {
for (int i = 0; i < tensor->size; i++) {
result_data[i] = 1.0;
}
}
void zeros_like_tensor_cpu(Tensor* tensor, float* result_data) {
for (int i = 0; i < tensor->size; i++) {
result_data[i] = 0.0;
}
}