code: def test_bn_init(self):
      params = (64,)
      assert equal_distribution(
          lambda *_: nn.BatchNorm2d(*params).weight,
          lambda _: torch.nn.BatchNorm2d(*params).weight.detach(),
      )
      assert equal_distribution(
          lambda *_: nn.BatchNorm2d(*params).bias,
          lambda _: torch.nn.BatchNorm2d(*params).bias.detach(),
      )
  
imports: import math
  import unittest
  import numpy
  import torch
  from tinygrad import nn, dtypes, Tensor
  from functools import partial
label_desc: Use image node_id 14 for calling the TestRandomness obj's underlying member method code with example usage: obj.test_bn_init() without return types