关于Tensorflow上Keras的数组输入形状

时间:2016-03-09 02:45:55

标签: python tensorflow keras

我对keras关于Convolution2D图层的4D张量有疑问。

Keras医生说: 具有形状的4D张量:(样本,通道,行,列),如果dim_ordering =' th'或4D张量与形状:(样本,行,列,通道),如果dim_ordering =' tf'。

我使用' tf',我的输入怎么样?当我使用(samples, channels, rows, cols)时,它没问题,但是当我使用(samples, rows, cols, channels)作为输入时,它有一些问题。

1 个答案:

答案 0 :(得分:0)

Keras假设如果您使用tensorflow,那么您将使用(samples, channels, rows, cols)