I need to use elementwise mutliplication (torch.mul) and matrix multiplication (torch.mm 29) inside model definition in forward pass in CPU/Cuda. Is there any CUDA version of these operations which can be used w/o switching into CPU mode?