Function random_tensor

tests/gemm_dequantize/th_gemm_dequantize.py:4–5

(shape, dtype, device, mean=0, std=1)

import unittest

def random_tensor(shape, dtype, device, mean=0, std=1):
    return torch.empty(shape, dtype=dtype, device=device).normal_(mean, std)
  
class TestGemmDequantize(unittest.TestCase):
    def setUp(self) -> None: