Commit 989e78c4 authored by Eugene Brevdo's avatar Eugene Brevdo Committed by TensorFlower Gardener
Browse files

[TF Keras] keras.Flatten()(tensor : shaped (batch,)) -> tensor : shaped (batch, 1) 

This provides a consistent API for users who have scalar "channels".

PiperOrigin-RevId: 224528301
parent 840be302
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment