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)