code: def __init__(self, dropout=0, lock_dim=None, inplace=False):
      super(LockedDropout, self).__init__()
      self.dropout = nn.Dropout(p=dropout, inplace=inplace)
      self.lock_dim = lock_dim
  
imports: from typing import List, Optional
  from collections import deque
  import math
  import torch
  from torch import Tensor
  from torch import nn
  from torch.distributions import Distribution
  from torch.distributions import Normal
label_desc: Use image node_id 1 to create a new LockedDropout object from inherited base classes: nn with example: obj = LockedDropout(dropout, lock_dim, inplace)