jax.lax.conv_dimension_numbers#
- jax.lax.conv_dimension_numbers(lhs_shape, rhs_shape, dimension_numbers)[源代码]#
将卷积 dimension_numbers 转换为 ConvDimensionNumbers。
- 参数:
lhs_shape – 非负整数元组，卷积输入的形状。
rhs_shape – 非负整数元组，卷积核的形状。
dimension_numbers – None 或字符串元组/列表或 ConvDimensionNumbers 对象。
- 返回:
一个 ConvDimensionNumbers 对象，它以 lax 函数使用的规范形式表示 dimension_numbers。
- 返回类型: