jax.lax.conv_dimension_numbers#
- jax.lax.conv_dimension_numbers(lhs_shape, rhs_shape, dimension_numbers)[source]#
將卷積 dimension_numbers 轉換為 ConvDimensionNumbers。
- 參數:
lhs_shape – 非負整數的元組，卷積輸入的形狀。
rhs_shape – 非負整數的元組，卷積核的形狀。
dimension_numbers – None 或字串的元組/列表或 ConvDimensionNumbers 物件。
- 回傳:
一個 ConvDimensionNumbers 物件，表示 lax 函數使用的規範形式的 dimension_numbers。
- 回傳型別: