galax
Contents:
API Documentation
galax.heterograph
galax.function
galax.function.apply_edges
galax.function.apply_nodes
galax.function.copy_e
galax.function.copy_edge
galax.function.copy_src
galax.function.copy_u
galax.function.e_add_u
galax.function.e_add_v
galax.function.e_div_u
galax.function.e_div_v
galax.function.e_dot_u
galax.function.e_dot_v
galax.function.e_mul_u
galax.function.e_mul_v
galax.function.e_sub_u
galax.function.e_sub_v
galax.function.segment_max
galax.function.segment_mean
galax.function.segment_min
galax.function.segment_softmax
galax.function.u_add_e
galax.function.u_add_v
galax.function.u_div_e
galax.function.u_div_v
galax.function.u_dot_e
galax.function.u_dot_v
galax.function.u_mul_e
galax.function.u_mul_v
galax.function.u_sub_e
galax.function.u_sub_v
galax.function.v_add_e
galax.function.v_add_u
galax.function.v_div_e
galax.function.v_div_u
galax.function.v_dot_e
galax.function.v_dot_u
galax.function.v_mul_e
galax.function.v_mul_u
galax.function.v_sub_e
galax.function.v_sub_u
galax.function.ReduceFunction
galax.core
galax.data
galax.nn
galax
»
API Documentation
»
galax.function
»
galax.function.segment_max
View page source
galax.function.segment_max
galax.function.
segment_max
(
*
args
,
**
kwargs
)
[source]
Alias of jax.ops.segment_max with nan_to_num.