Expose DirectMLX Identity in PyDirectML about directml HOT 2 CLOSED

import pydirectml as dml
import numpy as np

input_data = [
    [1,2,3],
    [4,5,6]
]

device = dml.Device(True, True)
builder = dml.GraphBuilder(device)

input_desc = dml.TensorDesc(dml.TensorDataType.FLOAT32, (1,1,2,3), (6,6,3,1))
input = dml.input_tensor(builder, 0, input_desc)
input_identity = dml.activation_identity(input)

input_transposed = dml.activation_identity(
    dml.reinterpret(input_identity, dml.TensorDataType.FLOAT32, (1,1,3,2), (6,6,1,3)))

input_broadcast = dml.activation_identity(
    dml.reinterpret(input_identity, dml.TensorDataType.FLOAT32, (1,1,6,3), (0,0,0,1)))

op = builder.build(dml.ExecutionFlags.NONE, [input_identity, input_transposed, input_broadcast])

input_bindings = (dml.Binding(input, input_data),)
output_data = device.compute(op, input_bindings, [input_identity, input_transposed, input_broadcast])

print(f"\nOriginal:\n{np.array(output_data[0], np.float32)}")
print(f"\nTransposed:\n{np.array(output_data[1], np.float32)}")
print(f"\nBroadcast First Row:\n{np.array(output_data[2], np.float32)}")