AI模型的算子形状的确定方法、装置及相关设备

一种AI模型的算子形状的确定方法、装置及相关设备，涉及人工智能技术领域。确定运行AI模型的计算系统所提供的多种并行策略中，每种并行策略对AI模型所包括的第一算子的形状的影响程度；获取AI模型的目标并行配置，目标并行配置包括多种并行策略中每种并行策略的配置值；根据多种并行策略中每种并行策略的配置值、第一算子的基础形状及每种并行策略对第一算子的形状的影响程度，确定第一算子在目标并行配置下的形状。如此，根据每种并行策略对于AI模型中的算子的形状的影响程度，能够确定在目标并行配置下该算子的形状，无需依赖AI模型的具体结构，这能够有效提高确定算子形状的泛化性和通用性，同时也能提高确定算子形状的效率。