예제 #1
0
 def print_shape(self, input_shape):
     print("super-resolution {")
     SequentialMaker.print_shape(self.seq_,
                                 torch.FloatTensor(np.zeros(input_shape)),
                                 intent_size=2)
     print("}")
예제 #2
0
 def print_shape(self, input_shape):
     print("text-encode {")
     SequentialMaker.print_shape(self.seq_,
                                 torch.LongTensor(np.zeros(input_shape)),
                                 intent_size=2)
     print("}")
예제 #3
0
 def print_shape(self, input_shape):
     print("audio-decoder {")
     SequentialMaker.print_shape(self.seq_,
                                 torch.FloatTensor(np.zeros(input_shape)),
                                 intent_size=2)
     print("}")