Is there a recommended way to implement an analogous operation to torch.where?

[balisujohn]

I am trying to get analogous behavior to

torch.where(score < 0, score * self.penalty, score / self.penalty)

where score is a tensor and self.penalty is a scalar.

[balisujohn]

I actually don't need an answer to this (going to this operation outside of ggml graph), so closing for now.