MCPcopy Create free account
hub / github.com/PaddlePaddle/Paddle2ONNX / forward

Method forward

tests/test_tensor_array.py:24–34  ·  view source on GitHub ↗
(self, input)

Source from the content-addressed store, hash-verified

22 super(BaseNet, self).__init__()
23
24 def forward(self, input):
25 arr = paddle.tensor.create_array(dtype="float32")
26 x = paddle.full(shape=[1, 3], fill_value=5, dtype="float32")
27 i = paddle.zeros(shape=[1], dtype="int64")
28 paddle.tensor.array_write(x, i, array=arr)
29 item = paddle.tensor.array_read(arr, i)
30 new_arr = paddle.tensor.create_array(dtype="float32")
31 j = paddle.to_tensor([paddle.tensor.array_length(new_arr)], dtype="int64")
32 paddle.tensor.array_write(item, j, array=new_arr)
33 new_item = paddle.tensor.array_read(new_arr, j)
34 return input + new_item
35
36
37@_test_only_pir

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected