probly.method.het_nets.torch.remove_layer¶ probly.method.het_nets.torch.remove_layer(obj: nn.Softmax, state: State) → tuple[nn.Module, State][source]¶ Remove the softmax layer.