bits(key[, shape, dtype]) 以无符号整数的形式采样均匀比特。
categorical(key, logits[, axis, shape]) 从分类分布中采样随机值。...choice(key, a[, shape, replace, p, axis]) 从给定数组中生成随机样本。...addressable_devices_indices_map(global_shape)
从可寻址设备到它们包含的数组数据切片的映射。...shard_shape(global_shape)
返回每个设备上数据的形状。
此函数返回的分片形状是从global_shape和分片属性计算得出的。...这不是有效的也不是可靠的序列化。输出在不同调用之间没有一致性的保证。
如果不可用,则返回None,例如基于后端、编译器或运行时。