replaceFirst

fun replaceFirst(dim: Long): TensorShape
Content copied to clipboard

Replace the first dimension with a new dimension size.

Return

The new changed TensorShape.

Parameters

dim

New size for first dimension