replace

fun replace(i: Int, dim: Long): TensorShape
Content copied to clipboard

Replace dimension i with a new dimension size.

Return

The new changed TensorShape

Parameters

i

The target dimension to change.

dim

The new dimension size.