Flatten the dimensions of input. Examples: m = torch.nn.Flatten() x = torch.ones(1, 2, 4, 4) y = m(x) print(y.size()) # (1, 32). See also. torch.flatten(…) ...
確定! 回上一頁