tf.keras.ops.meshgrid
Создаёт сетки координат на основе векторов координат.
tf.keras.ops.meshgrid(
*x, indexing='xy'
)
Если в качестве входных данных заданы N одномерные тензоры T0, T1, ..., TN-1 соответствующей длины S0, S1, ..., SN-1, функция создаёт N N-мерных тензоров G0, G1, ..., GN-1, каждый из которых имеет форму (S0, ..., SN-1), где выходной тензор Gi формируется путём расширения Ti до формы результата.
| Аргументы | |
|---|---|
x | Одномерные тензоры, представляющие координаты сетки. |
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