def forward(self, x, t1_indices, t0_indices):
self.input_shape = x.shape[0]
return HybridBlock.forward(self, x, t1_indices, t0_indices)
def forward(self, x):
self.layer_shape = x.shape
return HybridBlock.forward(self, x)
