spherical_harmonics(ir_vec, ls[, layout])
|
下标: vector[],...,vector[],Yl[] |
fully_connected_tensor_product(irreps1, ...)
|
下标: weights[uvw],lhs[iu],rhs[jv],output[kw] |
channelwise_tensor_product(irreps1, irreps2)
|
下标: weights[uv],lhs[iu],rhs[jv],output[kuv] |
elementwise_tensor_product(irreps1, irreps2)
|
下标: lhs[iu],rhs[ju],output[ku] |
linear(irreps_in, irreps_out)
|
下标: weights[uv],input[iu],output[iv] |
symmetric_contraction(irreps_in, irreps_out, ...)
|
下标: weights[u],input[u],output[u] |
x_rotation(irreps[, lmax])
|
下标: phi[],input[u],output[u] |
y_rotation(irreps[, lmax])
|
下标: phi[],input[u],output[u] |
yxy_rotation(irreps[, lmax])
|
下标: gamma[],beta[],alpha[],input[u],output[u] |
