lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix vectorized map for numpy backend

Browse Source
This commit is contained in:
Francois Chollet 2023-11-02 12:42:28 -07:00
parent 2ff6f13c01
commit 11cbb29b30
1 changed files with 5 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
keras/backend/numpy/core.py
View File

@ -61,15 +61,11 @@ def cond(pred, true_fn, false_fn):
return false_fn()
def vectorized_map(function, elements):
if len(elements) == 1:
return function(elements)
else:
batch_size = elements[0].shape[0]
output_store = list()
for index in range(batch_size):
output_store.append(function([x[index] for x in elements]))
return np.stack(output_store)
def vectorized_map(function, x):
output_store = []
for element in x:
output_store.append(function(element))
return np.stack(output_store)
# Shape / dtype inference util