Searched refs:MyModuleParamDim (Results 1 – 1 of 1) sorted by relevance
43 class MyModuleParamDim(MyModuleBase): class140 mymod = MyModuleParamDim(torch.nn.Parameter(torch.randn(2, 5, 3)))141 mymod2 = MyModuleParamDim(torch.nn.Parameter(torch.randn(15, 3)))