Searched defs:ArgmaxModule (Results 1 – 1 of 1) sorted by relevance
778 class ArgmaxModule(torch.nn.Module): class794 class ArgmaxModule(torch.nn.Module): class