Open
Description
Investigate if it makes sense to return non-Tensors from vmap. For example,
def f(x):
return 0.
vmap(f)(x)
{{ message }}
No one assigned
No type
No projects
No milestone
None yet
No branches or pull requests
Investigate if it makes sense to return non-Tensors from vmap. For example,
def f(x):
return 0.
vmap(f)(x)