Python tensor_list_split примеры использования

Пример #1

Файл: test_tensor_list_input_output.py Проект: Sodu-Qinming/Oneflow

 def foo_job(input_def: oft.ListListNumpy.Placeholder(shape=(2, 5, 4),
                                                      dtype=flow.float)):
     output_0, output_1 = flow.tensor_list_split(input_def)
     test_case.assertTrue(
         np.array_equal(output_0.numpy(), input_0.squeeze()))
     test_case.assertTrue(
         np.array_equal(output_1.numpy(), input_1.squeeze()))

Пример #2

Файл: test_tensor_list_split.py Проект: zyg11/oneflow

 def tensor_list_split_job(
     input_def: oft.ListListNumpy.Placeholder(
         shape=tuple(input_static_shape), dtype=flow.float
     ),
 ):
     outputs = flow.tensor_list_split(input_def)
     return outputs