Allow for more elemwise torch functions using broadcast_tensor
and vmap
#3224
This job was skipped
Loading
broadcast_tensor
and vmap
#3224