I would like to check which elements of an mxnet Tensor
are in a python list. How can I do it?
For example, let's say I've got the Tensor
[[1 2 3]
[4 5 6]]
and I ask if elements are in list [2, 5]
, then I should get the answer
[[0 1 0]
[0 1 0]]