Contents

init(dimensions:)

Creates a transpose layer with the dimensions you specify.

Declaration

convenience init?(dimensions: [Int])

Parameters

  • dimensions:

    An array that represents the ordering of dimensions.

Discussion

The dimensions array contains an input axis source for each output axis. In other words, the nth element in the dimensions array specifies the input axis source for the nth axis in the output. You can’t transpose the batch dimension, which is typically axis 0.