Spec-Zone.ru › TensorFlow C++ 2.9

tensorflow::ops::BroadcastTo

#include <array_ops.h>

Распространение массива для совместимой формы.

Краткое описание

Распространение — это процесс приведения массивов к совместимым формам для арифметических операций. Две формы совместимы, если для каждой пары измерений либо они равны, либо одно из них равно единице. При попытке распространить Tensor на форму, процесс начинается с последующих измерений и идёт в направлении старших измерений.

Например,

x = tf.constant([1, 2, 3]) y = tf.broadcast_to(x, [3, 3]) print(y) tf.Tensor( [[1 2 3] [1 2 3] [1 2 3]], shape=(3, 3), dtype=int32)

В приведённом примере входной Tensor с формой [1, 3] распространяется на выходной Tensor с формой [3, 3].

При выполнении операций с распространением, таких как умножение тензора на скаляр, распространение (обычно) даёт выигрыш во времени или памяти, так как распространяемый тензор никогда не материализуется.

Однако, broadcast_to не несёт с собой никаких таких преимуществ. Создаваемый тензор занимает всю память, соответствующую форме распространения. (В контексте графа broadcast_to может быть объединён с последующей операцией и затем оптимизирован.)

Аргументы:

  • scope: Объект Scope
  • input: Tensor для распространения.
  • shape: Одномерный int Tensor. Форма желаемого результата.

Возвращаемые значения:

  • Output: Tensor.
Конструкторы и деструкторы
BroadcastTo(const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input shape)
Публичные атрибуты
operation
Operation
output
::tensorflow::Output
Публичные функции
node() const
::tensorflow::Node *
operator::tensorflow::Input() const
operator::tensorflow::Output() const

Публичные атрибуты

operation

Operation operation

output

::tensorflow::Output output

Публичные функции

BroadcastTo

 BroadcastTo(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input shape
)

node

::tensorflow::Node * node() const 

operator::tensorflow::Input

operator::tensorflow::Input() const 

operator::tensorflow::Output

operator::tensorflow::Output() const 

© 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/versions/r2.9/api_docs/cc/class/tensorflow/ops/broadcast-to

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API