0

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]]
cserpell
  • 716
  • 1
  • 7
  • 17

0 Answers0