def draw(self, input_names=None, output_names=None, gray=False): if self.use_torch: Vs = [V.detach().cpu().numpy() for V in self.Vs] W = self.W.detach().cpu().numpy() else: Vs = self.Vs W = self.W ml.draw(Vs, W, input_names, output_names, gray)
def draw(self, inputNames=None, outputNames=None, gray=False): ml.draw(self.Vs, self.W, inputNames, outputNames, gray)
def draw(self, inputNames=None, outputNames=None): ml.draw(self.Vs, self.W, inputNames, outputNames)