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

2024-04-29 17:19:45 发布

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

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

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

tf.reshape(w, [-1])

以及

tf.squeeze(w)

是吗?


Tags: 方法tf矢量水平差异性能squeezereshape
1条回答
网友
1楼 · 发布于 2024-04-29 17:19:45

^{}^{}都是“便宜的”,因为它们只对给定张量的元数据(即形状)进行操作,而不修改数据本身。两者中的tf.reshape()稍有simpler logic internally,但两者的性能应该是不可区分的。

相关问题 更多 >