torch::Tensor nonzero process time is too long?what is the reason?

How can i reduce the processing time about torch::Tensor nonzero about gpu

1 Like