ttnn.move

ttnn.move() None

Moves the elements of the input tensor arg0 to a location in memory with specified memory layout.

If no memory layout is specified, output memory will be the same as the input tensor memory config.

Argument

Description

Data type

Valid range

Required

arg0

Tensor to move

Tensor

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

Yes

arg1

MemoryConfig of tensor of TT accelerator device

tt_lib.tensor.MemoryConfig

Default is same as input tensor

No