在TensorFlow中展平包含向量的二维张量的最佳方法?

2024-05-20 09:10:08 发布

您现在位置:Python中文网/ 问答频道 /正文

把二维张量(实际上是一个水平或垂直矢量)展平成一维张量的最有效方法是什么?

在性能方面是否存在以下差异:

tf.reshape(w, [-1])

以及

tf.squeeze(w)

是吗?


Tags: 方法tf矢量水平差异性能squeezereshape