tensorflow tf.dynamic_stitch因分段错误而崩溃

2ledvvac 于 2022-10-29 发布在其他

关注(0)|答案(1)|浏览(153)

问题类型

错误

来源

tensorflow 版本

tf 2.10

自定义代码

是的

操作系统平台和分发

Linux操作系统Ubuntu 20.04

移动的设备

没有回应 *

Python版本

没有回应 *

Bazel版本

没有回应 *

GCC/编译器版本

没有回应 *

CUDA/cuDNN版本

不适用

GPU型号和内存

没有回应 *

当前行为？

`tf.dynamic_stitch` crash with segmentation fault

重现问题的独立代码

import tensorflow as tf
import numpy as np
print(tf.__version__)

indices_data = np.array([[4, 3, 1, 7], [1, 5, 1, 0]])
updates_data = np.array([9, 10, 11, 12])
indices = tf.Variable(indices_data, dtype=tf.int32)
updates = tf.Variable(updates_data, dtype=tf.int32)
stitch = tf.dynamic_stitch(indices, updates)

1条答案

按热度按时间

yyhrrdl81#

是的，它在TF 2.10中得到了确认，但是它已经在TF 2.11.0rc1中得到了解决/保护，所以我不认为我们会在2.10补丁版本中挑选它。

import os
import tensorflow as tf
import numpy as np
print(tf.__version__)

indices_data = np.array([[4, 3, 1, 7], [1, 5, 1, 0]])
updates_data = np.array([9, 10, 11, 12])
indices = tf.Variable(indices_data, dtype=tf.int32)
updates = tf.Variable(updates_data, dtype=tf.int32)
stitch = tf.dynamic_stitch(indices, updates)

InvalidArgumentError: NodeDef expected inputs 'int32, int32, int32, int32' do not match 6 inputs specified; Op<name=DynamicStitch; signature=indices:N*int32, data:N*T -> merged:T; attr=N:int,min=1; attr=T:type>; NodeDef: {{node DynamicStitch}} [Op:DynamicStitch]

当然，如果你遵守第二个参数的文档形状约束，你也可以在TF 2.10中解决这个问题：
数据库|与具有相同类型的Tensor对象的索引具有相同长度的列表。

赞(0）回复(0）举报 2022-10-29

我来回答