tf.keras.ops.meshgrid
Создаёт сетки координат из векторов координат.
tf.keras.ops.meshgrid(
*x, indexing='xy'
)
Учитывая N 1-мерные тензоры T0, T1, ..., TN-1 в качестве входных данных с соответствующими длинами S0, S1, ..., SN-1, это создаёт N N-мерные тензоры G0, G1, ..., GN-1, каждый из которых имеет форму (S0, ..., SN-1), где выходной Gi тензор формируется путем расширения Ti до формы результата.
| Аргументы | |
|---|---|
x | 1-мерные тензоры, представляющие координаты сетки. |
indexing | "xy" или "ij". "xy" — декартовы координаты; "ij" — матричная индексация вывода. По умолчанию — "xy". |
| Возвращаемые значения | |
|---|---|
| Последовательность из N тензоров. |
Пример:
from keras.src import ops x = ops.array([1, 2, 3]) y = ops.array([4, 5, 6])
grid_x, grid_y = ops.meshgrid(x, y, indexing="ij")
grid_x
array([[1, 1, 1],
[2, 2, 2],
[3, 3, 3]])
grid_y
array([[4, 5, 6],
[4, 5, 6],
[4, 5, 6]])
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/ops/meshgrid