我试图从(None,7)
中获取tf.keras.utils.timeseries_dataset_from_array
形状,但它返回了(None, None,7)
形状。
tensorflow不允许设置batch_size=None
或batch_size=0
以减少batch_size维度。
那么有什么办法可以减少这个维度吗?
PS:我正在尝试将它安装到我的自定义层,它只接受2的n_dim数据。
发布于 2021-12-13 09:26:45
试着做一个批次,并得到它的大小为1,然后使用tf.squeeze删除该尺寸1?确保您指定了正确的轴。
https://stackoverflow.com/questions/70338539
复制