Physics

An extension module which optimizes common physics operators. The input tensors can be either polar or cartesian, however one needs to specify the source coordinate type explicitly using the Cartesian or Polar submodules. Similar to what was previously discussed, the inputs can be either combined or separate, depending on preference.

Physics for Combined Tensors (Cartesian)

pyc.Physics.Cartesian.P2(torch.tensor pmc) torch.tensor

Computes the scalar square of the 3-momentum vector of a cartesian tensor.

Parameters:

pmc (torch.tensor) – A tensor with the four vector being in cartesian form.

pyc.Physics.Cartesian.P(torch.tensor pmc) torch.tensor

Computes the scalar of the 3-momentum vector of a cartesian tensor.

Parameters:

pmc (torch.tensor) – A tensor with the four vector being in cartesian form.

pyc.Physics.Cartesian.Beta2(torch.tensor pmc) torch.tensor

Computes the \beta^2 for the Lorentz Factor from the 3-momentum vector of a cartesian tensor.

Parameters:

pmc (torch.tensor) – A tensor with the four vector being in cartesian form.

pyc.Physics.Cartesian.Beta(torch.tensor pmc) torch.tensor

Computes the \beta for the Lorentz Factor from the 3-momentum vector of a cartesian tensor.

Parameters:

pmc (torch.tensor) – A tensor with the four vector being in cartesian form.

pyc.Physics.Cartesian.M2(torch.tensor pmc) torch.tensor

Computes the square of the invariant mass from a cartesian 4-vector.

Parameters:

pmc (torch.tensor) – A tensor with the four vector being in cartesian form.

pyc.Physics.Cartesian.M(torch.tensor pmc) torch.tensor

Computes the invariant mass from a cartesian 4-vector.

Parameters:

pmc (torch.tensor) – A tensor with the four vector being in cartesian form.

pyc.Physics.Cartesian.Mt2(torch.tensor pmc) torch.tensor

Computes the square of the transverse invariant mass from a cartesian 4-vector.

Parameters:

pmc (torch.tensor) – A tensor with the four vector being in cartesian form.

pyc.Physics.Cartesian.Mt(torch.tensor pmc) torch.tensor

Computes the transverse invariant mass from a cartesian 4-vector.

Parameters:

pmc (torch.tensor) – A tensor with the four vector being in cartesian form.

pyc.Physics.Cartesian.Theta(torch.tensor pmc) torch.tensor

Computes the particle’s \theta from a cartesian 4-vector.

Parameters:

pmc (torch.tensor) – A tensor with the four vector being in cartesian form.

pyc.Physics.Cartesian.DeltaR(torch.tensor pmc1, torch.tensor pmc2) torch.tensor

Computes the \Delta R between two particle 4-vectors.

Parameters:
  • pmc1 (torch.tensor) – The cartesian 4-vector of particle-1.

  • pmc2 (torch.tensor) – The cartesian 4-vector of particle-2.

Physics for Separated Tensors (Cartesian)

pyc.Physics.Cartesian.P2(torch.tensor px, torch.tensor py, torch.tensor pz) torch.tensor
pyc.Physics.Cartesian.P(torch.tensor px, torch.tensor py, torch.tensor pz) torch.tensor
pyc.Physics.Cartesian.Beta2(torch.tensor px, torch.tensor py, torch.tensor pz, torch.tensor e) torch.tensor
pyc.Physics.Cartesian.Beta(torch.tensor px, torch.tensor py, torch.tensor pz, torch.tensor e) torch.tensor
pyc.Physics.Cartesian.M2(torch.tensor px, torch.tensor py, torch.tensor pz, torch.tensor e) torch.tensor
pyc.Physics.Cartesian.M(torch.tensor px, torch.tensor py, torch.tensor pz, torch.tensor e) torch.tensor
pyc.Physics.Cartesian.Mt2(torch.tensor pz, torch.tensor e) torch.tensor
pyc.Physics.Cartesian.Mt(torch.tensor pz, torch.tensor e) torch.tensor
pyc.Physics.Cartesian.Theta(torch.tensor px, torch.tensor py, torch.tensor pz) torch.tensor
pyc.Physics.Cartesian.DeltaR(torch.tensor px1, torch.tensor px2, torch.tensor py1, torch.tensor py2, torch.tensor pz1, torch.tensor pz2) torch.tensor

Physics for Combined Tensors (Polar)

pyc.Physics.Polar.P2(torch.tensor pmu) torch.tensor
pyc.Physics.Polar.P(torch.tensor pmu) torch.tensor
pyc.Physics.Polar.Beta2(torch.tensor pmu) torch.tensor
pyc.Physics.Polar.Beta(torch.tensor pmu) torch.tensor
pyc.Physics.Polar.M2(torch.tensor pmu) torch.tensor
pyc.Physics.Polar.M(torch.tensor pmu) torch.tensor
pyc.Physics.Polar.Mt2(torch.tensor pmu) torch.tensor
pyc.Physics.Polar.Mt(torch.tensor pmu) torch.tensor
pyc.Physics.Polar.Theta(torch.tensor pmu) torch.tensor
pyc.Physics.Polar.DeltaR(torch.tensor pmu1, torch.tensor pmu2) torch.tensor

Physics for Separated Tensors (Polar)

pyc.Physics.Polar.P2(torch.tensor pt, torch.tensor eta, torch.tensor phi) torch.tensor
pyc.Physics.Polar.P(torch.tensor pt, torch.tensor eta, torch.tensor phi) torch.tensor
pyc.Physics.Polar.Beta2(torch.tensor pt, torch.tensor eta, torch.tensor phi, torch.tensor e) torch.tensor
pyc.Physics.Polar.Beta(torch.tensor pt, torch.tensor eta, torch.tensor phi, torch.tensor e) torch.tensor
pyc.Physics.Polar.M2(torch.tensor pt, torch.tensor eta, torch.tensor phi, torch.tensor e) torch.tensor
pyc.Physics.Polar.M(torch.tensor pt, torch.tensor eta, torch.tensor phi, torch.tensor e) torch.tensor
pyc.Physics.Polar.Mt2(torch.tensor pt, torch.tensor eta, torch.tensor e) torch.tensor
pyc.Physics.Polar.Mt(torch.tensor pt, torch.tensor eta, torch.tensor e) torch.tensor
pyc.Physics.Polar.Theta(torch.tensor pt, torch.tensor eta, torch.tensor phi) torch.tensor
pyc.Physics.Polar.DeltaR(torch.tensor eta1, torch.tensor eta2, torch.tensor phi1, torch.tensor phi2) torch.tensor