AlphaBetaGamma96:

if you want to change it across the entire script you need to use torch.set_default_dtype(torch.cfloat)


I am using google colab. When I try this command I get the following: “Your session crashed for an unknown reason.” This seems to occur for “cfloat” and “complex64”, but the code runs fine for “float32” and “float64.” What could be the issue here?