replaceLast

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

Replace the last dimension with a new dimension size.

Return

The new changed TensorShape

Parameters

dim

New size for the last dimensions