... x2 = x2.view(len(x2),1,28,28) next(iter(custom_dl)) Fetch the flatten (fully ... the model until they reach the flattened layer: flatten_layer = nn.
確定！ 回上一頁
