以下python代码返回错误: CosDistance操作的输入操作数的形状最多应该有一个轴。
cosine_distance(query_output,document_output)
我想调试参数形状。是否有调试接口输出query_output和document_output的形状?
答案 0 :(得分:3)
对于python中的“debug”接口,你可以import pdb
,然后在你想要进入调试器的那一点插入一行:
pdb.set_trace()
从那里你可以做query_output.shape
这样的事情,它会告诉你形状。
答案 1 :(得分:0)
您可以通过输入变量的shape属性打印出形状。