Hi,

I am trying to apply condition on tensors. What I am trying to achieve is: When mask is true then use the value from X otherwise Y. Logic works fine using `np.where`

function. I am trying to achieve same using `torch.where`

. Anything i am missing on the following?

`weights_x`

and `weights_y`

both have size:

```
torch.Size([32, 1, 5, 5])
torch.Size([32, 1, 5, 5])
```

`mask= th.tensor([False, True, False, True, True, False, False, True, True, True, False, True, False, True, True, True, False, True, True, False, False, True, False, False, False, False, True, False, False, False, True, True])`

**Numpy version**

`np.where(mask, weights_x, weights_y)`

**Pytorch version**

`th.where(mask == True, weights_x, weights_y)`

**Error:**

```
if isinstance(args_, tuple):
--> 414 response = command_method(*args_, **kwargs_)
415 else:
416 response = command_method(args_, **kwargs_)
RuntimeError: The size of tensor a (32) must match the size of tensor b (5) at non-singleton dimension 3
```