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