reshape

open override fun reshape(vararg dims: Long)

Chain-like setter to set up input shape.

Parameters

dims

The input shape.