code: def conv_output_size(image_size, kernel_size, stride, padding=0):
      return int(
          ((image_size - kernel_size + (2 * padding)) / stride) + 1
      )
  
imports: from math import sqrt
  import torch
  from torch import nn, einsum
  import torch.nn.functional
  from einops import rearrange, repeat
  from einops.layers.torch import Rearrange
label_desc: Use image node_id 2 for calling a global function with example usage: conv_output_size(image_size, kernel_size, stride, padding) and returns: int