我试着根据mask中的值合并两个图像，在mask为1的所有点上，结果图像具有第一个图像的值，否则具有第二个图像的值。有人知道如何在pytorch中实现吗？使用numpy，可以使用

>>> import numpy as np
>>> img1 = np.random.rand(100,100,3)
>>> img2 = np.random.rand(100,100,3)
>>> mask = np.random.rand(100,100)>.5
>>> res = img2.copy()
>>> res[mask] = img1[mask]