Flatten ( full = False ) :: Module. Flatten x to a single dimension, e.g. at end of a model. full for rank-1 tensor. tst = Flatten() x = torch.randn(10,5,4) ...
確定! 回上一頁