How to make an if statement using a boolean Tensor

前端 未结 2 781
小鲜肉
小鲜肉 2021-01-11 12:38

How do I make an if statement using a boolean tensor? To be more precise, I\'m trying to compare a tensor of size 1 to a constant, checking to see if the value in the tenso

2条回答
提交回复
热议问题