kenmikanmi:

In this context, “2 dimensions” corresponds to batch_size = 2?


Two dimensions means that the size is a tuple containing two values, in this case the first value is the batch size and the second value is 3072.
The batch size is determined by your data loader.



 kenmikanmi:

In my model includes lambda layers as described. Can I correctly save them by…


Those Lambda layers have no parameters. There is nothing to save.