Closed mikeheddes closed 1 year ago
Selects elements from the input tensor if p=1 and from the other tensor when p=0. Fixes #161 .
input
p=1
other
p=0
While there seem to be some problems in the tests, they are due to problems with the dataset server. The tests that cover the changes are passing so I'm merging this PR.
Description
Selects elements from the
input
tensor ifp=1
and from theother
tensor whenp=0
. Fixes #161 .Checklist