问题描述:
add.shard(((dp,1),()))
上面代码第二个输入张量的切分策略(),表示第二个输入张量不进行切分吗?
mul.shard(((dp,1), (1,)))
上面代码第二个输入张量的切分策略(1,),表示第二个输入张量不进行切分吗?
解决方法:
()这样表示不切分,(1,)这样表示张量第一个维度切成1份,实际效果是一样的。
add.shard(((dp,1),()))
上面代码第二个输入张量的切分策略(),表示第二个输入张量不进行切分吗?
mul.shard(((dp,1), (1,)))
上面代码第二个输入张量的切分策略(1,),表示第二个输入张量不进行切分吗?
()这样表示不切分,(1,)这样表示张量第一个维度切成1份,实际效果是一样的。