How to get indices of a specific value in a tensor in tensorflow-js?

前端 未结 0 1078
悲&欢浪女
悲&欢浪女 2021-01-20 00:10

For example, If I have a 2D tensor of [[1,3],[2,1]], how can I get indices that value 1 has occurre ? (it should give [0,0] and [1,1])

I too

相关标签:
回答
  • 消灭零回复
提交回复
热议问题