ttnn.assign

ttnn.assign(input_tensor_a: ttnn.Tensor, input_tensor_b: ttnn.Tensor) → None

Returns a new tensor which is a new copy of input tensor. Alternatively, copies input tensor input_a to input_b if their shapes and memory layouts match, and returns input_b tensor.

Input tensor must have BFLOAT16 data type. Output tensor will have BFLOAT16 data type.

Parameters:

• input_tensor_a (ttnn.Tensor) – Tensor assign is applied to. Tensor of shape [W, Z, Y, X].

• input_tensor_b (ttnn.Tensor) – Input tensor. Tensor of shape [W, Z, Y, X]. Only if performing assign to from tensor_a.