Contents

batchFirst

A Boolean that indicates whether the input and output shape is batch first.

Declaration

var batchFirst: Bool { get }

Discussion

If batchFirst is true, the input and output shape is [batch, feature, 1, time steps]. Otherwise, the shape is [time steps, batch, feature], which is more efficient.

See Also

Inspecting LSTM Descriptors