ReshapeTensorExtension<T> extension

on

Methods

flatten() Tensor<T>
Return the tensor collapsed into one dimension.
reshape(List<int> shape) Tensor<T>
Returns a reshaped view, if possible. If a dimension is set to -1 it is inferred automatically to keep the length constant.