Expand description
Tensor utilities.
Enums§
- Reshape
Action - The action to take for a reshape operation.
- Reshape
Analysis - The reshape kind.
Functions§
- broadcast_
strides - Calculate the new strides given added batch dimensions.
- contiguous_
strides - Computes the strides for a contiguous tensor with the given shape.
- is_
contiguous - Check if the current tensor is contiguous.
- reshape_
action - Returns the proper action to take when reshaping a tensor.
- reshape_
analysis - Returns the analysis of a reshape operation.