package onnx
Ordering
- Alphabetic
Visibility
- Public
- All
Type Members
-
class
Reshape[T] extends AbstractModule[Activity, Tensor[T], T]
Reshape the input tensor similar to numpy.reshape.
Reshape the input tensor similar to numpy.reshape. First input is the data tensor, second input is a shape tensor which specifies the output shape. It outputs the reshaped tensor.
- T
The numeric type in this module parameters.