Broadcast

BroadcastOperator(API) broadcasts all input tensors according to attribute targetShape.

Few notes:

  • The inputTensors and outputTensors should have same number of tensors.

We follow same braodcasting rule in Scipy.

Initialization

let shape = TensorShape(dataType: .float, shape: [3, 3, 15])
let op = BroadcastOperator(targetShape: shape)