Skip to contents

Broadcast scalar tensors to match the shape of non-scalar tensors. All non-scalar tensors must have the same shape.

Usage

nv_broadcast_scalars(...)

Arguments

...

(tensorish)
Tensors to broadcast. Scalars will be broadcast to the common non-scalar shape.

Value

(list() of tensorish)
List of broadcasted tensors.