並前往https://stackoverflow.com/questions/68924374/use-torch-square-inside-a-nn-sequential-layer-in-pytorch