Contents

weights

The key you use to access the weights of a layer in a neural network model.

Declaration

class var weights: MLParameterKey { get }

Discussion

The value type for the weights key is an MLMultiArray. You must scope this key with the name of the specific neural network layer whose weights you’d like to access. See scoped(to:).

See Also

Accessing neural network layer parameters