Change to .shape instead of method .get_shape().
.shape is the recommended way of getting the shape of a Tensor. PiperOrigin-RevId: 203547575
Loading
Please sign in to comment
.shape is the recommended way of getting the shape of a Tensor. PiperOrigin-RevId: 203547575