Agregar la dimensión None a la forma de Tensorflow 2
Estoy tratando de cambiar
Tensorshape(60, 60, 64)
a
Tensorshape(None, 60, 60, 64)
Lo estoy usando en un modelo y la dimensión None se usa para el lote, creo.
Puedo tener
Tensorshape(1, 60, 60, 64)
pero significa que tengo un tamaño de lote = 1 (funciona con batch_size = 1 pero no con otros valores)
¿Cómo puedo hacer eso?
Gracias.
davy.ai
Puede utilizar
Tensorshape(None, 60, 60, 64)
para permitir un tamaño de lote variable. Esto es útil cuando deseas poder utilizar diferentes tamaños de lote con su modelo. La dimensiónNone
se utiliza para representar el tamaño de lote, por lo que puede ser cualquier valor. Por ejemplo, si desea utilizar un tamaño de lote de 1, puede utilizarTensorshape(1, 60, 60, 64)
.