Finally, we use the numpy expand_dims function to change the array (or Tensor) from (num_samples, 28, 28) to (num_samples, 28, 28, 1)—one dimension.
確定! 回上一頁