nn_permute_output_shape: Apply a permute layer's 'dims' to a per-sample shape

View source: R/nn_layers.R

nn_permute_output_shapeR Documentation

Apply a permute layer's 'dims' to a per-sample shape

Description

Shared by the sequential and functional shape passes so the two cannot disagree.

Usage

nn_permute_output_shape(dims, shape)

Arguments

dims

Permutation in 'aperm' (source) order.

shape

Per-sample input shape.

Value

The permuted per-sample shape.


ggmlR documentation built on Oct. 23, 2026, 1:07 a.m.