code: def forward(self, x):
      maxes = x.amax(dim=self.dim, keepdim=True).detach()
      return x / maxes
  
imports: import torch
  import torch.nn.functional
  from einops import rearrange
  from torch import nn
label_desc: Use image node_id 2 for calling the DivideMax obj's underlying member method code with example usage: obj.forward(x) and returns: unknown