Class BroadcastTo

• Defined in File broadcast_to.h

Inheritance Relationships

Base Type

• public mindspore::ops::PrimitiveC (Class PrimitiveC)

Class Documentation

class BroadcastTo : public mindspore::ops::PrimitiveC

Broadcasts input tensor to a given shape. Refer to Python API mindspore.ops.BroadcastTo for more details.

Public Functions

inline BroadcastTo()

Constructor.

~BroadcastTo() = default

Destructor.

void Init(const std::vector<int64_t> &shape)

Init. Refer to the parameters of Python API mindspore.ops.BroadcastTo for the inputs.

void set_shape(const std::vector<int64_t> &shape)

Set shape.

std::vector<int64_t> get_shape() const

Get shape.

Returns

shape.