skainet-data-transform/sk.ainet.data.transform/TransformScope/reshape

reshape

fun <I> Transform<I, Tensor<T, V>>.reshape(shape: Shape): Transform<I, Tensor<T, V>>(source)

fun <I> Transform<I, Tensor<T, V>>.reshape(vararg dims: Int): Transform<I, Tensor<T, V>>(source)

Chains a Reshape transform.