var a = torch.randn(3, 3).to(torch.bfloat16);
a.print();