Searched refs:get_random_tensor (Results 1 – 1 of 1) sorted by relevance
84 def get_random_tensor(self, size, dtype): member in BaseArg118 return self.get_random_tensor([], dtype).item()146 return self.get_random_tensor(self.size, dtype=dtype)