Expand description
The Shape type and utilities.
Structs§
- A shape with each dimension being a fixed
usize. - A shape with each dimension corresponding to Size.
- A size expression of the form
F * pow(batch_size, N).
Enums§
Functions§
- Infer what the batch size should be for the given input shapes.
- Same as infer_batch_size, except for individual dimensions.