ttnn.copy

ttnn.copy() None

Copies the elements from input_a into input_b. input_b is modified in place.

Both input tensors must be of equal shape.

Argument

Description

Data type

Valid range

Required

input_a

First tensor to copy

Tensor

Tensor of shape [W, Z, Y, X]

Yes

input_b

Second tensor to copy

Tensor

Tensor of shape [W, Z, Y, X]

Yes