nn.Dense的输入和输出
时间: 2024-06-04 21:13:48 浏览: 182
tensorflow-MNIST-nn.rar_MNIST_tensorflow
输入:nn.Dense接受一个张量作为输入,该张量的形状为(batch_size, input_size),其中batch_size表示一批数据中样本的数量,input_size表示每个样本的特征数。
输出:nn.Dense输出一个张量,该张量的形状为(batch_size, units),其中units表示输出的特征数。每个样本将被转换为具有units个特征的向量。
阅读全文