深度学习入门:基于Python的理论与实现
# 可以同时处理单个数据和批量数据
def cross_entropy_error(y, t):
'''判断维数,将单个数据和批量数据处理成同样的形式,
方便后面shape[0]取出batch_size'''
if y.ndim == 1:
t = t.reshape(1, t.size)
y = y.reshape(1, y.size)
# shape[0]即最外层的个数,即batch_size(每一批的数据量)
batch_size = y.shape[0]
print("y.shape[0]:", y.shape[0])
print("y.shape:", y.shape)
# 返回平均交叉熵误差
return